Improve tests for the --dir option

fastly · Jan 17, 2025 · e16105d · e16105d
1 parent 228f618
commit e16105d
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/pkg/commands/kvstoreentry/kvstoreentry_test.go b/pkg/commands/kvstoreentry/kvstoreentry_test.go
@@ -77,8 +77,11 @@ func TestCreateCommand(t *testing.T) {
 			Args:  fmt.Sprintf("--store-id %s --dir %s", storeID, filepath.Join("testdata", "example")),
 			Stdin: []string{"y"},
 			API: mock.API{
-				InsertKVStoreKeyFn: func(_ *fastly.InsertKVStoreKeyInput) error {
-					return nil
+				InsertKVStoreKeyFn: func(i *fastly.InsertKVStoreKeyInput) error {
+					if i.Key == "foo.txt" {
+						return nil
+					}
+					return errors.New("invalid request")
 				},
 			},
 			WantOutput: "SUCCESS: Inserted 1 keys into KV Store",
@@ -87,8 +90,11 @@ func TestCreateCommand(t *testing.T) {
 			Args:  fmt.Sprintf("--store-id %s --dir %s --dir-allow-hidden", storeID, filepath.Join("testdata", "example")),
 			Stdin: []string{"y"},
 			API: mock.API{
-				InsertKVStoreKeyFn: func(_ *fastly.InsertKVStoreKeyInput) error {
-					return nil
+				InsertKVStoreKeyFn: func(i *fastly.InsertKVStoreKeyInput) error {
+					if i.Key == "foo.txt" || i.Key == ".hiddenfile" {
+						return nil
+					}
+					return errors.New("invalid request")
 				},
 			},
 			WantOutput: "SUCCESS: Inserted 2 keys into KV Store",