Add dynamic disk CLI support: disks --dynamic and delete-disk --dynamic

cmd/delete_disk.go

@@ -21,6 +21,10 @@ func (c DeleteDiskCmd) Run(opts DeleteDiskOpts) error {
 		return err
 	}
 
+	if opts.Dynamic {
+		return c.director.DeleteDynamicDisk(opts.Args.CID)
+	}
+
 	disk, err := c.director.FindOrphanDisk(opts.Args.CID)
 	if err != nil {
 		return err

cmd/delete_disk_test.go

@@ -80,5 +80,39 @@ var _ = Describe("DeleteDiskCmd", func() {
 			Expect(err).To(HaveOccurred())
 			Expect(err.Error()).To(ContainSubstring("fake-err"))
 		})
+
+		Context("when --dynamic", func() {
+			BeforeEach(func() {
+				deleteDiskOpts = opts.DeleteDiskOpts{
+					Args:    opts.DeleteDiskArgs{CID: "my-disk"},
+					Dynamic: true,
+				}
+			})
+
+			It("deletes the dynamic disk", func() {
+				err := act()
+				Expect(err).ToNot(HaveOccurred())
+
+				Expect(director.DeleteDynamicDiskArgsForCall(0)).To(Equal("my-disk"))
+			})
+
+			It("returns error if deletion fails", func() {
+				director.DeleteDynamicDiskReturns(errors.New("fake-err"))
+
+				err := act()
+				Expect(err).To(HaveOccurred())
+				Expect(err.Error()).To(ContainSubstring("fake-err"))
+			})
+
+			It("does not delete if confirmation is rejected", func() {
+				ui.AskedConfirmationErr = errors.New("stop")
+
+				err := act()
+				Expect(err).To(HaveOccurred())
+				Expect(err.Error()).To(ContainSubstring("stop"))
+
+				Expect(director.DeleteDynamicDiskCallCount()).To(Equal(0))
+			})
+		})
 	})
 })

cmd/disks.go

