Add invalid variable set apply/remove test cases

hashicorp · Apr 13, 2022 · bd4f215 · bd4f215
1 parent 9b47554
commit bd4f215
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/variable_set_test.go b/variable_set_test.go
@@ -254,6 +254,41 @@ func TestVariableSetsApplyToAndRemoveFromWorkspaces(t *testing.T) {
 		assert.Equal(t, 1, len(vsAfter.Workspaces))
 		assert.Equal(t, wTest2.ID, vsAfter.Workspaces[0].ID)
 	})
+
+	t.Run("when variable set ID is invalid", func(t *testing.T) {
+		applyOptions := VariableSetApplyToWorkspacesOptions{
+			Workspaces: []*Workspace{wTest1},
+		}
+
+		err := client.VariableSets.ApplyToWorkspaces(ctx, badIdentifier, &applyOptions)
+		assert.EqualError(t, err, ErrInvalidVariableID.Error())
+
+		removeOptions := VariableSetRemoveFromWorkspacesOptions{
+			Workspaces: []*Workspace{wTest1},
+		}
+		err = client.VariableSets.RemoveFromWorkspaces(ctx, badIdentifier, &removeOptions)
+		assert.EqualError(t, err, ErrInvalidVariableID.Error())
+	})
+
+	t.Run("when workspace ID is invalid", func(t *testing.T) {
+		badWorkspace := &Workspace{
+			ID: badIdentifier,
+		}
+
+		applyOptions := VariableSetApplyToWorkspacesOptions{
+			Workspaces: []*Workspace{badWorkspace},
+		}
+
+		err := client.VariableSets.ApplyToWorkspaces(ctx, vsTest.ID, &applyOptions)
+		assert.EqualError(t, err, ErrRequiredWorkspaceID.Error())
+
+		removeOptions := VariableSetRemoveFromWorkspacesOptions{
+			Workspaces: []*Workspace{badWorkspace},
+		}
+
+		err = client.VariableSets.RemoveFromWorkspaces(ctx, vsTest.ID, &removeOptions)
+		assert.EqualError(t, err, ErrRequiredWorkspaceID.Error())
+	})
 }
 
 func TestVariableSetsUpdateWorkspaces(t *testing.T) {