def test_getObject(self): oid_1 = self.getOID(1) oid_2 = self.getOID(2) txn = Transaction(self.getClientUUID(), self.getNextTID()) object_info = oid_1, None, None txn.addObject(*object_info) self.assertRaises(KeyError, txn.getObject, oid_2) self.assertEqual(txn.getObject(oid_1), object_info)
def testObjects(self): txn = Transaction(self.getClientUUID(), self.getNextTID()) oid1, oid2 = self.getOID(1), self.getOID(2) object1 = oid1, "0" * 20, None object2 = oid2, "1" * 20, None self.assertEqual(txn.getObjectList(), []) self.assertEqual(txn.getOIDList(), []) txn.addObject(*object1) self.assertEqual(txn.getObjectList(), [object1]) self.assertEqual(txn.getOIDList(), [oid1]) txn.addObject(*object2) self.assertEqual(txn.getObjectList(), [object1, object2]) self.assertEqual(txn.getOIDList(), [oid1, oid2])