implement compute tests for StopInstances

daisy/compute/compute_test.go

@@ -169,6 +169,30 @@ func TestCreates(t *testing.T) {
 	}
 }
 
+func TestStops(t *testing.T) {
+	var stopURL, opGetURL string
+	svr, c, err := NewTestClient(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "POST" && r.URL.String() == stopURL {
+			fmt.Fprint(w, `{}`)
+		} else if r.Method == "GET" && r.URL.String() == opGetURL {
+			fmt.Fprint(w, `{"Status":"DONE"}`)
+		} else {
+			w.WriteHeader(500)
+			fmt.Fprintln(w, "URL and Method not recognized:", r.Method, r.URL)
+		}
+	}))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer svr.Close()
+
+	stopURL = fmt.Sprintf("/%s/zones/%s/instances/%s/stop?alt=json", testProject, testZone, testInstance)
+	opGetURL = fmt.Sprintf("/%s/zones/%s/operations/?alt=json", testProject, testZone)
+	if err := c.StopInstance(testProject, testZone, testInstance); err != nil {
+		t.Errorf("error running Stop: %v", err)
+	}
+}
+
 func TestDeletes(t *testing.T) {
 	var deleteURL, opGetURL *string
 	svr, c, err := NewTestClient(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

daisy/compute/test_client.go

@@ -51,6 +51,7 @@ type TestClient struct {
 	CreateImageFn         func(project string, i *compute.Image) error
 	CreateInstanceFn      func(project, zone string, i *compute.Instance) error
 	CreateNetworkFn       func(project string, i *compute.Network) error
+	StopInstanceFn        func(project, zone, name string) error
 	DeleteDiskFn          func(project, zone, name string) error
 	DeleteImageFn         func(project, name string) error
 	DeleteInstanceFn      func(project, zone, name string) error
@@ -127,6 +128,14 @@ func (c *TestClient) CreateNetwork(project string, n *compute.Network) error {
 	return c.client.CreateNetwork(project, n)
 }
 
+// StopInstance uses the override method StopInstanceFn or the real implementation.
+func (c *TestClient) StopInstance(project, zone, name string) error {
+	if c.StopInstanceFn != nil {
+		return c.StopInstanceFn(project, zone, name)
+	}
+	return c.client.StopInstance(project, zone, name)
+}
+
 // DeleteDisk uses the override method DeleteDiskFn or the real implementation.
 func (c *TestClient) DeleteDisk(project, zone, name string) error {
 	if c.DeleteDiskFn != nil {

daisy/compute/test_client_test.go

@@ -48,6 +48,7 @@ func TestTestClient(t *testing.T) {
 		{"create image", func() { c.CreateImage("a", &compute.Image{}) }, "/a/global/images?alt=json"},
 		{"create instance", func() { c.CreateInstance("a", "b", &compute.Instance{}) }, "/a/zones/b/instances?alt=json"},
 		{"create network", func() { c.CreateNetwork("a", &compute.Network{}) }, "/a/global/networks?alt=json"},
+		{"instances stop", func() { c.StopInstance("a", "b", "c") }, "/a/zones/b/instances/c/stop?alt=json"},
 		{"delete disk", func() { c.DeleteDisk("a", "b", "c") }, "/a/zones/b/disks/c?alt=json"},
 		{"delete image", func() { c.DeleteImage("a", "b") }, "/a/global/images/b?alt=json"},
 		{"delete instance", func() { c.DeleteInstance("a", "b", "c") }, "/a/zones/b/instances/c?alt=json"},
@@ -106,6 +107,7 @@ func TestTestClient(t *testing.T) {
 	c.CreateImageFn = func(_ string, _ *compute.Image) error { fakeCalled = true; return nil }
 	c.CreateInstanceFn = func(_, _ string, _ *compute.Instance) error { fakeCalled = true; return nil }
 	c.CreateNetworkFn = func(_ string, _ *compute.Network) error { fakeCalled = true; return nil }
+	c.StopInstanceFn = func(_, _, _ string) error { fakeCalled = true; return nil }
 	c.DeleteDiskFn = func(_, _, _ string) error { fakeCalled = true; return nil }
 	c.DeleteImageFn = func(_, _ string) error { fakeCalled = true; return nil }
 	c.DeleteInstanceFn = func(_, _, _ string) error { fakeCalled = true; return nil }