class AddMachineTest(CommandBase): def setUp(self): super(AddMachineTest, self).setUp() self.cmd = AddMachine(self.config, self.provider, self.env) @mock.patch('juju_linode.constraints.get_images') def test_add_machine(self, mock_get_images): mock_get_images.return_value = IMAGE_MAP self.setup_env() self.cmd.run()
def setUp(self): super(AddMachineTest, self).setUp() self.cmd = AddMachine(self.config, self.provider, self.env)