| 1 | 1 |
new file mode 100644 |
| ... | ... |
@@ -0,0 +1,55 @@ |
| 0 |
+package engine |
|
| 1 |
+ |
|
| 2 |
+import ( |
|
| 3 |
+ "testing" |
|
| 4 |
+) |
|
| 5 |
+ |
|
| 6 |
+func TestRegister(t *testing.T) {
|
|
| 7 |
+ if err := Register("dummy1", nil); err != nil {
|
|
| 8 |
+ t.Fatal(err) |
|
| 9 |
+ } |
|
| 10 |
+ |
|
| 11 |
+ if err := Register("dummy1", nil); err == nil {
|
|
| 12 |
+ t.Fatalf("Expecting error, got none")
|
|
| 13 |
+ } |
|
| 14 |
+ |
|
| 15 |
+ eng := newTestEngine(t) |
|
| 16 |
+ |
|
| 17 |
+ //Should fail because globan handlers are copied |
|
| 18 |
+ //at the engine creation |
|
| 19 |
+ if err := eng.Register("dummy1", nil); err == nil {
|
|
| 20 |
+ t.Fatalf("Expecting error, got none")
|
|
| 21 |
+ } |
|
| 22 |
+ |
|
| 23 |
+ if err := eng.Register("dummy2", nil); err != nil {
|
|
| 24 |
+ t.Fatal(err) |
|
| 25 |
+ } |
|
| 26 |
+ |
|
| 27 |
+ if err := eng.Register("dummy2", nil); err == nil {
|
|
| 28 |
+ t.Fatalf("Expecting error, got none")
|
|
| 29 |
+ } |
|
| 30 |
+} |
|
| 31 |
+ |
|
| 32 |
+func TestJob(t *testing.T) {
|
|
| 33 |
+ eng := newTestEngine(t) |
|
| 34 |
+ job1 := eng.Job("dummy1", "--level=awesome")
|
|
| 35 |
+ |
|
| 36 |
+ if job1.handler != nil {
|
|
| 37 |
+ t.Fatalf("job1.handler should be empty")
|
|
| 38 |
+ } |
|
| 39 |
+ |
|
| 40 |
+ h := func(j *Job) string {
|
|
| 41 |
+ return j.Name |
|
| 42 |
+ } |
|
| 43 |
+ |
|
| 44 |
+ eng.Register("dummy2", h)
|
|
| 45 |
+ job2 := eng.Job("dummy2", "--level=awesome")
|
|
| 46 |
+ |
|
| 47 |
+ if job2.handler == nil {
|
|
| 48 |
+ t.Fatalf("job2.handler shouldn't be nil")
|
|
| 49 |
+ } |
|
| 50 |
+ |
|
| 51 |
+ if job2.handler(job2) != job2.Name {
|
|
| 52 |
+ t.Fatalf("handler dummy2 was not found in job2")
|
|
| 53 |
+ } |
|
| 54 |
+} |
| ... | ... |
@@ -23,7 +23,101 @@ func TestSetenv(t *testing.T) {
|
| 23 | 23 |
if val := job.Getenv("foo"); val != "bar" {
|
| 24 | 24 |
t.Fatalf("Getenv returns incorrect value: %s", val)
|
| 25 | 25 |
} |
| 26 |
+ |
|
| 27 |
+ job.Setenv("bar", "")
|
|
| 28 |
+ if val := job.Getenv("bar"); val != "" {
|
|
| 29 |
+ t.Fatalf("Getenv returns incorrect value: %s", val)
|
|
| 30 |
+ } |
|
| 26 | 31 |
if val := job.Getenv("nonexistent"); val != "" {
|
| 27 | 32 |
t.Fatalf("Getenv returns incorrect value: %s", val)
|
| 28 | 33 |
} |
| 29 | 34 |
} |
| 35 |
+ |
|
| 36 |
+func TestSetenvBool(t *testing.T) {
|
|
| 37 |
+ job := mkJob(t, "dummy") |
|
| 38 |
+ job.SetenvBool("foo", true)
|
|
| 39 |
+ if val := job.GetenvBool("foo"); !val {
|
|
| 40 |
+ t.Fatalf("GetenvBool returns incorrect value: %b", val)
|
|
| 41 |
+ } |
|
| 42 |
+ |
|
| 43 |
+ job.SetenvBool("bar", false)
|
|
| 44 |
+ if val := job.GetenvBool("bar"); val {
|
|
| 45 |
+ t.Fatalf("GetenvBool returns incorrect value: %b", val)
|
|
| 46 |
+ } |
|
| 47 |
+ |
|
| 48 |
+ if val := job.GetenvBool("nonexistent"); val {
|
|
| 49 |
+ t.Fatalf("GetenvBool returns incorrect value: %b", val)
|
|
| 50 |
+ } |
|
| 51 |
+} |
|
| 52 |
+ |
|
| 53 |
+func TestSetenvInt(t *testing.T) {
|
|
| 54 |
+ job := mkJob(t, "dummy") |
|
| 55 |
+ |
|
| 56 |
+ job.SetenvInt("foo", -42)
|
|
| 57 |
+ if val := job.GetenvInt("foo"); val != -42 {
|
|
| 58 |
+ t.Fatalf("GetenvInt returns incorrect value: %d", val)
|
|
| 59 |
+ } |
|
| 60 |
+ |
|
| 61 |
+ job.SetenvInt("bar", 42)
|
|
| 62 |
+ if val := job.GetenvInt("bar"); val != 42 {
|
|
| 63 |
+ t.Fatalf("GetenvInt returns incorrect value: %d", val)
|
|
| 64 |
+ } |
|
| 65 |
+ if val := job.GetenvInt("nonexistent"); val != -1 {
|
|
| 66 |
+ t.Fatalf("GetenvInt returns incorrect value: %d", val)
|
|
| 67 |
+ } |
|
| 68 |
+} |
|
| 69 |
+ |
|
| 70 |
+func TestSetenvList(t *testing.T) {
|
|
| 71 |
+ job := mkJob(t, "dummy") |
|
| 72 |
+ |
|
| 73 |
+ job.SetenvList("foo", []string{"bar"})
|
|
| 74 |
+ if val := job.GetenvList("foo"); len(val) != 1 || val[0] != "bar" {
|
|
| 75 |
+ t.Fatalf("GetenvList returns incorrect value: %v", val)
|
|
| 76 |
+ } |
|
| 77 |
+ |
|
| 78 |
+ job.SetenvList("bar", nil)
|
|
| 79 |
+ if val := job.GetenvList("bar"); val != nil {
|
|
| 80 |
+ t.Fatalf("GetenvList returns incorrect value: %v", val)
|
|
| 81 |
+ } |
|
| 82 |
+ if val := job.GetenvList("nonexistent"); val != nil {
|
|
| 83 |
+ t.Fatalf("GetenvList returns incorrect value: %v", val)
|
|
| 84 |
+ } |
|
| 85 |
+} |
|
| 86 |
+ |
|
| 87 |
+func TestImportEnv(t *testing.T) {
|
|
| 88 |
+ type dummy struct {
|
|
| 89 |
+ DummyInt int |
|
| 90 |
+ DummyStringArray []string |
|
| 91 |
+ } |
|
| 92 |
+ |
|
| 93 |
+ job := mkJob(t, "dummy") |
|
| 94 |
+ if err := job.ImportEnv(&dummy{42, []string{"foo", "bar"}}); err != nil {
|
|
| 95 |
+ t.Fatal(err) |
|
| 96 |
+ } |
|
| 97 |
+ |
|
| 98 |
+ dmy := dummy{}
|
|
| 99 |
+ if err := job.ExportEnv(&dmy); err != nil {
|
|
| 100 |
+ t.Fatal(err) |
|
| 101 |
+ } |
|
| 102 |
+ |
|
| 103 |
+ if dmy.DummyInt != 42 {
|
|
| 104 |
+ t.Fatalf("Expected 42, got %d", dmy.DummyInt)
|
|
| 105 |
+ } |
|
| 106 |
+ |
|
| 107 |
+ if len(dmy.DummyStringArray) != 2 || dmy.DummyStringArray[0] != "foo" || dmy.DummyStringArray[1] != "bar" {
|
|
| 108 |
+ t.Fatalf("Expected {foo, bar}, got %v", dmy.DummyStringArray)
|
|
| 109 |
+ } |
|
| 110 |
+ |
|
| 111 |
+} |
|
| 112 |
+ |
|
| 113 |
+func TestEnviron(t *testing.T) {
|
|
| 114 |
+ job := mkJob(t, "dummy") |
|
| 115 |
+ job.Setenv("foo", "bar")
|
|
| 116 |
+ val, exists := job.Environ()["foo"] |
|
| 117 |
+ if !exists {
|
|
| 118 |
+ t.Fatalf("foo not found in the environ")
|
|
| 119 |
+ } |
|
| 120 |
+ if val != "bar" {
|
|
| 121 |
+ t.Fatalf("bar not found in the environ")
|
|
| 122 |
+ } |
|
| 123 |
+} |
| 30 | 124 |
new file mode 100644 |
| ... | ... |
@@ -0,0 +1,38 @@ |
| 0 |
+package engine |
|
| 1 |
+ |
|
| 2 |
+import ( |
|
| 3 |
+ "fmt" |
|
| 4 |
+ "github.com/dotcloud/docker/utils" |
|
| 5 |
+ "io/ioutil" |
|
| 6 |
+ "runtime" |
|
| 7 |
+ "strings" |
|
| 8 |
+ "testing" |
|
| 9 |
+) |
|
| 10 |
+ |
|
| 11 |
+var globalTestID string |
|
| 12 |
+ |
|
| 13 |
+func newTestEngine(t *testing.T) *Engine {
|
|
| 14 |
+ // Use the caller function name as a prefix. |
|
| 15 |
+ // This helps trace temp directories back to their test. |
|
| 16 |
+ pc, _, _, _ := runtime.Caller(1) |
|
| 17 |
+ callerLongName := runtime.FuncForPC(pc).Name() |
|
| 18 |
+ parts := strings.Split(callerLongName, ".") |
|
| 19 |
+ callerShortName := parts[len(parts)-1] |
|
| 20 |
+ if globalTestID == "" {
|
|
| 21 |
+ globalTestID = utils.RandomString()[:4] |
|
| 22 |
+ } |
|
| 23 |
+ prefix := fmt.Sprintf("docker-test%s-%s-", globalTestID, callerShortName)
|
|
| 24 |
+ root, err := ioutil.TempDir("", prefix)
|
|
| 25 |
+ if err != nil {
|
|
| 26 |
+ t.Fatal(err) |
|
| 27 |
+ } |
|
| 28 |
+ eng, err := New(root) |
|
| 29 |
+ if err != nil {
|
|
| 30 |
+ t.Fatal(err) |
|
| 31 |
+ } |
|
| 32 |
+ return eng |
|
| 33 |
+} |
|
| 34 |
+ |
|
| 35 |
+func mkJob(t *testing.T, name string, args ...string) *Job {
|
|
| 36 |
+ return newTestEngine(t).Job(name, args...) |
|
| 37 |
+} |
| ... | ... |
@@ -205,8 +205,12 @@ func (job *Job) SetenvInt(key string, value int64) {
|
| 205 | 205 |
job.Setenv(key, fmt.Sprintf("%d", value))
|
| 206 | 206 |
} |
| 207 | 207 |
|
| 208 |
+// Returns nil if key not found |
|
| 208 | 209 |
func (job *Job) GetenvList(key string) []string {
|
| 209 | 210 |
sval := job.Getenv(key) |
| 211 |
+ if sval == "" {
|
|
| 212 |
+ return nil |
|
| 213 |
+ } |
|
| 210 | 214 |
l := make([]string, 0, 1) |
| 211 | 215 |
if err := json.Unmarshal([]byte(sval), &l); err != nil {
|
| 212 | 216 |
l = append(l, sval) |
| ... | ... |
@@ -234,7 +238,7 @@ func (job *Job) Setenv(key, value string) {
|
| 234 | 234 |
// DecodeEnv decodes `src` as a json dictionary, and adds |
| 235 | 235 |
// each decoded key-value pair to the environment. |
| 236 | 236 |
// |
| 237 |
-// If `text` cannot be decoded as a json dictionary, an error |
|
| 237 |
+// If `src` cannot be decoded as a json dictionary, an error |
|
| 238 | 238 |
// is returned. |
| 239 | 239 |
func (job *Job) DecodeEnv(src io.Reader) error {
|
| 240 | 240 |
m := make(map[string]interface{})
|
| 241 | 241 |
deleted file mode 100644 |
| ... | ... |
@@ -1,42 +0,0 @@ |
| 1 |
-package engine |
|
| 2 |
- |
|
| 3 |
-import ( |
|
| 4 |
- "fmt" |
|
| 5 |
- "github.com/dotcloud/docker/utils" |
|
| 6 |
- "io/ioutil" |
|
| 7 |
- "runtime" |
|
| 8 |
- "strings" |
|
| 9 |
- "testing" |
|
| 10 |
-) |
|
| 11 |
- |
|
| 12 |
-var globalTestID string |
|
| 13 |
- |
|
| 14 |
-func init() {
|
|
| 15 |
- Register("dummy", func(job *Job) string { return "" })
|
|
| 16 |
-} |
|
| 17 |
- |
|
| 18 |
-func newTestEngine(t *testing.T) *Engine {
|
|
| 19 |
- // Use the caller function name as a prefix. |
|
| 20 |
- // This helps trace temp directories back to their test. |
|
| 21 |
- pc, _, _, _ := runtime.Caller(1) |
|
| 22 |
- callerLongName := runtime.FuncForPC(pc).Name() |
|
| 23 |
- parts := strings.Split(callerLongName, ".") |
|
| 24 |
- callerShortName := parts[len(parts)-1] |
|
| 25 |
- if globalTestID == "" {
|
|
| 26 |
- globalTestID = utils.RandomString()[:4] |
|
| 27 |
- } |
|
| 28 |
- prefix := fmt.Sprintf("docker-test%s-%s-", globalTestID, callerShortName)
|
|
| 29 |
- root, err := ioutil.TempDir("", prefix)
|
|
| 30 |
- if err != nil {
|
|
| 31 |
- t.Fatal(err) |
|
| 32 |
- } |
|
| 33 |
- eng, err := New(root) |
|
| 34 |
- if err != nil {
|
|
| 35 |
- t.Fatal(err) |
|
| 36 |
- } |
|
| 37 |
- return eng |
|
| 38 |
-} |
|
| 39 |
- |
|
| 40 |
-func mkJob(t *testing.T, name string, args ...string) *Job {
|
|
| 41 |
- return newTestEngine(t).Job(name, args...) |
|
| 42 |
-} |