class BaseRemoveCommandTests(rpm_support_base.PulpClientTests): def setUp(self): super(BaseRemoveCommandTests, self).setUp() self.command = BaseRemoveCommand(self.context, "remove", "base-remove", "base-type") def test_structure(self): self.assertTrue(isinstance(self.command, UnitRemoveCommand)) @mock.patch("pulp_rpm.extension.admin.units_display.display_units") def test_succeeded(self, mock_display): # Setup fake_units = "u" fake_task = mock.MagicMock() fake_task.result = fake_units # Test self.command.succeeded(fake_task) # Verify mock_display.assert_called_once_with(self.prompt, fake_units, self.command.unit_threshold)
class BaseRemoveCommandTests(rpm_support_base.PulpClientTests): def setUp(self): super(BaseRemoveCommandTests, self).setUp() self.command = BaseRemoveCommand(self.context, 'remove', 'base-remove', 'base-type') def test_structure(self): self.assertTrue(isinstance(self.command, UnitRemoveCommand)) @mock.patch('pulp_rpm.extension.admin.units_display.get_formatter_for_type') def test_get_formatter_for_type(self, mock_display): # Setup fake_units = 'u' fake_task = mock.MagicMock() fake_task.result = fake_units # Test self.command.get_formatter_for_type('foo-type') # Verify mock_display.assert_called_once_with('foo-type')
def setUp(self): super(BaseRemoveCommandTests, self).setUp() self.command = BaseRemoveCommand(self.context, "remove", "base-remove", "base-type")
def setUp(self): super(BaseRemoveCommandTests, self).setUp() self.command = BaseRemoveCommand(self.context, 'remove', 'base-remove', 'base-type')