-
Notifications
You must be signed in to change notification settings - Fork 808
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: torredil <[email protected]>
- Loading branch information
Showing
5 changed files
with
221 additions
and
32 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
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,143 @@ | ||
| package driver | ||
|
|
||
| import ( | ||
| "testing" | ||
| "time" | ||
|
|
||
| flag "github.com/spf13/pflag" | ||
| ) | ||
|
|
||
| func TestAddFlags(t *testing.T) { | ||
| o := &Options{} | ||
|
|
||
| f := flag.NewFlagSet("test", flag.ExitOnError) | ||
| o.AddFlags(f) | ||
|
|
||
| if err := f.Set("endpoint", "custom-endpoint"); err != nil { | ||
| t.Errorf("error setting endpoint: %v", err) | ||
| } | ||
| if err := f.Set("http-endpoint", ":8080"); err != nil { | ||
| t.Errorf("error setting http-endpoint: %v", err) | ||
| } | ||
| if err := f.Set("enable-otel-tracing", "true"); err != nil { | ||
| t.Errorf("error setting enable-otel-tracing: %v", err) | ||
| } | ||
| if err := f.Set("extra-tags", "key1=value1,key2=value2"); err != nil { | ||
| t.Errorf("error setting extra-tags: %v", err) | ||
| } | ||
| if err := f.Set("k8s-tag-cluster-id", "cluster-123"); err != nil { | ||
| t.Errorf("error setting k8s-tag-cluster-id: %v", err) | ||
| } | ||
| if err := f.Set("aws-sdk-debug-log", "true"); err != nil { | ||
| t.Errorf("error setting aws-sdk-debug-log: %v", err) | ||
| } | ||
| if err := f.Set("warn-on-invalid-tag", "true"); err != nil { | ||
| t.Errorf("error setting warn-on-invalid-tag: %v", err) | ||
| } | ||
| if err := f.Set("user-agent-extra", "extra-info"); err != nil { | ||
| t.Errorf("error setting user-agent-extra: %v", err) | ||
| } | ||
| if err := f.Set("batching", "true"); err != nil { | ||
| t.Errorf("error setting batching: %v", err) | ||
| } | ||
| if err := f.Set("modify-volume-request-handler-timeout", "1m"); err != nil { | ||
| t.Errorf("error setting modify-volume-request-handler-timeout: %v", err) | ||
| } | ||
| if err := f.Set("volume-attach-limit", "10"); err != nil { | ||
| t.Errorf("error setting volume-attach-limit: %v", err) | ||
| } | ||
| if err := f.Set("reserved-volume-attachments", "5"); err != nil { | ||
| t.Errorf("error setting reserved-volume-attachments: %v", err) | ||
| } | ||
|
|
||
| if o.Endpoint != "custom-endpoint" { | ||
| t.Errorf("unexpected Endpoint: got %s, want custom-endpoint", o.Endpoint) | ||
| } | ||
| if o.HttpEndpoint != ":8080" { | ||
| t.Errorf("unexpected HttpEndpoint: got %s, want :8080", o.HttpEndpoint) | ||
| } | ||
| if !o.EnableOtelTracing { | ||
| t.Error("unexpected EnableOtelTracing: got false, want true") | ||
| } | ||
| if len(o.ExtraTags) != 2 || o.ExtraTags["key1"] != "value1" || o.ExtraTags["key2"] != "value2" { | ||
| t.Errorf("unexpected ExtraTags: got %v, want map[key1:value1 key2:value2]", o.ExtraTags) | ||
| } | ||
| if o.KubernetesClusterID != "cluster-123" { | ||
| t.Errorf("unexpected KubernetesClusterID: got %s, want cluster-123", o.KubernetesClusterID) | ||
| } | ||
| if !o.AwsSdkDebugLog { | ||
| t.Error("unexpected AwsSdkDebugLog: got false, want true") | ||
| } | ||
| if !o.WarnOnInvalidTag { | ||
| t.Error("unexpected WarnOnInvalidTag: got false, want true") | ||
| } | ||
| if o.UserAgentExtra != "extra-info" { | ||
| t.Errorf("unexpected UserAgentExtra: got %s, want extra-info", o.UserAgentExtra) | ||
| } | ||
| if !o.Batching { | ||
| t.Error("unexpected Batching: got false, want true") | ||
| } | ||
| if o.ModifyVolumeRequestHandlerTimeout != time.Minute { | ||
| t.Errorf("unexpected ModifyVolumeRequestHandlerTimeout: got %v, want 1m", o.ModifyVolumeRequestHandlerTimeout) | ||
| } | ||
| if o.VolumeAttachLimit != 10 { | ||
| t.Errorf("unexpected VolumeAttachLimit: got %d, want 10", o.VolumeAttachLimit) | ||
| } | ||
| if o.ReservedVolumeAttachments != 5 { | ||
| t.Errorf("unexpected ReservedVolumeAttachments: got %d, want 5", o.ReservedVolumeAttachments) | ||
| } | ||
| } | ||
|
|
||
| func TestValidate(t *testing.T) { | ||
| tests := []struct { | ||
| name string | ||
| volumeAttachLimit int64 | ||
| reservedAttachments int | ||
| expectedErr bool | ||
| errMsg string | ||
| }{ | ||
| { | ||
| name: "both options not set", | ||
| volumeAttachLimit: -1, | ||
| reservedAttachments: -1, | ||
| expectedErr: false, | ||
| }, | ||
| { | ||
| name: "volumeAttachLimit set", | ||
| volumeAttachLimit: 10, | ||
| reservedAttachments: -1, | ||
| expectedErr: false, | ||
| }, | ||
| { | ||
| name: "reservedVolumeAttachments set", | ||
| volumeAttachLimit: -1, | ||
| reservedAttachments: 2, | ||
| expectedErr: false, | ||
| }, | ||
| { | ||
| name: "both options set", | ||
| volumeAttachLimit: 10, | ||
| reservedAttachments: 2, | ||
| expectedErr: true, | ||
| errMsg: "only one of --volume-attach-limit and --reserved-volume-attachments may be specified", | ||
| }, | ||
| } | ||
|
|
||
| for _, tt := range tests { | ||
| t.Run(tt.name, func(t *testing.T) { | ||
| o := &Options{ | ||
| VolumeAttachLimit: tt.volumeAttachLimit, | ||
| ReservedVolumeAttachments: tt.reservedAttachments, | ||
| } | ||
|
|
||
| err := o.Validate() | ||
| if (err != nil) != tt.expectedErr { | ||
| t.Errorf("Options.Validate() error = %v, wantErr %v", err, tt.expectedErr) | ||
| } | ||
|
|
||
| if err != nil && err.Error() != tt.errMsg { | ||
| t.Errorf("Options.Validate() error message = %v, wantErrMsg %v", err.Error(), tt.errMsg) | ||
| } | ||
| }) | ||
| } | ||
| } |