def test_basic(self): pth_seg_db = PathSegmentDB() pth_seg_db._db = create_mock(['delete']) pth_seg_db._db.return_value = "data1" ntools.eq_(pth_seg_db.delete("data2"), DBResult.ENTRY_DELETED) pth_seg_db._db.assert_called_once_with(id="data2") pth_seg_db._db.delete.assert_called_once_with("data1")
def test_outdated(self, db_rec): inst = PathSegmentDB() pcb = self._mk_pcb(-1) cur_rec = create_mock_full({"pcb": self._mk_pcb(0)}) inst._db = create_mock_full(return_value={0: {'record': cur_rec}}) record = create_mock_full({'id': "idstr"}) db_rec.return_value = record # Call ntools.eq_(inst.update(pcb), DBResult.NONE) # Tests pcb.get_expiration_time.assert_called_once_with() cur_rec.pcb.get_expiration_time.assert_called_once_with()
def test_add(self, db_rec): inst = PathSegmentDB() inst._db = create_mock_full({'insert()': None}, return_value=[]) pcb = self._mk_pcb() record = create_mock_full({'id': "id str"}) db_rec.return_value = record # Call ntools.eq_(inst.update(pcb), DBResult.ENTRY_ADDED) # Tests db_rec.assert_called_once_with(pcb) inst._db.assert_called_once_with(id="id str", sibra=True) inst._db.insert.assert_called_once_with(record, "id str", 1, 2, 3, 4, True)
def test(self, time): inst = PathSegmentDB() inst._db = create_mock(['delete']) recs = [] for i in range(5): rec = create_mock(['exp_time', 'pcb']) rec.exp_time = i rec.pcb = create_mock(["short_desc"]) recs.append({'record': rec}) time.return_value = 2 # Call ntools.eq_(inst._exp_call_records(recs), recs[2:]) # Tests inst._db.delete.assert_called_once_with(recs[:2])
def test(self): inst = PathSegmentDB() inst._parse_call_kwargs = create_mock() inst._parse_call_kwargs.return_value = {"arg1": "val1"} inst._exp_call_records = create_mock() inst._sort_call_pcbs = create_mock() inst._db = create_mock() # Call ntools.eq_(inst("data", a="b"), inst._sort_call_pcbs.return_value) # Tests inst._parse_call_kwargs.assert_called_once_with({"a": "b"}) inst._db.assert_called_once_with("data", arg1="val1") inst._exp_call_records.assert_called_once_with(inst._db.return_value) inst._sort_call_pcbs.assert_called_once_with( False, inst._exp_call_records.return_value)
def test_with_segment_ttl(self, db_rec, time): segment_ttl = 300 inst = PathSegmentDB(segment_ttl) cur_rec = create_mock(['pcb', 'id', 'exp_time']) cur_rec.pcb = self._mk_pcb(0) cur_rec.exp_time = 10 inst._db = create_mock_full(return_value={0: {'record': cur_rec}}) pcb = self._mk_pcb(1) db_rec.return_value = create_mock(['id']) time.return_value = 1 # Call inst.update(pcb) # Tests db_rec.assert_called_once_with(pcb, segment_ttl + time.return_value) ntools.eq_(cur_rec.exp_time, 301)
def test_update(self, db_rec): inst = PathSegmentDB() pcb = self._mk_pcb(1) cur_rec = create_mock_full({ "pcb": self._mk_pcb(0), "id": "cur rec", "exp_time": 44 }) inst._db = create_mock_full(return_value={0: {'record': cur_rec}}) db_rec.return_value = create_mock_full({ 'id': "record", 'exp_time': 32 }) # Call ntools.eq_(inst.update(pcb), DBResult.ENTRY_UPDATED) # Tests ntools.eq_(cur_rec.pcb, pcb)
def test_not_present(self): pth_seg_db = PathSegmentDB() pth_seg_db._db = create_mock() pth_seg_db._db.return_value = False ntools.eq_(pth_seg_db.delete("data"), DBResult.NONE)