From 4f1a1d76b7ca44b8ee8e85b0ee014ae5eca2592f Mon Sep 17 00:00:00 2001
From: Alexander Hellbom <alexander.hellbom@gmail.com>
Date: Thu, 4 Jan 2018 10:49:18 +0100
Subject: [PATCH] Try to perform an exact match before partial matching

---
 pagerduty/data_source_pagerduty_vendor.go     | 15 +++++-
 .../data_source_pagerduty_vendor_test.go      | 52 +++++++++----------
 2 files changed, 38 insertions(+), 29 deletions(-)

diff --git a/pagerduty/data_source_pagerduty_vendor.go b/pagerduty/data_source_pagerduty_vendor.go
index fee4820d4..5635dbc07 100644
--- a/pagerduty/data_source_pagerduty_vendor.go
+++ b/pagerduty/data_source_pagerduty_vendor.go
@@ -49,15 +49,26 @@ func dataSourcePagerDutyVendorRead(d *schema.ResourceData, meta interface{}) err
 
 	var found *pagerduty.Vendor
 
-	r := regexp.MustCompile("(?i)" + searchName)
+	er := regexp.MustCompile(fmt.Sprintf("^(?i)%s$", searchName))
 
 	for _, vendor := range resp.Vendors {
-		if r.MatchString(vendor.Name) {
+		if er.MatchString(vendor.Name) {
 			found = vendor
 			break
 		}
 	}
 
+	// We didn't find an exact match, so let's fallback to partial matching.
+	if found == nil {
+		pr := regexp.MustCompile("(?i)" + searchName)
+		for _, vendor := range resp.Vendors {
+			if pr.MatchString(vendor.Name) {
+				found = vendor
+				break
+			}
+		}
+	}
+
 	if found == nil {
 		return fmt.Errorf("Unable to locate any vendor with the name: %s", searchName)
 	}
diff --git a/pagerduty/data_source_pagerduty_vendor_test.go b/pagerduty/data_source_pagerduty_vendor_test.go
index 59b639aa2..8d019711e 100644
--- a/pagerduty/data_source_pagerduty_vendor_test.go
+++ b/pagerduty/data_source_pagerduty_vendor_test.go
@@ -1,14 +1,13 @@
 package pagerduty
 
 import (
-	"fmt"
 	"testing"
 
 	"github.com/hashicorp/terraform/helper/resource"
-	"github.com/hashicorp/terraform/terraform"
 )
 
 func TestAccDataSourcePagerDutyVendor_Basic(t *testing.T) {
+	dataSourceName := "data.pagerduty_vendor.foo"
 	resource.Test(t, resource.TestCase{
 		PreCheck:     func() { testAccPreCheck(t) },
 		Providers:    testAccProviders,
@@ -17,37 +16,30 @@ func TestAccDataSourcePagerDutyVendor_Basic(t *testing.T) {
 			{
 				Config: testAccDataSourcePagerDutyVendorConfig,
 				Check: resource.ComposeTestCheckFunc(
-					testAccDataSourcePagerDutyVendor("data.pagerduty_vendor.foo"),
+					resource.TestCheckResourceAttr(dataSourceName, "id", "PZQ6AUS"),
+					resource.TestCheckResourceAttr(dataSourceName, "name", "Amazon Cloudwatch"),
 				),
 			},
 		},
 	})
 }
 
-func testAccDataSourcePagerDutyVendor(n string) resource.TestCheckFunc {
-	return func(s *terraform.State) error {
-
-		r := s.RootModule().Resources[n]
-		a := r.Primary.Attributes
-
-		if a["id"] == "" {
-			return fmt.Errorf("Expected to get a vendor ID from PagerDuty")
-		}
-
-		if a["id"] != "PZQ6AUS" {
-			return fmt.Errorf("Expected the Datadog Vendor ID to be: PZQ6AUS, but got: %s", a["id"])
-		}
-
-		if a["name"] != "Amazon Cloudwatch" {
-			return fmt.Errorf("Expected the Datadog Vendor Name to be: Datadog, but got: %s", a["name"])
-		}
-
-		if a["type"] != "api" {
-			return fmt.Errorf("Expected the Datadog Vendor Type to be: api, but got: %s", a["type"])
-		}
-
-		return nil
-	}
+func TestAccDataSourcePagerDutyVendor_ExactMatch(t *testing.T) {
+	dataSourceName := "data.pagerduty_vendor.foo"
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckPagerDutyScheduleDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourcePagerDutyExactMatchConfig,
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(dataSourceName, "id", "PKG4M95"),
+					resource.TestCheckResourceAttr(dataSourceName, "name", "Sentry"),
+				),
+			},
+		},
+	})
 }
 
 const testAccDataSourcePagerDutyVendorConfig = `
@@ -55,3 +47,9 @@ data "pagerduty_vendor" "foo" {
   name = "cloudwatch"
 }
 `
+
+const testAccDataSourcePagerDutyExactMatchConfig = `
+data "pagerduty_vendor" "foo" {
+  name = "sentry"
+}
+`