@@ -19,8 +19,12 @@ func NewDisksCmd(ui boshui.UI, director boshdir.Director) DisksCmd {
 }
 
 func (c DisksCmd) Run(opts DisksOpts) error {
+	if opts.Dynamic {
+		return c.runDynamic()
+	}
+
 	if !opts.Orphaned {
-		return errors.New("Only --orphaned is supported") //nolint:staticcheck
+		return errors.New("Only --orphaned or --dynamic is supported") //nolint:staticcheck
 	}
 
 	disks, err := c.director.OrphanDisks()
@@ -56,3 +60,42 @@ func (c DisksCmd) Run(opts DisksOpts) error {
 
 	return nil
 }
+
+func (c DisksCmd) runDynamic() error {
+	disks, err := c.director.DynamicDisks()
+	if err != nil {
+		return err
+	}
+
+	table := boshtbl.Table{
+		Content: "dynamic disks",
+		Header: []boshtbl.Header{
+			boshtbl.NewHeader("Name"),
+			boshtbl.NewHeader("Disk CID"),
+			boshtbl.NewHeader("Size"),
+			boshtbl.NewHeader("Disk Pool"),
+			boshtbl.NewHeader("Deployment"),
+			boshtbl.NewHeader("Instance"),
+			boshtbl.NewHeader("AZ"),
+			boshtbl.NewHeader("CPI"),
+		},
+		SortBy: []boshtbl.ColumnSort{{Column: 0}},
+	}
+
+	for _, d := range disks {
+		table.Rows = append(table.Rows, []boshtbl.Value{
+			boshtbl.NewValueString(d.Name()),
+			boshtbl.NewValueString(d.DiskCID()),
+			boshtbl.NewValueMegaBytes(d.Size()),
+			boshtbl.NewValueString(d.DiskPoolName()),
+			boshtbl.NewValueString(d.DeploymentName()),
+			boshtbl.NewValueString(d.InstanceName()),
+			boshtbl.NewValueString(d.AvailabilityZone()),
+			boshtbl.NewValueString(d.CPI()),
+		})
+	}
+
+	c.ui.PrintTable(table)
+
+	return nil
+}

cmd/disks_test.go

@@ -106,7 +106,71 @@ var _ = Describe("DisksCmd", func() {
 		})
 
 		It("returns error if orphaned disks were not requested", func() {
-			Expect(act()).To(Equal(errors.New("Only --orphaned is supported")))
+			Expect(act()).To(Equal(errors.New("Only --orphaned or --dynamic is supported")))
+		})
+
+		Context("when dynamic disks requested", func() {
+			BeforeEach(func() {
+				disksOpts.Dynamic = true
+			})
+
+			It("lists dynamic disks", func() {
+				disks := []boshdir.DynamicDisk{
+					&fakedir.FakeDynamicDisk{
+						NameStub:             func() string { return "my-disk" },
+						DiskCIDStub:          func() string { return "disk-cid-1" },
+						SizeStub:             func() uint64 { return 2048 },
+						DiskPoolNameStub:     func() string { return "large" },
+						DeploymentNameStub:   func() string { return "my-deployment" },
+						InstanceNameStub:     func() string { return "api/abc123" },
+						AvailabilityZoneStub: func() string { return "z1" },
+						CPIStub:              func() string { return "aws_cpi" },
+					},
+				}
+
+				director.DynamicDisksReturns(disks, nil)
+
+				err := act()
+				Expect(err).ToNot(HaveOccurred())
+
+				Expect(ui.Table).To(Equal(boshtbl.Table{
+					Content: "dynamic disks",
+
+					Header: []boshtbl.Header{
+						boshtbl.NewHeader("Name"),
+						boshtbl.NewHeader("Disk CID"),
+						boshtbl.NewHeader("Size"),
+						boshtbl.NewHeader("Disk Pool"),
+						boshtbl.NewHeader("Deployment"),
+						boshtbl.NewHeader("Instance"),
+						boshtbl.NewHeader("AZ"),
+						boshtbl.NewHeader("CPI"),
+					},
+
+					SortBy: []boshtbl.ColumnSort{{Column: 0}},
+
+					Rows: [][]boshtbl.Value{
+						{
+							boshtbl.NewValueString("my-disk"),
+							boshtbl.NewValueString("disk-cid-1"),
+							boshtbl.NewValueMegaBytes(2048),
+							boshtbl.NewValueString("large"),
+							boshtbl.NewValueString("my-deployment"),
+							boshtbl.NewValueString("api/abc123"),
+							boshtbl.NewValueString("z1"),
+							boshtbl.NewValueString("aws_cpi"),
+						},
+					},
+				}))
+			})
+
+			It("returns error if dynamic disks cannot be retrieved", func() {
+				director.DynamicDisksReturns(nil, errors.New("fake-err"))
+
+				err := act()
+				Expect(err).To(HaveOccurred())
+				Expect(err.Error()).To(ContainSubstring("fake-err"))
+			})
 		})
 	})
 })

cmd/opts/opts.go

@@ -745,11 +745,13 @@ type DeleteNetworkArgs struct {
 
 type DisksOpts struct {
 	Orphaned bool `long:"orphaned" short:"o" description:"List orphaned disks"`
+	Dynamic  bool `long:"dynamic" description:"List dynamic disks"`
 	cmd
 }
 
 type DeleteDiskOpts struct {
-	Args DeleteDiskArgs `positional-args:"true" required:"true"`
+	Args    DeleteDiskArgs `positional-args:"true" required:"true"`
+	Dynamic bool           `long:"dynamic" description:"Delete a dynamic disk by name"`
 	cmd
 }
 

cmd/opts/opts_test.go

@@ -2219,6 +2219,14 @@ var _ = Describe("Opts", func() {
 				Expect(getStructTagForName("Args", opts)).To(Equal(`positional-args:"true" required:"true"`))
 			})
 		})
+
+		Describe("Dynamic", func() {
+			It("contains desired values", func() {
+				Expect(getStructTagForName("Dynamic", opts)).To(Equal(
+					`long:"dynamic" description:"Delete a dynamic disk by name"`,
+				))
+			})
+		})
 	})
 
 	Describe("DeleteDiskArgs", func() {