Merge pull request #1 from circa10a/provider-test

Add tests

Author: circa10a

.github/workflows/test.yml

@@ -0,0 +1,27 @@
+on: [
+  push,
+  pull_request
+]
+name: Test
+jobs:
+  test:
+    strategy:
+      matrix:
+        go-version: [
+          1.15.x
+        ]
+        os: [
+          ubuntu-latest,
+        ]
+    runs-on: ${{ matrix.os }}
+    steps:
+    - name: Install Go
+      uses: actions/setup-go@v2
+      with:
+        go-version: ${{ matrix.go-version }}
+    - name: Checkout code
+      uses: actions/checkout@v2
+    - name: Test
+      env:
+        TF_ACC: 1
+      run: go test ./...

.gitignore

@@ -1,3 +1,4 @@
 .terraform/
 *tfstate*
-terraform-provider-mcbroken
+terraform-provider-mcbroken
+*.log

Makefile

@@ -15,6 +15,10 @@ build-linux:
 	test -d $(PLUGIN_DIR) || mkdir -p $(PLUGIN_DIR)
 	go build -o $(PLUGIN_DIR)/$(PROVIDER_FULL_NAME)
 
+test: export TF_ACC=true
+test:
+	go test -v ./...
+
 lint:
 	@if ! command -v golangci-lint 1>/dev/null; then\
 		echo "Need to install golangci-lint";\

README.md

@@ -11,6 +11,7 @@ Base the count of your infrastucture resources on the current number of broken m
   * [Usage](#usage)
   * [Development](#development)
     + [Linting](#linting)
+    + [Testing](#testing)
     + [Mac](#mac)
     + [Linux](#linux)
     + [Windows](#windows)
@@ -138,6 +139,12 @@ output "user_specified_city_not_found" {
 make lint
 ```
 
+### Testing
+
+```bash
+make test
+```
+
 ### Mac
 
 ```bash

mcbroken/data_source_cities_test.go

@@ -0,0 +1,48 @@
+package mcbroken
+
+import (
+	"regexp"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccMcbrokenCities(t *testing.T) {
+	brokenNumberRegex, _ := regexp.Compile(`\d{1,3}.\d{1,2}`)
+	cityRegex, _ := regexp.Compile(`\w`)
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:          func() { /* no precheck needed testAccPreCheck(t) */ },
+		ProviderFactories: testAccProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccCheckCities(),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrSet(
+						"data.mcbroken_cities.all", "broken"),
+					resource.TestMatchResourceAttr(
+						"data.mcbroken_cities.all", "broken", brokenNumberRegex),
+					resource.TestCheckResourceAttrSet(
+						"data.mcbroken_cities.all", "cities.0.city"),
+					resource.TestCheckResourceAttrSet(
+						"data.mcbroken_cities.all", "cities.0.broken"),
+					resource.TestMatchResourceAttr(
+						"data.mcbroken_cities.all", "cities.0.city", cityRegex),
+					resource.TestMatchResourceAttr(
+						"data.mcbroken_cities.all", "cities.0.broken", brokenNumberRegex),
+					resource.TestCheckResourceAttrSet(
+						"data.mcbroken_cities.all", "cities.1.city"),
+					resource.TestCheckResourceAttrSet(
+						"data.mcbroken_cities.all", "cities.1.broken"),
+					resource.TestMatchResourceAttr(
+						"data.mcbroken_cities.all", "cities.1.city", cityRegex),
+					resource.TestMatchResourceAttr(
+						"data.mcbroken_cities.all", "cities.1.broken", brokenNumberRegex),
+				),
+			},
+		},
+	})
+}
+
+func testAccCheckCities() string {
+	return `data "mcbroken_cities" "all" {}`
+}

mcbroken/data_source_city_test.go

@@ -0,0 +1,50 @@
+package mcbroken
+
+import (
+	"fmt"
+	"regexp"
+	"strings"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccMcbrokenCity(t *testing.T) {
+	brokenNumberRegex, _ := regexp.Compile(`\d{1,3}.\d{1,2}`)
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:          func() { /* no precheck needed testAccPreCheck(t) */ },
+		ProviderFactories: testAccProviderFactories,
+		Steps: []resource.TestStep{
+			// valid city
+			{
+				Config: testAccCheckCity("Dallas"),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(
+						"data.mcbroken_city.dallas", "city", "dallas"),
+					resource.TestCheckResourceAttrSet(
+						"data.mcbroken_city.dallas", "broken"),
+					resource.TestMatchResourceAttr(
+						"data.mcbroken_city.dallas", "broken", brokenNumberRegex),
+				),
+			},
+			// invalid city
+			{
+				Config: testAccCheckCity("not_found"),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(
+						"data.mcbroken_city.not_found", "city", "not_found"),
+					resource.TestCheckResourceAttr(
+						"data.mcbroken_city.not_found", "broken", "-1"),
+				),
+			},
+		},
+	})
+}
+
+func testAccCheckCity(city string) string {
+	return fmt.Sprintf(`
+data "mcbroken_city" "%[1]v" {
+  city = "%[1]v"
+}
+`, strings.ToLower(city))
+}

mcbroken/provider.go

@@ -2,20 +2,21 @@ package mcbroken
 
 import (
 	"context"
-	"net/url"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
 )
 
 // Provider -
 func Provider() *schema.Provider {
 	return &schema.Provider{
 		Schema: map[string]*schema.Schema{
 			"url": {
-				Type:     schema.TypeString,
-				Optional: true,
-				Default:  "https://mcbroken.com/stats.json",
+				Type:         schema.TypeString,
+				Optional:     true,
+				Default:      "https://mcbroken.com/stats.json",
+				ValidateFunc: validation.IsURLWithHTTPorHTTPS,
 			},
 		},
 		DataSourcesMap: map[string]*schema.Resource{
@@ -29,18 +30,6 @@ func Provider() *schema.Provider {
 func providerConfigure(ctx context.Context, d *schema.ResourceData) (interface{}, diag.Diagnostics) {
 	var diags diag.Diagnostics
 	providerConfig := make(map[string]interface{})
-	mcbrokenEndpoint := d.Get("url").(string)
-
-	_, err := url.Parse(mcbrokenEndpoint)
-	if err != nil {
-		diags = append(diags, diag.Diagnostic{
-			Severity: diag.Error,
-			Summary:  "Unable to parse url",
-			Detail:   "Mcbroken endpoint provided is not a valid url",
-		})
-		return nil, diags
-	}
-
-	providerConfig["url"] = mcbrokenEndpoint
+	providerConfig["url"] = d.Get("url").(string)
 	return providerConfig, diags
 }

mcbroken/provider_test.go

@@ -0,0 +1,33 @@
+package mcbroken
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+var testAccProviders map[string]*schema.Provider
+var testAccProvider *schema.Provider
+var testAccProviderFactories map[string]func() (*schema.Provider, error)
+
+func init() {
+	testAccProvider = Provider()
+	testAccProviders = map[string]*schema.Provider{
+		"mcbroken": testAccProvider,
+	}
+	testAccProviderFactories = map[string]func() (*schema.Provider, error){
+		"mcbroken": func() (*schema.Provider, error) {
+			return testAccProvider, nil
+		},
+	}
+}
+
+func TestProvider(t *testing.T) {
+	if err := testAccProvider.InternalValidate(); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+}
+
+func TestProvider_Impl(t *testing.T) {
+	var _ *schema.Provider = Provider()
+}