workflow-cli/internal/loader/loader_test.go at dbe9fa9a561a8e6eec81fee95c8ce0f0e637fba7 · drycc/workflow-cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
// Package loader provides common utility functions and helper methods for the workflow CLI.
package loader

import (
	"os"
	"path/filepath"
	"testing"

	drycc "github.com/drycc/controller-sdk-go"
	"github.com/drycc/workflow-cli/pkg/git"
	"github.com/drycc/workflow-cli/pkg/settings"
	"github.com/stretchr/testify/assert"
)

func TestLoadProject(t *testing.T) {
	name, err := os.MkdirTemp("", "drycc-cli-unit-test-load")
	assert.NoError(t, err)
	defer os.RemoveAll(name)

	filename := filepath.Join(name, "test.json")
	host := "drycc.example.com"
	client, err := drycc.New(false, host, "")
	assert.NoError(t, err)

	config := settings.Settings{
		Username:  "test",
		Client:    client,
		Workspace: "test-workspace",
	}

	filename, err = config.Save(filename)
	assert.NoError(t, err)

	appID, _, err := LoadAppSettings(filename, "test")
	assert.NoError(t, err)
	assert.Equal(t, appID, "test", "app")

	assert.NoError(t, os.Chdir(name))

	os.Setenv("DRYCC_APP", "testapp")
	appID, _, err = LoadAppSettings(filename, "")
	assert.NoError(t, err)
	assert.Equal(t, appID, "testapp", "app")
	os.Unsetenv("DRYCC_APP")

	appID, _, err = LoadAppSettings(filename, "")
	assert.NoError(t, err)
	assert.Equal(t, appID, filepath.Base(name), "app")

	assert.NoError(t, git.Init(git.DefaultCmd))
	assert.NoError(t, git.CreateRemote(git.DefaultCmd, host, "drycc", "testing"))

	appID, _, err = LoadAppSettings(filename, "")
	assert.NoError(t, err)
	assert.Equal(t, appID, "testing", "app")
}

func TestLoadAppSettingsNoWorkspace(t *testing.T) {
	name, err := os.MkdirTemp("", "drycc-cli-unit-test-no-workspace")
	assert.NoError(t, err)
	defer os.RemoveAll(name)

	filename := filepath.Join(name, "test.json")
	client, err := drycc.New(false, "drycc.example.com", "")
	assert.NoError(t, err)

	config := settings.Settings{
		Username: "test",
		Client:   client,
	}
	filename, err = config.Save(filename)
	assert.NoError(t, err)

	_, _, err = LoadAppSettings(filename, "test")
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "no workspace specified")
}

func TestLoadWorkspace(t *testing.T) {
	name, err := os.MkdirTemp("", "drycc-cli-unit-test-workspace")
	assert.NoError(t, err)
	defer os.RemoveAll(name)

	filename := filepath.Join(name, "test.json")
	client, err := drycc.New(false, "drycc.example.com", "")
	assert.NoError(t, err)

	// Test with no workspace set
	config := settings.Settings{
		Username: "test",
		Client:   client,
	}
	filename, err = config.Save(filename)
	assert.NoError(t, err)

	_, _, err = LoadWorkspace(filename)
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "no workspace specified")

	// Test with default workspace in config
	config.Workspace = "default-workspace"
	filename, err = config.Save(filename)
	assert.NoError(t, err)

	workspace, _, err := LoadWorkspace(filename)
	assert.NoError(t, err)
	assert.Equal(t, "default-workspace", workspace)
}