Added new test

Added data source instances size test
Fix documentation for template import cmd

Signed-off-by: Alejandro JNM <[email protected]>

civo/datasource_intances_size_test.go

@@ -0,0 +1,136 @@
+package civo
+
+import (
+	"fmt"
+	"strconv"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/terraform"
+)
+
+func TestAccDataSourceCivoInstanceSize_basic(t *testing.T) {
+	datasourceName := "data.civo_instances_size.foobar"
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceCivoInstanceSizeConfig(),
+				Check: resource.ComposeAggregateTestCheckFunc(
+					testAccCheckDataSourceCivoInstanceSizeExist(datasourceName),
+				),
+			},
+		},
+	})
+}
+
+func TestAccDataSourceCivoInstanceSize_WithFilterAndSort(t *testing.T) {
+	datasourceName := "data.civo_instances_size.foobar"
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceCivoInstanceSizeConfigWhitFilterAndSort(),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckDataSourceCivoInstanceSizeExist(datasourceName),
+					testAccCheckDataSourceCivoInstanceSizeFilteredAndSorted(datasourceName),
+				),
+			},
+		},
+	})
+}
+
+func testAccCheckDataSourceCivoInstanceSizeExist(n string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		rs, ok := s.RootModule().Resources[n]
+
+		if !ok {
+			return fmt.Errorf("Not found: %s", n)
+		}
+
+		if rs.Primary.ID == "" {
+			return fmt.Errorf("No Record ID is set")
+		}
+
+		rawTotal := rs.Primary.Attributes["sizes.#"]
+		total, err := strconv.Atoi(rawTotal)
+		if err != nil {
+			return err
+		}
+
+		if total < 1 {
+			return fmt.Errorf("No Civo sizes retrieved")
+		}
+
+		return nil
+	}
+}
+
+func testAccCheckDataSourceCivoInstanceSizeFilteredAndSorted(n string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		rs, ok := s.RootModule().Resources[n]
+
+		if !ok {
+			return fmt.Errorf("Not found: %s", n)
+		}
+
+		rawTotal := rs.Primary.Attributes["sizes.#"]
+		total, err := strconv.Atoi(rawTotal)
+		if err != nil {
+			return err
+		}
+
+		stringInSlice := func(value string, slice []string) bool {
+			for _, item := range slice {
+				if item == value {
+					return true
+				}
+			}
+			return false
+		}
+
+		var prevCPUCore float64
+		for i := 0; i < total; i++ {
+			name := rs.Primary.Attributes[fmt.Sprintf("sizes.%d.name", i)]
+			if !stringInSlice(name, []string{"g2.large", "g2.xlarge", "g2.2xlarge"}) {
+				return fmt.Errorf("Name is not in expected test filter values")
+			}
+
+			CPUCore, _ := strconv.ParseFloat(rs.Primary.Attributes[fmt.Sprintf("sizes.%d.cpu_cores", i)], 64)
+			if prevCPUCore > 0 && prevCPUCore < prevCPUCore {
+				return fmt.Errorf("Sizes is not sorted by CPU Core in descending order")
+			}
+			prevCPUCore = CPUCore
+
+		}
+
+		return nil
+	}
+}
+
+func testAccDataSourceCivoInstanceSizeConfig() string {
+	return fmt.Sprintf(`
+data "civo_instances_size" "foobar" {
+}
+`)
+}
+
+func testAccDataSourceCivoInstanceSizeConfigWhitFilterAndSort() string {
+	return fmt.Sprintf(`
+data "civo_instances_size" "foobar" {
+	filter {
+        key = "name"
+        values = ["large"]
+	}
+	
+	sort {
+        key = "cpu_cores"
+        direction = "desc"
+    }
+}
+`)
+}

website/docs/r/template.html.markdown

@@ -64,8 +64,8 @@ the instance is first booted.
 
 ## Import
 
-Template can be imported using the template `id`, e.g.
+Template can be imported using the template `code`, e.g.
 
 ```
-terraform import civo_template.my-custom-template aa589909-6bcc-42cc-8be9-b9c3648f74c1 
+terraform import civo_template.my-custom-template my-template-code 
 ```