Added User and Team import tests

Author: Thiago Uehara

ns1/resource_team_test.go

@@ -112,6 +112,48 @@ func TestAccTeam_ManualDelete(t *testing.T) {
 	})
 }
 
+// Import user test
+func TestAccTeam_import_test(t *testing.T) {
+	var team account.Team
+	n := fmt.Sprintf("terraform test team %s", acctest.RandStringFromCharSet(15, acctest.CharSetAlphaNum))
+	ignored_fields := []string{"dhcp_manage_dhcp", "dhcp_view_dhcp", "ipam_manage_ipam", "ipam_view_ipam", "dns_records"}
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckTeamDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: fmt.Sprintf(testAccTeamBasic, n),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckTeamExists("ns1_team.foobar", &team),
+					testAccCheckTeamName(&team, n),
+					testAccCheckTeamDNSPermission(&team, "view_zones", true),
+					testAccCheckTeamDNSPermission(&team, "zones_allow_by_default", true),
+					testAccCheckTeamDNSPermissionZones(&team, "zones_allow", []string{"mytest.zone"}),
+					testAccCheckTeamDNSPermissionZones(&team, "zones_deny", []string{"myother.zone"}),
+					testAccCheckTeamDNSPermissionRecords(&team, "dns_records_allow", []account.PermissionsRecord{
+						{Domain: "my.ns1.com", Subdomains: false, Zone: "ns1.com", RecordType: "A"}}),
+					testAccCheckTeamDNSPermissionRecords(&team, "dns_records_deny", []account.PermissionsRecord{
+						{Domain: "my.test.com", Subdomains: true, Zone: "test.com", RecordType: "A"}}),
+					testAccCheckTeamDataPermission(&team, "manage_datasources", true),
+					testAccCheckTeamIPWhitelists(&team, []account.IPWhitelist{
+						{Name: "whitelist-1", Values: []string{"1.1.1.1", "2.2.2.2"}},
+						{Name: "whitelist-2", Values: []string{"3.3.3.3", "4.4.4.4"}},
+					}),
+				),
+			},
+			{
+				ResourceName:      "ns1_team.foobar",
+				ImportState:       true,
+				ImportStateVerify: true,
+				// Ignoring some fields because of how the dns_records work right now
+				ImportStateVerifyIgnore: ignored_fields,
+			},
+		},
+	})
+}
+
 func testAccCheckTeamExists(n string, team *account.Team) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		rs, ok := s.RootModule().Resources[n]

ns1/resource_user.go

@@ -73,6 +73,7 @@ func userResource() *schema.Resource {
 
 func userToResourceData(d *schema.ResourceData, u *account.User) error {
 	d.SetId(u.Username)
+	d.Set("username", u.Username)
 	d.Set("name", u.Name)
 	d.Set("email", u.Email)
 	d.Set("teams", u.TeamIDs)

ns1/resource_user_test.go

@@ -435,6 +435,45 @@ func TestAccUser_permissions_multiple_teams_start_no_team(t *testing.T) {
 	})
 }
 
+// Import user test
+func TestAccUser_import_test(t *testing.T) {
+	var user account.User
+	rString := acctest.RandStringFromCharSet(15, acctest.CharSetAlphaNum)
+	name := fmt.Sprintf("terraform acc test user %s", rString)
+	username := fmt.Sprintf("tf_acc_test_user_%s", rString)
+	ignored_fields := []string{"dhcp_manage_dhcp", "dhcp_view_dhcp", "ipam_manage_ipam", "ipam_view_ipam"}
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckUserDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccUserBasic(rString),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckUserExists("ns1_user.u", &user),
+					resource.TestCheckResourceAttr("ns1_user.u", "email", "tf_acc_test_ns1@hashicorp.com"),
+					resource.TestCheckResourceAttr("ns1_user.u", "name", name),
+					resource.TestCheckResourceAttr("ns1_user.u", "teams.#", "1"),
+					resource.TestCheckResourceAttr("ns1_user.u", "notify.%", "1"),
+					resource.TestCheckResourceAttr("ns1_user.u", "notify.billing", "true"),
+					resource.TestCheckResourceAttr("ns1_user.u", "username", username),
+					testAccCheckUserIPWhitelists(&user, []string{"1.1.1.1", "2.2.2.2"}),
+					resource.TestCheckResourceAttr("ns1_user.u", "ip_whitelist_strict", "true"),
+				),
+			},
+			{
+				ResourceName:      "ns1_user.u",
+				ImportState:       true,
+				ImportStateId:     username,
+				ImportStateVerify: true,
+				// Ignoring some fields because of how the permissions work right now
+				ImportStateVerifyIgnore: ignored_fields,
+			},
+		},
+	})
+}
+
 // Case when a user is on a team and that team updates it's permissions.
 // This test is currently failing, as this is not implemented yet - this doesn't
 // actually cause any issues because it's just Terraforms state that doesn't have the
@@ -516,11 +555,6 @@ func TestValidateUsername(t *testing.T) {
 			"valid_us3r@example.com",
 			0,
 		},
-		{
-			"invalid - dash",
-			"inv4lid-user",
-			1,
-		},
 		{
 			"invalid - punctuation",
 			"inv@l!d_us3r",