Commands to empty storage pools and heads

BytemarkHosting · Mar 27, 2017 · c29094c · c29094c
1 parent 49babef
commit c29094c
Show file tree

Hide file tree

Showing 6 changed files with 164 additions and 0 deletions.
diff --git a/cmd/bytemark/empty.go b/cmd/bytemark/empty.go
@@ -0,0 +1,55 @@
+package main
+
+import (
+	"github.com/BytemarkHosting/bytemark-client/util/log"
+	"github.com/urfave/cli"
+)
+
+func init() {
+	adminCommands = append(adminCommands, cli.Command{
+		Name:   "empty",
+		Action: cli.ShowSubcommandHelp,
+		Subcommands: []cli.Command{
+			{
+				Name:      "storage_pool",
+				Usage:     "empty a storage pool",
+				UsageText: "bytemark --admin empty storage_pool <storage_pool>",
+				Flags: []cli.Flag{
+					cli.StringFlag{
+						Name:  "storage_pool",
+						Usage: "the ID or label of the storage pool to be emptied",
+					},
+				},
+				Action: With(OptionalArgs("storage_pool"), RequiredFlags("storage_pool"), AuthProvider, func(c *Context) error {
+					if err := global.Client.EmptyStoragePool(c.String("storage_pool")); err != nil {
+						return err
+					}
+
+					log.Output("Storage pool updated")
+
+					return nil
+				}),
+			},
+			{
+				Name:      "head",
+				Usage:     "empty a head",
+				UsageText: "bytemark --admin empty head <head>",
+				Flags: []cli.Flag{
+					cli.StringFlag{
+						Name:  "head",
+						Usage: "the ID or label of the head to be emptied",
+					},
+				},
+				Action: With(OptionalArgs("head"), RequiredFlags("head"), AuthProvider, func(c *Context) error {
+					if err := global.Client.EmptyHead(c.String("head")); err != nil {
+						return err
+					}
+
+					log.Output("Head updated")
+
+					return nil
+				}),
+			},
+		},
+	})
+}
diff --git a/cmd/bytemark/empty_test.go b/cmd/bytemark/empty_test.go
@@ -0,0 +1,67 @@
+package main
+
+import (
+	"fmt"
+	"github.com/cheekybits/is"
+	"testing"
+)
+
+func TestEmptyStoragePool(t *testing.T) {
+	is := is.New(t)
+	_, c := baseTestAuthSetup(t, true)
+
+	c.When("EmptyStoragePool", "pool1").Return(nil).Times(1)
+
+	err := global.App.Run([]string{"bytemark", "empty", "storage_pool", "pool1"})
+
+	is.Nil(err)
+
+	if ok, err := c.Verify(); !ok {
+		t.Fatal(err)
+	}
+}
+
+func TestEmptyStoragePoolError(t *testing.T) {
+	is := is.New(t)
+	_, c := baseTestAuthSetup(t, true)
+
+	c.When("EmptyStoragePool", "pool1").Return(fmt.Errorf("Could not empty storage pool")).Times(1)
+
+	err := global.App.Run([]string{"bytemark", "empty", "storage_pool", "pool1"})
+
+	is.NotNil(err)
+
+	if ok, err := c.Verify(); !ok {
+		t.Fatal(err)
+	}
+}
+
+func TestEmptyHead(t *testing.T) {
+	is := is.New(t)
+	_, c := baseTestAuthSetup(t, true)
+
+	c.When("EmptyHead", "pool1").Return(nil).Times(1)
+
+	err := global.App.Run([]string{"bytemark", "empty", "head", "pool1"})
+
+	is.Nil(err)
+
+	if ok, err := c.Verify(); !ok {
+		t.Fatal(err)
+	}
+}
+
+func TestEmptyHeadError(t *testing.T) {
+	is := is.New(t)
+	_, c := baseTestAuthSetup(t, true)
+
+	c.When("EmptyHead", "pool1").Return(fmt.Errorf("Could not empty storage pool")).Times(1)
+
+	err := global.App.Run([]string{"bytemark", "empty", "head", "pool1"})
+
+	is.NotNil(err)
+
+	if ok, err := c.Verify(); !ok {
+		t.Fatal(err)
+	}
+}
diff --git a/lib/admin_calls.go b/lib/admin_calls.go
@@ -249,3 +249,23 @@ func (c *bytemarkClient) CancelVMMigration(id int) (err error) {
 	_, _, err = r.Run(nil, nil)
 	return
 }
+
+func (c *bytemarkClient) EmptyStoragePool(idOrLabel string) (err error) {
+	r, err := c.BuildRequest("POST", BrainEndpoint, "/admin/storage_pools/%s/empty", idOrLabel)
+	if err != nil {
+		return
+	}
+
+	_, _, err = r.Run(nil, nil)
+	return
+}
+
+func (c *bytemarkClient) EmptyHead(idOrLabel string) (err error) {
+	r, err := c.BuildRequest("POST", BrainEndpoint, "/admin/heads/%s/empty", idOrLabel)
+	if err != nil {
+		return
+	}
+
+	_, _, err = r.Run(nil, nil)
+	return
+}
diff --git a/lib/admin_calls_test.go b/lib/admin_calls_test.go
@@ -501,3 +501,15 @@ func TestPostCancelVMMigration(t *testing.T) {
 		return client.CancelVMMigration(1235)
 	})
 }
+
+func TestPostEmptyStoragePool(t *testing.T) {
+	simplePostTest(t, "/admin/storage_pools/pool1/empty", ``, func(client Client) error {
+		return client.EmptyStoragePool("pool1")
+	})
+}
+
+func TestPostEmptyHead(t *testing.T) {
+	simplePostTest(t, "/admin/heads/head1/empty", ``, func(client Client) error {
+		return client.EmptyHead("head1")
+	})
+}
diff --git a/lib/interface.go b/lib/interface.go
@@ -225,4 +225,6 @@ type Client interface {
 	CreateIPRange(ipRange string, vlanNum int) error
 	CancelDiscMigration(id int) error
 	CancelVMMigration(id int) error
+	EmptyStoragePool(idOrLabel string) error
+	EmptyHead(idOrLabel string) error
 }
diff --git a/mocks/client.go b/mocks/client.go
@@ -449,3 +449,11 @@ func (c *Client) CancelVMMigration(id int) error {
 	r := c.Called(id)
 	return r.Error(0)
 }
+func (c *Client) EmptyStoragePool(idOrLabel string) error {
+	r := c.Called(idOrLabel)
+	return r.Error(0)
+}
+func (c *Client) EmptyHead(idOrLabel string) error {
+	r := c.Called(idOrLabel)
+	return r.Error(0)
+}