def test_update_assembly_status(self, mock_registry, mock_uas): mock_assembly = mock.MagicMock() mock_registry.Assembly.get_by_id.return_value = mock_assembly shell_handler.update_assembly_status(self.ctx, '1234', 'BUILDING') self.assertEqual(mock_registry.Assembly.get_by_id.call_count, 0) self.assertEqual(mock_registry.save.call_count, 0) self.assertEqual(mock_uas.call_count, 1)
def test_update_assembly_status(self, mock_registry): mock_assembly = mock.MagicMock() mock_registry.Assembly.get_by_id.return_value = mock_assembly shell_handler.update_assembly_status(self.ctx, '1234', 'BUILDING') mock_registry.Assembly.get_by_id.assert_called_once_with( self.ctx, '1234') mock_assembly.save.assert_called_once_with(self.ctx) self.assertEqual(mock_assembly.status, 'BUILDING')
def test_update_assembly_status(self, mock_registry): mock_assembly = mock.MagicMock() mock_registry.Assembly.get_by_id.return_value = mock_assembly shell_handler.update_assembly_status(self.ctx, '1234', 'BUILDING') mock_registry.Assembly.get_by_id.assert_called_once_with(self.ctx, '1234') mock_assembly.save.assert_called_once_with(self.ctx) self.assertEqual(mock_assembly.status, 'BUILDING')
def test_update_assembly_status_pass(self, mock_registry, mock_uas): shell_handler.update_assembly_status(self.ctx, None, 'BUILDING') self.assertEqual(mock_registry.call_count, 0)