Exemplo n.º 1
0
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)
Exemplo n.º 2
0
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)