def test_get_by(self): uuid = str(uuid4()) t2 = Table2(id=1, user_uuid=uuid) save_to_database(t2) t2 = Table2.get_by('user_uuid', uuid) assert t2.id == 1
def test_get_by_deleted_contains(self): uuid = str(uuid4()) t2 = Table2(id=1, user_uuid=uuid) save_to_database(t2) t2 = Table2.get_by('user_uuid', uuid, contains_deleted=True) assert t2.id == 1
def test_safe_save_to_database_multi(self): uuid = str(uuid4()) user = User(uuid=uuid, email='*****@*****.**') t2 = Table2(id=1, user_uuid=uuid) assert safe_save_to_database([user, t2]) user = User.get(uuid) assert user.email == '*****@*****.**' t2 = Table2.get(1) assert t2.user_uuid == uuid
def test_get_by_for_update(self): uuid = str(uuid4()) t2 = Table2(id=1, user_uuid=uuid) save_to_database(t2) t2 = Table2.get_by('user_uuid', uuid, for_update=True) assert t2.id == 1 t2.key2 = 10 save_to_database(t2) t2 = Table2.get_by('user_uuid', uuid) assert t2.key2 == 10
def test_get_by_user(self): uuid = str(uuid4()) t21 = Table2(id=1, user_uuid=uuid, key=1, key2=1, created_at=arrow.get('2011-10-21')) t22 = Table2(id=2, user_uuid=uuid, key=1, key2=2, created_at=arrow.get('2012-10-21')) t23 = Table2(id=3, user_uuid=uuid, key=1, key2=3, created_at=arrow.get('2013-10-21')) t24 = Table2(id=4, user_uuid=uuid, key=1, key2=4, created_at=arrow.get('2014-10-21')) t25 = Table2(id=5, user_uuid=uuid, key=1, key2=5, created_at=arrow.get('2015-10-21')) save_to_database([t21, t22, t23, t24, t25]) results = Table2.get_by_user(uuid, offset=2, limit=2) assert len(results) == 2 assert results[0].key2 == 3 assert results[1].key2 == 2
def test_from_dict(self): uuid = str(uuid4()) now = arrow.get() t2_dict = { 'id': 1, 'user_uuid': uuid, 'key': 1, 'key2': 1, 'created_at': now.timestamp, } t2 = Table2.from_dict(t2_dict) assert t2.created_at == arrow.get(now.timestamp) assert not t2.key2
def test_string(self): t2 = Table2(id=1, key=1, key2=1, user_type=UserType.admin, created_at=arrow.get('2017-10-21')) t2_str = str(t2) assert t2_str.startswith('<Table2 ') assert ast.literal_eval(t2_str[8:-1]) == { 'id': 1, 'user_uuid': None, 'user_type': 'admin', 'key2': True, 'created_at': 1508544000, }
def test_to_json(self): uuid = str(uuid4()) now = arrow.get() t2 = Table2(id=1, user_uuid=uuid, key=1, key2=1, user_type=UserType.regular, created_at=now) t2_json = t2.to_json() assert json.loads(t2_json) == { 'id': 1, 'user_uuid': uuid, 'user_type': 'regular', 'key2': True, 'created_at': now.timestamp, }
def test_get_multi(self): uuid = str(uuid4()) t21 = Table2(id=1, user_uuid=uuid, key=1, key2=1) t22 = Table2(id=2, user_uuid=uuid, key=1, key2=2) t23 = Table2(id=3, user_uuid=uuid, key=1, key2=3) t24 = Table2(id=4, user_uuid=uuid, key=1, key2=4) t25 = Table2(id=5, user_uuid=uuid, key=1, key2=5) save_to_database([t21, t22, t23, t24, t25]) results = Table2.get_multi('key', 1, 'key2', offset=2, limit=2) assert len(results) == 2 assert results[0].key2 == 3 assert results[1].key2 == 2 results = Table2.get_multi('key', 1, 'key2', desc=False, limit=7) assert len(results) == 5 assert results[0].key2 == 1
def test_to_dict(self): uuid = str(uuid4()) now = arrow.get() t2 = Table2(id=1, user_uuid=uuid, key=1, key2=1, user_type=UserType.regular, created_at=now) t2_dict = t2.to_dict() t2_inflate = t2.inflate_to_dict() assert t2_inflate == t2_dict assert t2_dict == { 'id': 1, 'user_uuid': uuid, 'user_type': 'regular', 'key2': True, 'created_at': now.timestamp, }
def test_get_by_deleted(self): uuid = str(uuid4()) t2 = Table2(id=1, user_uuid=uuid, deleted_at=arrow.get()) save_to_database(t2) assert not Table2.get_by('user_uuid', uuid)
def test_get_by_empty(self): t2 = Table2(id=1, user_uuid=str(uuid4())) save_to_database(t2) assert not Table2.get_by('user_uuid', None)