def _mock_call_log(self,
cel_ids=None,
date=None,
id=None,
source_line_identity=None,
destination_line_identity=None,
answered=False):
if cel_ids is None:
cel_ids = ()
if date is None:
date = datetime.now()
call_log = CallLog(id=id,
date=date,
duration=timedelta(0),
source_line_identity=source_line_identity,
destination_line_identity=destination_line_identity,
answered=answered)
call_log.add_related_cels(cel_ids)
self.call_log_rows.append(
CallLogSchema(id=id,
date=date,
source_line_identity=source_line_identity,
destination_line_identity=destination_line_identity,
answered=answered,
duration=timedelta(3)))
return call_log
def _mock_call_log(self, cel_ids=None, date=None, id=None, source_line_identity=None, destination_line_identity=None, answered=False):
if cel_ids is None:
cel_ids = ()
if date is None:
date = datetime.now()
call_log = CallLog(id=id,
date=date,
duration=timedelta(0),
source_line_identity=source_line_identity,
destination_line_identity=destination_line_identity,
answered=answered)
call_log.add_related_cels(cel_ids)
self.call_log_rows.append(CallLogSchema(id=id,
date=date,
source_line_identity=source_line_identity,
destination_line_identity=destination_line_identity,
answered=answered,
duration=timedelta(3)))
return call_log
class TestCallLogModel(TestCase):
def setUp(self):
self.call_log = CallLog()
def tearDown(self):
pass
def test_get_related_cels_empty(self):
result = self.call_log.get_related_cels()
assert_that(result, contains())
def test_add_then_get_related_cel(self):
cel_id_1, cel_id_2, cel_id_3 = Mock(), Mock(), Mock()
self.call_log.add_related_cels([cel_id_1, cel_id_2])
result = self.call_log.get_related_cels()
assert_that(result, contains(cel_id_1, cel_id_2))
self.call_log.add_related_cels([cel_id_3])
result = self.call_log.get_related_cels()
assert_that(result, contains(cel_id_1, cel_id_2, cel_id_3))
class TestCallLogModel(TestCase):
def setUp(self):
self.call_log = CallLog()
def tearDown(self):
pass
def test_get_related_cels_empty(self):
result = self.call_log.get_related_cels()
assert_that(result, contains())
def test_add_then_get_related_cel(self):
cel_id_1, cel_id_2, cel_id_3 = Mock(), Mock(), Mock()
self.call_log.add_related_cels([cel_id_1, cel_id_2])
result = self.call_log.get_related_cels()
assert_that(result, contains(cel_id_1, cel_id_2))
self.call_log.add_related_cels([cel_id_3])
result = self.call_log.get_related_cels()
assert_that(result, contains(cel_id_1, cel_id_2, cel_id_3))
