Project data source and test

shelley.bess · cehoffman · commit 25ebbc344b4d · 2018-01-23T15:27:28.000-06:00
Signed-off-by: shelley.bess &lt;shelley.bess@vertivco.com&gt;
diff --git a/gitlab/data_source_gitlab_project.go b/gitlab/data_source_gitlab_project.go
@@ -0,0 +1,69 @@
+package gitlab
+
+import (
+	"fmt"
+	"log"
+	"strings"
+
+	"github.com/hashicorp/terraform/helper/schema"
+	"github.com/xanzy/go-gitlab"
+)
+
+func dataSourceGitlabProject() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceGitlabProjectRead,
+
+		Schema: map[string]*schema.Schema{
+			"name": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+		},
+	}
+}
+
+// Performs the lookup
+func dataSourceGitlabProjectRead(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*gitlab.Client)
+
+	log.Printf("[INFO] Reading Gitlab project")
+
+	searchName := strings.ToLower(d.Get("name").(string))
+
+	o := &gitlab.ListProjectsOptions{
+		Search: &searchName,
+	}
+
+	projects, _, err := client.Projects.ListProjects(o)
+	if err != nil {
+		return err
+	}
+
+	var found *gitlab.Project
+
+	for _, project := range projects {
+		if strings.ToLower(project.Name) == searchName {
+			found = project
+			break
+		}
+	}
+
+	if found == nil {
+		return fmt.Errorf("Unable to locate any project with the name: %s", searchName)
+	}
+	d.SetId(fmt.Sprintf("%d", found.ID))
+	d.Set("name", found.Name)
+	d.Set("path", found.Path)
+	d.Set("description", found.Description)
+	d.Set("default_branch", found.DefaultBranch)
+	d.Set("issues_enabled", found.IssuesEnabled)
+	d.Set("merge_requests_enabled", found.MergeRequestsEnabled)
+	d.Set("wiki_enabled", found.WikiEnabled)
+	d.Set("snippets_enabled", found.SnippetsEnabled)
+	d.Set("visibility_level", string(found.Visibility))
+	d.Set("namespace_id", found.Namespace.ID)
+	d.Set("ssh_url_to_repo", found.SSHURLToRepo)
+	d.Set("http_url_to_repo", found.HTTPURLToRepo)
+	d.Set("web_url", found.WebURL)
+	return nil
+}
diff --git a/gitlab/data_source_gitlab_project_test.go b/gitlab/data_source_gitlab_project_test.go
@@ -0,0 +1,66 @@
+package gitlab
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+	"github.com/hashicorp/terraform/terraform"
+)
+
+func TestAccDataGitlabProject_basic(t *testing.T) {
+	projectname := fmt.Sprintf("tf-%s", acctest.RandString(5))
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataGitlabProjectConfig(projectname),
+				Check: resource.ComposeTestCheckFunc(
+					testAccDataSourceGitlabProject("gitlab_project.test", "data.gitlab_project.foo"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceGitlabProject(src, n string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+
+		project := s.RootModule().Resources[src]
+		projectResource := project.Primary.Attributes
+
+		search := s.RootModule().Resources[n]
+		searchResource := search.Primary.Attributes
+
+		if searchResource["id"] == "" {
+			return fmt.Errorf("Expected to get a project ID from Gitlab")
+		}
+
+		testAttributes := []string{"id", "name", "visibility"}
+
+		for _, attribute := range testAttributes {
+			if searchResource[attribute] != projectResource[attribute] {
+				return fmt.Errorf("Expected the project %s to be: %s, but got: %s", attribute, projectResource[attribute], searchResource[attribute])
+			}
+		}
+		return nil
+	}
+}
+
+func testAccDataGitlabProjectConfig(projectname string) string {
+	return fmt.Sprintf(`
+resource "gitlab_project" "test"{
+	name = "%s"
+	path = "%s"
+	description = "Terraform acceptance tests"
+	visibility_level = "public"
+}
+
+data "gitlab_project" "foo" {
+	name = "${gitlab_project.test.name}"
+}
+	`, projectname, projectname)
+}
diff --git a/gitlab/data_source_gitlab_user.go b/gitlab/data_source_gitlab_user.go
@@ -0,0 +1,56 @@
+package gitlab
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"github.com/hashicorp/terraform/helper/schema"
+	gitlab "github.com/xanzy/go-gitlab"
+)
+
+// Search by email required
+func dataSourceGitlabUser() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceGitlabUserRead,
+		Schema: map[string]*schema.Schema{
+			//Search option
+			"email": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+		},
+	}
+}
+
+// Performs the lookup
+func dataSourceGitlabUserRead(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*gitlab.Client)
+	// Create the query, grab the email for the query and set it for use
+	var query *gitlab.ListUsersOptions
+	email := strings.ToLower(d.Get("email").(string))
+	*query.Search = email
+	// Query to find the email. Returns a list
+	users, _, err := client.Users.ListUsers(query)
+	if err != nil {
+		return err
+	}
+
+	// Create a user to save userdata to
+	var user *gitlab.User
+	// Grab User data out of list
+	for _, a := range users {
+		if a.Email == email {
+			user = a
+			break
+		}
+	}
+	if user == nil {
+		return fmt.Errorf("The email '%s' does not match any user email", email)
+	}
+	d.SetId(strconv.Itoa(user.ID))
+	d.Set("name", user.Name)
+	d.Set("username", user.Username)
+	d.Set("email", user.Email)
+	return nil
+}
diff --git a/gitlab/data_source_gitlab_user_test.go b/gitlab/data_source_gitlab_user_test.go
@@ -0,0 +1,63 @@
+package gitlab
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+	"github.com/hashicorp/terraform/terraform"
+)
+
+func TestAccDataGitlabUser_basic(t *testing.T) {
+	userEmail := fmt.Sprintf("tf-%s", acctest.RandString(5))
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataGitlabUserConfig(userEmail),
+				Check: resource.ComposeTestCheckFunc(
+					testAccDataSourceGitlabUser("gitlab_user.foo", "data.gitlab_user.foo"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceGitlabUser(src, n string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+
+		srcR := s.RootModule().Resources[src]
+		srcA := srcR.Primary.Attributes
+
+		r := s.RootModule().Resources[n]
+		a := r.Primary.Attributes
+
+		if a["id"] == "" {
+			return fmt.Errorf("Expected to get a user email from Gitlab")
+		}
+
+		testAtts := []string{"id", "email"}
+
+		for _, att := range testAtts {
+			if a[att] != srcA[att] {
+				return fmt.Errorf("Expected the user %s to be: %s, but got: %s", att, srcA[att], a[att])
+			}
+		}
+		return nil
+	}
+}
+
+func testAccDataGitlabUserConfig(userEmail string) string {
+	return fmt.Sprintf(`
+resource "gitlab_user" "foo"{
+	email = "%s"
+}
+
+data "gitlab_user" "foo" {
+	name = "${gitlab_user.foo.email}"
+}
+	`, userEmail)
+}
diff --git a/gitlab/provider.go b/gitlab/provider.go
@@ -40,6 +40,12 @@ func Provider() terraform.ResourceProvider {
 				Description: descriptions["insecure"],
 			},
 		},
+
+		DataSourcesMap: map[string]*schema.Resource{
+			"gitlab_project": dataSourceGitlabProject(),
+			"gitlab_user":    dataSourceGitlabUser(),
+		},
+
 		ResourcesMap: map[string]*schema.Resource{
 			"gitlab_group":              resourceGitlabGroup(),
 			"gitlab_project":            resourceGitlabProject(),
