workflow/client-go/controller/client/client_test.go at ad932b3e20cab08dbab59f84ae9e8be23b693c1e · drycc/workflow · 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
package client

import (
	"io/ioutil"
	"net/url"
	"os"
	"path"
	"testing"
)

const sFile string = `{"username":"t","ssl_verify":false,"controller":"http://d.t","token":"a"}`

func createTempProfile(contents string) error {
	name, err := ioutil.TempDir("", "client")

	if err != nil {
		return err
	}

	os.Unsetenv("DEIS_PROFILE")
	os.Setenv("HOME", name)
	folder := path.Join(name, "/.deis/")
	if err = os.Mkdir(folder, 0755); err != nil {
		return err
	}

	if err = ioutil.WriteFile(path.Join(folder, "client.json"), []byte(contents), 0775); err != nil {
		return err
	}

	return nil
}

func TestLoadSave(t *testing.T) {
	if err := createTempProfile(sFile); err != nil {
		t.Fatal(err)
	}

	client, err := New()

	if err != nil {
		t.Fatal(err)
	}

	expectedB := false
	if client.SSLVerify != expectedB {
		t.Errorf("Expected %t, Got %t", expectedB, client.SSLVerify)
	}

	expected := "a"
	if client.Token != expected {
		t.Errorf("Expected %s, Got %s", expected, client.Token)
	}

	expected = "t"
	if client.Username != expected {
		t.Errorf("Expected %s, Got %s", expected, client.Username)
	}

	expected = "http://d.t"
	if client.ControllerURL.String() != expected {
		t.Errorf("Expected %s, Got %s", expected, client.ControllerURL.String())
	}

	client.SSLVerify = true
	client.Token = "b"
	client.Username = "c"

	u, err := url.Parse("http://deis.test")

	if err != nil {
		t.Fatal(err)
	}

	client.ControllerURL = *u

	if err = client.Save(); err != nil {
		t.Fatal(err)
	}

	client, err = New()

	expectedB = true
	if client.SSLVerify != expectedB {
		t.Errorf("Expected %t, Got %t", expectedB, client.SSLVerify)
	}

	expected = "b"
	if client.Token != expected {
		t.Errorf("Expected %s, Got %s", expected, client.Token)
	}

	expected = "c"
	if client.Username != expected {
		t.Errorf("Expected %s, Got %s", expected, client.Username)
	}

	expected = "http://deis.test"
	if client.ControllerURL.String() != expected {
		t.Errorf("Expected %s, Got %s", expected, client.ControllerURL.String())
	}
}