class TestCallLogsWriter(TestCase): def setUp(self): self.writer = CallLogsWriter() def tearDown(self): pass @patch('xivo_dao.resources.call_log.dao.create_from_list') @patch('xivo_dao.resources.call_log.dao.delete_from_list') def test_write(self, mock_dao_delete, mock_dao_create): call_logs_creation = CallLogsCreation(new_call_logs=[Mock(), Mock()], call_logs_to_delete=None) self.writer.write(call_logs_creation) mock_dao_create.assert_called_once_with( call_logs_creation.new_call_logs) mock_dao_delete.assert_called_once_with( call_logs_creation.call_logs_to_delete)
class TestCallLogsWriter(TestCase): def setUp(self): self.dao = Mock() self.writer = CallLogsWriter(self.dao) def tearDown(self): pass def test_write(self): call_logs_creation = CallLogsCreation( new_call_logs=[Mock(recordings=[]), Mock(recordings=[])], call_logs_to_delete=None, ) self.writer.write(call_logs_creation) self.dao.call_log.create_from_list.assert_called_once_with( call_logs_creation.new_call_logs) self.dao.call_log.delete_from_list.assert_called_once_with( call_logs_creation.call_logs_to_delete)