-
Notifications
You must be signed in to change notification settings - Fork 54
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- added `--hetzner-user-data-file` and deprecated `--hetzner-user-data-from-file` in favor of it (#99, thanks @perlun)
- Loading branch information
1 parent
1b3281e
commit 13bf4f8
Showing
5 changed files
with
205 additions
and
19 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,122 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "github.com/docker/machine/commands/commandstest" | ||
| "github.com/docker/machine/libmachine/drivers" | ||
| "os" | ||
| "strings" | ||
| "testing" | ||
| ) | ||
|
|
||
| var defaultFlags = map[string]interface{}{ | ||
| flagAPIToken: "foo", | ||
| } | ||
|
|
||
| func makeFlags(args map[string]interface{}) drivers.DriverOptions { | ||
| combined := make(map[string]interface{}, len(defaultFlags)+len(args)) | ||
| for k, v := range defaultFlags { | ||
| combined[k] = v | ||
| } | ||
| for k, v := range args { | ||
| combined[k] = v | ||
| } | ||
|
|
||
| return &commandstest.FakeFlagger{Data: combined} | ||
| } | ||
|
|
||
| func TestUserData(t *testing.T) { | ||
| const fileContents = "User data from file" | ||
| const inlineContents = "User data" | ||
|
|
||
| file := t.TempDir() + string(os.PathSeparator) + "userData" | ||
| err := os.WriteFile(file, []byte(fileContents), 0644) | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
|
|
||
| // mutual exclusion data <=> data file | ||
| d := NewDriver() | ||
| err = d.setConfigFromFlagsImpl(makeFlags(map[string]interface{}{ | ||
| flagUserData: inlineContents, | ||
| flagUserDataFile: file, | ||
| })) | ||
| assertMutualExclusion(t, err, flagUserData, flagUserDataFile) | ||
|
|
||
| // mutual exclusion data file <=> legacy flag | ||
| d = NewDriver() | ||
| err = d.setConfigFromFlagsImpl(&commandstest.FakeFlagger{ | ||
| Data: map[string]interface{}{ | ||
| flagAPIToken: "foo", | ||
| legacyFlagUserDataFromFile: true, | ||
| flagUserDataFile: file, | ||
| }, | ||
| }) | ||
| assertMutualExclusion(t, err, legacyFlagUserDataFromFile, flagUserDataFile) | ||
|
|
||
| // inline user data | ||
| d = NewDriver() | ||
| err = d.setConfigFromFlagsImpl(makeFlags(map[string]interface{}{ | ||
| flagAPIToken: "foo", | ||
| flagUserData: inlineContents, | ||
| })) | ||
| if err != nil { | ||
| t.Fatalf("unexpected error, %v", err) | ||
| } | ||
|
|
||
| data, err := d.getUserData() | ||
| if err != nil { | ||
| t.Fatalf("unexpected error, %v", err) | ||
| } | ||
| if data != inlineContents { | ||
| t.Error("content did not match (inline)") | ||
| } | ||
|
|
||
| // file user data | ||
| d = NewDriver() | ||
| err = d.setConfigFromFlagsImpl(makeFlags(map[string]interface{}{ | ||
| flagAPIToken: "foo", | ||
| flagUserDataFile: file, | ||
| })) | ||
| if err != nil { | ||
| t.Fatalf("unexpected error, %v", err) | ||
| } | ||
|
|
||
| data, err = d.getUserData() | ||
| if err != nil { | ||
| t.Fatalf("unexpected error, %v", err) | ||
| } | ||
| if data != fileContents { | ||
| t.Error("content did not match (file)") | ||
| } | ||
|
|
||
| // legacy file user data | ||
| d = NewDriver() | ||
| err = d.setConfigFromFlagsImpl(makeFlags(map[string]interface{}{ | ||
| flagAPIToken: "foo", | ||
| flagUserData: file, | ||
| legacyFlagUserDataFromFile: true, | ||
| })) | ||
| if err != nil { | ||
| t.Fatalf("unexpected error, %v", err) | ||
| } | ||
|
|
||
| data, err = d.getUserData() | ||
| if err != nil { | ||
| t.Fatalf("unexpected error, %v", err) | ||
| } | ||
| if data != fileContents { | ||
| t.Error("content did not match (legacy-file)") | ||
| } | ||
| } | ||
|
|
||
| func assertMutualExclusion(t *testing.T, err error, flag1, flag2 string) { | ||
| if err == nil { | ||
| t.Errorf("expected mutually exclusive flags to fail, but no error was thrown: %v %v", flag1, flag2) | ||
| return | ||
| } | ||
|
|
||
| errstr := err.Error() | ||
| if !(strings.Contains(errstr, flag1) && strings.Contains(errstr, flag2) && strings.Contains(errstr, "mutually exclusive")) { | ||
| t.Errorf("expected mutually exclusive flags to fail, but message differs: %v %v %v", flag1, flag2, errstr) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters