From 5389ed246bdd3e8302a9bf96988d7ec399aa9e13 Mon Sep 17 00:00:00 2001
From: ecrupper <easton.crupper12@gmail.com>
Date: Tue, 22 Aug 2023 10:27:43 -0500
Subject: [PATCH] add integration test for executables

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

diff --git a/database/integration_test.go b/database/integration_test.go
index cd4e3f9a7..79ed59033 100644
--- a/database/integration_test.go
+++ b/database/integration_test.go
@@ -13,6 +13,7 @@ import (
 	"time"
 
 	"github.com/go-vela/server/database/build"
+	"github.com/go-vela/server/database/executable"
 	"github.com/go-vela/server/database/hook"
 	"github.com/go-vela/server/database/log"
 	"github.com/go-vela/server/database/pipeline"
@@ -32,6 +33,7 @@ import (
 type Resources struct {
 	Builds      []*library.Build
 	Deployments []*library.Deployment
+	Executables []*library.BuildExecutable
 	Hooks       []*library.Hook
 	Logs        []*library.Log
 	Pipelines   []*library.Pipeline
@@ -117,6 +119,8 @@ func TestDatabase_Integration(t *testing.T) {
 
 			t.Run("test_builds", func(t *testing.T) { testBuilds(t, db, resources) })
 
+			t.Run("test_executables", func(t *testing.T) { testExecutables(t, db, resources) })
+
 			t.Run("test_hooks", func(t *testing.T) { testHooks(t, db, resources) })
 
 			t.Run("test_logs", func(t *testing.T) { testLogs(t, db, resources) })
@@ -381,6 +385,53 @@ func testBuilds(t *testing.T, db Interface, resources *Resources) {
 	}
 }
 
+func testExecutables(t *testing.T, db Interface, resources *Resources) {
+	// create a variable to track the number of methods called for pipelines
+	methods := make(map[string]bool)
+	// capture the element type of the pipeline interface
+	element := reflect.TypeOf(new(executable.BuildExecutableInterface)).Elem()
+	// iterate through all methods found in the pipeline interface
+	for i := 0; i < element.NumMethod(); i++ {
+		// skip tracking the methods to create indexes and tables for pipelines
+		// since those are already called when the database engine starts
+		if strings.Contains(element.Method(i).Name, "Index") ||
+			strings.Contains(element.Method(i).Name, "Table") {
+			continue
+		}
+
+		// add the method name to the list of functions
+		methods[element.Method(i).Name] = false
+	}
+
+	// create the pipelines
+	for _, executable := range resources.Executables {
+		err := db.CreateBuildExecutable(executable)
+		if err != nil {
+			t.Errorf("unable to create executable %d: %v", executable.GetID(), err)
+		}
+	}
+	methods["CreateExecutable"] = true
+
+	// pop executables for builds
+	for _, executable := range resources.Executables {
+		got, err := db.PopBuildExecutable(executable.GetBuildID())
+		if err != nil {
+			t.Errorf("unable to get executable %d for build %d: %v", executable.GetID(), executable.GetBuildID(), err)
+		}
+		if !reflect.DeepEqual(got, executable) {
+			t.Errorf("PopBuildExecutable() is %v, want %v", got, executable)
+		}
+	}
+	methods["PopBuildExecutable"] = true
+
+	// ensure we called all the methods we expected to
+	for method, called := range methods {
+		if !called {
+			t.Errorf("method %s was not called for pipelines", method)
+		}
+	}
+}
+
 func testHooks(t *testing.T, db Interface, resources *Resources) {
 	// create a variable to track the number of methods called for hooks
 	methods := make(map[string]bool)
@@ -1862,6 +1913,16 @@ func newResources() *Resources {
 	buildTwo.SetRuntime("docker")
 	buildTwo.SetDistribution("linux")
 
+	executableOne := new(library.BuildExecutable)
+	executableOne.SetID(1)
+	executableOne.SetBuildID(1)
+	executableOne.SetData([]byte("version: 1"))
+
+	executableTwo := new(library.BuildExecutable)
+	executableTwo.SetID(2)
+	executableTwo.SetBuildID(2)
+	executableTwo.SetData([]byte("version: 2"))
+
 	deploymentOne := new(library.Deployment)
 	deploymentOne.SetID(1)
 	deploymentOne.SetRepoID(1)
@@ -2229,6 +2290,7 @@ func newResources() *Resources {
 	return &Resources{
 		Builds:      []*library.Build{buildOne, buildTwo},
 		Deployments: []*library.Deployment{deploymentOne, deploymentTwo},
+		Executables: []*library.BuildExecutable{executableOne, executableTwo},
 		Hooks:       []*library.Hook{hookOne, hookTwo},
 		Logs:        []*library.Log{logServiceOne, logServiceTwo, logStepOne, logStepTwo},
 		Pipelines:   []*library.Pipeline{pipelineOne, pipelineTwo},