test: add loader_test

pkg/executor/executor.go

@@ -31,8 +31,8 @@ type Executor struct {
 func NewExecutor(dpl *deployment.Deployment, cfg *Config, logChannel chan string) (*Executor, error) {
 	if cfg == nil {
 		cfg = &Config{
-			JobTimeout:        3600,
-			DeploymentTimeout: 86400,
+			JobTimeout:        3600,  // 1 hour
+			DeploymentTimeout: 86400, // 1 day
 			AllowFailure:      false,
 		}
 	}
@@ -43,7 +43,6 @@ func NewExecutor(dpl *deployment.Deployment, cfg *Config, logChannel chan string
 	}
 
 	filePath := fmt.Sprintf("logs/deployments/%s.log", randId.String())
-
 	file, err := os.OpenFile(filePath, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0660)
 	if err != nil {
 		return nil, err
@@ -85,7 +84,7 @@ func (e *Executor) ExecuteJobs() error {
 
 		for _, job := range e.Deployment.Jobs {
 			select {
-			// Prevent the deployment gorouting from running other jobs if deployment is cancelled
+			// Prevent the deployment goroutine from running other jobs if deployment is cancelled
 			case <-deploymentContext.Done():
 				return
 			default:
@@ -98,15 +97,10 @@ func (e *Executor) ExecuteJobs() error {
 				if host == "localhost" {
 					client = remote.ConnectToLocalhost()
 				} else {
-					hostStr, ok := host.(string)
-					if !ok {
-						errorChannel <- fmt.Errorf("error: host '%s' is not a string", host)
-						continue
-					}
-					c, ok := clients[hostStr]
+					c, ok := clients[host.(string)]
 					if !ok {
-						errorChannel <- fmt.Errorf("error: host '%s' does not exist", hostStr)
-						continue
+						errorChannel <- fmt.Errorf("error: host %#v does not exist", host)
+						return
 					}
 					client = c
 				}

pkg/loader/loader_test.go

@@ -0,0 +1,31 @@
+package loader_test
+
+import (
+	"testing"
+
+	loaders "github.com/SharkEzz/provisiond/internal/loader"
+	"github.com/SharkEzz/provisiond/pkg/loader"
+)
+
+func TestLoader(t *testing.T) {
+	ld := loader.GetLoader(`name: Test
+jobs:
+  - name: Test1
+  hosts: [localhost]
+  shell: echo oui > test.txt`)
+
+	if _, ok := ld.(loaders.StringLoader); !ok {
+		t.Errorf("expected StringLoader")
+	}
+
+	ld = loader.GetLoader("name: Test\n")
+
+	if _, ok := ld.(loaders.StringLoader); !ok {
+		t.Errorf("expected StringLoader")
+	}
+
+	ld = loader.GetLoader("./test.yaml")
+	if _, ok := ld.(loaders.FileLoader); !ok {
+		t.Errorf("expected FileLoader")
+	}
+}