From d40a96d350f479ee9a8375de777d1244cfcc6623 Mon Sep 17 00:00:00 2001
From: JordanBrockopp <Jordan_Brockopp@cargill.com>
Date: Thu, 29 Jun 2023 13:21:43 -0500
Subject: [PATCH] feat(database): add repos

---
 database/integration_test.go | 90 ++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)

diff --git a/database/integration_test.go b/database/integration_test.go
index 00c72c007..bbc8902b6 100644
--- a/database/integration_test.go
+++ b/database/integration_test.go
@@ -11,6 +11,8 @@ import (
 	"testing"
 	"time"
 
+	"github.com/go-vela/server/database/repo"
+
 	"github.com/go-vela/server/database/pipeline"
 
 	"github.com/go-vela/server/database/log"
@@ -288,6 +290,10 @@ func TestDatabase_Integration(t *testing.T) {
 				testPipelines(t, db, []*library.Repo{repoOne, repoTwo})
 			})
 
+			t.Run("test_repos", func(t *testing.T) {
+				testRepos(t, db, []*library.Repo{repoOne, repoTwo})
+			})
+
 			t.Run("test_services", func(t *testing.T) {
 				testServices(t, db, []*library.Build{buildOne, buildTwo}, []*library.Service{serviceOne, serviceTwo})
 			})
@@ -716,6 +722,90 @@ func testPipelines(t *testing.T, db Interface, repos []*library.Repo) {
 	}
 }
 
+func testRepos(t *testing.T, db Interface, repos []*library.Repo) {
+	// used to track the number of methods we call for repos
+	//
+	// we start at 2 for creating the table and indexes for repos
+	// since those are already called when the database engine starts
+	counter := 2
+
+	// create the repos
+	for _, repo := range repos {
+		err := db.CreateRepo(repo)
+		if err != nil {
+			t.Errorf("unable to create repo %s: %v", repo.GetFullName(), err)
+		}
+	}
+	counter++
+
+	// count the repos
+	count, err := db.CountRepos()
+	if err != nil {
+		t.Errorf("unable to count repos: %v", err)
+	}
+	if int(count) != len(repos) {
+		t.Errorf("CountRepos() is %v, want %v", count, len(repos))
+	}
+	counter++
+
+	// list the repos
+	list, err := db.ListRepos()
+	if err != nil {
+		t.Errorf("unable to list repos: %v", err)
+	}
+	if !reflect.DeepEqual(list, repos) {
+		t.Errorf("ListRepos() is %v, want %v", list, repos)
+	}
+	counter++
+
+	// lookup the repos by name
+	for _, repo := range repos {
+		got, err := db.GetRepoForOrg(repo.GetOrg(), repo.GetName())
+		if err != nil {
+			t.Errorf("unable to get repo %s by org: %v", repo.GetFullName(), err)
+		}
+		if !reflect.DeepEqual(got, repo) {
+			t.Errorf("GetRepoForOrg() is %v, want %v", got, repo)
+		}
+	}
+	counter++
+
+	// update the repos
+	for _, repo := range repos {
+		repo.SetActive(false)
+		err = db.UpdateRepo(repo)
+		if err != nil {
+			t.Errorf("unable to update repo %s: %v", repo.GetFullName(), err)
+		}
+
+		// lookup the repo by ID
+		got, err := db.GetRepo(repo.GetID())
+		if err != nil {
+			t.Errorf("unable to get repo %s by ID: %v", repo.GetFullName(), err)
+		}
+		if !reflect.DeepEqual(got, repo) {
+			t.Errorf("GetRepo() is %v, want %v", got, repo)
+		}
+	}
+	counter++
+	counter++
+
+	// delete the repos
+	for _, repo := range repos {
+		err = db.DeleteRepo(repo)
+		if err != nil {
+			t.Errorf("unable to delete repo %s: %v", repo.GetFullName(), err)
+		}
+	}
+	counter++
+
+	// ensure we called all the functions we should have
+	methods := reflect.TypeOf(new(repo.RepoInterface)).Elem().NumMethod()
+	if counter != methods {
+		t.Errorf("total number of methods called is %v, want %v", counter, methods)
+	}
+}
+
 func testServices(t *testing.T, db Interface, builds []*library.Build, services []*library.Service) {
 	// used to track the number of methods we call for services
 	//