def test_append(): rdoc = RDoc(_id=ObjectId(), i=10) obj = ReferencesList(RDoc, []) obj.append(rdoc) assert obj.ids == [rdoc.id] assert obj._documents == [rdoc]
def test_delitem(): rdoc = RDoc(_id=ObjectId(), i=10) obj = ReferencesList(RDoc, []) obj.append(rdoc) del obj[0] assert len(obj) == 0 assert not obj
def test_setitem(): rdoc_1 = RDoc(_id=ObjectId(), i=10) obj = ReferencesList(RDoc, []) obj.append(rdoc_1) rdoc_2 = RDoc(_id=ObjectId(), i=10) obj[0] = rdoc_2 assert obj.ids == [rdoc_2.id] assert obj._documents == [rdoc_2]
def test_pop(): rdoc_1 = RDoc(_id=ObjectId(), i=10) rdoc_2 = RDoc(_id=ObjectId(), i=12) obj = ReferencesList(RDoc, []) obj.append(rdoc_1) obj.append(rdoc_2) res = obj.pop(1) assert res == rdoc_2 assert obj.ids == [rdoc_1.id] assert obj._documents == [rdoc_1]
def test_insert(): rdoc_1 = RDoc(_id=ObjectId(), i=10) rdoc_2 = RDoc(_id=ObjectId(), i=12) rdoc_3 = RDoc(_id=ObjectId(), i=13) obj = ReferencesList(RDoc, []) obj.insert(0, rdoc_1) obj.insert(0, rdoc_2) obj.insert(1, rdoc_3) assert obj.ids == [rdoc_2.id, rdoc_3.id, rdoc_1.id] assert obj._documents == [rdoc_2, rdoc_3, rdoc_1]
def test_getitem(): rdoc = RDoc(_id=ObjectId(), i=10) obj = ReferencesList(RDoc, []) obj.append(rdoc) assert obj[0] == rdoc