Function TestRemoveVMAction_runAdapter

cmd/finch/virtual_machine_remove_test.go:26–81

(t *testing.T)

}

func TestRemoveVMAction_runAdapter(t *testing.T) {
	t.Parallel()

	testCases := []struct {
		name    string
		mockSvc func(logger *mocks.Logger, creator *mocks.NerdctlCmdCreator, dm *mocks.UserDataDiskManager, ctrl *gomock.Controller)
		args    []string
	}{
		{
			name: "should remove the instance",
			args: []string{},
			mockSvc: func(logger *mocks.Logger, creator *mocks.NerdctlCmdCreator, dm *mocks.UserDataDiskManager, ctrl *gomock.Controller) {
				getVMStatusC := mocks.NewCommand(ctrl)
				creator.EXPECT().CreateWithoutStdio("ls", "-f", "{{.Status}}", limaInstanceName).Return(getVMStatusC)
				getVMStatusC.EXPECT().Output().Return([]byte("Stopped"), nil)
				logger.EXPECT().Debugf("Status of virtual machine: %s", "Stopped")
				logger.EXPECT().Warnf(gomock.Any(), gomock.Any()).AnyTimes()
				dm.EXPECT().DetachUserDataDisk().Return(nil)
				command := mocks.NewCommand(ctrl)
				creator.EXPECT().CreateWithoutStdio("remove", limaInstanceName).Return(command)
				command.EXPECT().CombinedOutput()
				logger.EXPECT().Info(gomock.Any()).AnyTimes()
			},
		},
		{
			name: "should forcibly remove the instance",
			args: []string{
				"--force",
			},
			mockSvc: func(logger *mocks.Logger, creator *mocks.NerdctlCmdCreator, dm *mocks.UserDataDiskManager, ctrl *gomock.Controller) {
				logger.EXPECT().Warnf(gomock.Any(), gomock.Any()).AnyTimes()
				command := mocks.NewCommand(ctrl)
				dm.EXPECT().DetachUserDataDisk().Return(nil)
				creator.EXPECT().CreateWithoutStdio("remove", "--force", limaInstanceName).Return(command)
				command.EXPECT().CombinedOutput()
				logger.EXPECT().Info(gomock.Any()).AnyTimes()
			},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			ctrl := gomock.NewController(t)
			dm := mocks.NewUserDataDiskManager(ctrl)
			logger := mocks.NewLogger(ctrl)
			ncc := mocks.NewNerdctlCmdCreator(ctrl)
			tc.mockSvc(logger, ncc, dm, ctrl)

			cmd := newRemoveVMCommand(ncc, dm, logger)
			cmd.SetArgs(tc.args)
			assert.NoError(t, cmd.Execute())
		})
	}
}

func TestRemoveVMAction_run(t *testing.T) {