fix delete vlan test 🤦🏼‍♀️

BytemarkHosting · May 9, 2018 · 4e76cc3 · 4e76cc3
1 parent 90c511a
commit 4e76cc3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/cmd/bytemark/commands/admin/delete_test.go b/cmd/bytemark/commands/admin/delete_test.go
@@ -13,9 +13,9 @@ func TestDeleteVLAN(t *testing.T) {
 	is := is.New(t)
 	_, c, app := testutil.BaseTestAuthSetup(t, true, admin.Commands)
 
-	c.When("ReapVMs").Return(nil).Times(1)
+	c.When("DeleteVLAN", 25).Return(nil).Times(1)
 
-	err := app.Run([]string{"bytemark", "reap", "servers"})
+	err := app.Run([]string{"bytemark", "delete", "vlan", "25"})
 
 	is.Nil(err)
 
@@ -28,9 +28,9 @@ func TestDeleteVLANError(t *testing.T) {
 	is := is.New(t)
 	_, c, app := testutil.BaseTestAuthSetup(t, true, admin.Commands)
 
-	c.When("ReapVMs").Return(fmt.Errorf("Could not delete VLAN")).Times(1)
+	c.When("DeleteVLAN", 204).Return(fmt.Errorf("Could not delete VLAN")).Times(1)
 
-	err := app.Run([]string{"bytemark", "reap", "servers"})
+	err := app.Run([]string{"bytemark", "delete", "vlan", "204"})
 
 	is.NotNil(err)
 

diff --git a/mocks/client.go b/mocks/client.go
@@ -439,7 +439,7 @@ func (c *Client) ReapVMs() error {
 	return r.Error(0)
 }
 func (c *Client) DeleteVLAN(id int) error {
-	r := c.Called()
+	r := c.Called(id)
 	return r.Error(0)
 }
 func (c *Client) AdminCreateGroup(name lib.GroupName, vlanNum int) error {