def test_repr_getter_setter(self): from core.Test import Test test_id = str(uuid.uuid4()) owner = str(uuid.uuid4()) test_type = str(uuid.uuid4()) test = Test(test_id, owner, test_type) assert '{0}'.format(test) == '<Test {0} ({1}) by {2}>'.format(test_id, test_type, owner) assert test.to_dict() == {'test_id': test_id, 'owner': owner, 'type': test_type} test._last_seen = datetime.datetime.now() assert test.to_dict()['last_seen'] == test._last_seen test2 = test.from_dict(test.to_dict()) assert test2.to_dict() == test.to_dict()
def test_get_one(self): from core.Test import Test test_id1 = str(uuid.uuid4()) test_id2 = str(uuid.uuid4()) test_id3 = str(uuid.uuid4()) owner1 = str(uuid.uuid4()) owner2 = str(uuid.uuid4()) test_type = str(uuid.uuid4()) Test(test_id1, owner1, test_type).save() Test(test_id2, owner1, test_type).save() Test(test_id2, owner2, test_type).save() test = Test(test_id=test_id1) one = Test.get_one(test.to_dict()) assert one is not None assert one._test_id == test_id1 assert one._owner == owner1 test = Test(test_id=test_id2) one = Test.get_one(test.to_dict()) assert one._test_id == test_id2 assert one._owner == owner2 assert one is not None test = Test(test_id=test_id3) one = Test.get_one(test.to_dict()) assert one is None
def test_get_all(self): from core.Test import Test test_id1 = str(uuid.uuid4()) test_id2 = str(uuid.uuid4()) test_id3 = str(uuid.uuid4()) owner = str(uuid.uuid4()) owner2 = str(uuid.uuid4()) test_type = str(uuid.uuid4()) Test(test_id1, owner, test_type).save() Test(test_id2, owner, test_type).save() Test(test_id3, owner2, test_type).save() test = Test(owner=owner) atl = Test.get_all(test.to_dict()) assert len(atl) == 2 assert atl[0]._test_id == test_id1 assert atl[1]._test_id == test_id2