provide default values for workflowconfiguration (#59)

Author: Spartakovic

pkg/digger/digger_config.go

@@ -25,6 +25,23 @@ type Project struct {
 	WorkflowConfiguration WorkflowConfiguration `yaml:"workflow_configuration"`
 }
 
+func (p *Project) UnmarshalYAML(unmarshal func(interface{}) error) error {
+	type rawProject Project
+	raw := rawProject{
+		WorkflowConfiguration: WorkflowConfiguration{
+			OnPullRequestPushed: []string{"digger plan"},
+			OnPullRequestClosed: []string{"digger unlock"},
+			OnCommitToDefault:   []string{"digger apply"},
+		},
+	}
+	if err := unmarshal(&raw); err != nil {
+		return err
+	}
+	*p = Project(raw)
+	return nil
+
+}
+
 func NewDiggerConfig(workingDir string) (*DiggerConfig, error) {
 	config := &DiggerConfig{}
 	var fileName string

pkg/digger/digger_config_test.go

@@ -1,6 +1,8 @@
 package digger
 
 import (
+	"log"
+	"os"
 	"testing"
 )
 import "github.com/stretchr/testify/assert"
@@ -11,3 +13,51 @@ func TestDiggerConfigFileDoesNotExist(t *testing.T) {
 	assert.Equal(t, dg.Projects[0].Name, "default", "expected default project to have name 'default'")
 	assert.Equal(t, dg.Projects[0].Dir, ".", "expected default project dir to be '.'")
 }
+
+func TestDefaultValuesForWorkflowConfiguration(t *testing.T) {
+	tempDir := CreateTempDir()
+	defer func(name string) {
+		err := os.RemoveAll(name)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}(tempDir)
+
+	f, err := os.Create(tempDir + "/digger.yml")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	defer func(f *os.File) {
+		err := f.Close()
+		if err != nil {
+
+		}
+	}(f)
+
+	digger_yml := `
+projects:
+- name: dev
+  branch: /main/
+  dir: .
+  workspace: default
+`
+	_, err2 := f.WriteString(digger_yml)
+	if err2 != nil {
+		log.Fatal(err2)
+	}
+
+	dg, err := NewDiggerConfig(tempDir)
+	assert.NoError(t, err, "expected error to be not nil")
+	assert.Equal(t, dg.Projects[0].WorkflowConfiguration.OnPullRequestPushed[0], "digger plan")
+	assert.Equal(t, dg.Projects[0].WorkflowConfiguration.OnPullRequestClosed[0], "digger unlock")
+	assert.Equal(t, dg.Projects[0].WorkflowConfiguration.OnCommitToDefault[0], "digger apply")
+}
+
+func CreateTempDir() string {
+	dir, err := os.MkdirTemp("", "tmp")
+	if err != nil {
+		log.Fatal(err)
+	}
+	return dir
+}