def test_query_users_basic(self):
user = self.dal.user.create(User(self.user_input_dict))
assert len(self.dal.user.query({"id": user.id})) == 1
_ = self.dal.user.create(User(self.user_input_dict))
assert len(self.dal.user.query({"name":
self.user_input_dict['name']})) == 2
def test_query_users_range_query(self):
_ = self.dal.user.create(User(self.user_input_dict))
_ = self.dal.user.create(User(self.user_input_dict))
_ = self.dal.user.create(User(self.user_input_dict))
users = self.dal.user.query(
{}, sort_key="created_at", sort_order="descending")
result = self.dal.user.query({
"created_at": {
"$lt": users[1].created_at.strftime('%Y-%m-%dT%H:%M:%S.%fZ')
}
})
assert len(users) == 3
assert len(result) == 1
def test_create_user_by_dictionary(self):
user = self.dal.user.create(User(self.user_input_dict))
assert user.id
assert user.name == self.user_input_dict['name']
assert user.created_at
assert user.updated_at
user_2 = self.dal.user.create(User(self.user_input_dict))
assert user_2.id != user.id
test_user_input_dict = self.user_input_dict.copy()
test_user_input_dict['id'] = "user_id"
user_3 = self.dal.user.create(User(test_user_input_dict))
assert user_3.id == test_user_input_dict['id']
def test_query_users_multiple(self):
user_1 = self.dal.user.create(User(self.user_input_dict))
user_2 = self.dal.user.create(User(self.user_input_dict))
user_3 = self.dal.user.create(User(self.user_input_dict))
results = self.dal.user.query({},
sort_key="created_at",
sort_order="ascending")
assert len(results) == 3
assert results[0].created_at == user_1.created_at
assert results[1].created_at == user_2.created_at
results = self.dal.user.query({},
sort_key="created_at",
sort_order="descending")
assert len(results) == 3
assert results[0].created_at == user_3.created_at
assert results[1].created_at == user_2.created_at
# Wrong order being passed in
failed = False
try:
_ = self.dal.user.query({},
sort_key='created_at',
sort_order='wrong_order')
except InvalidArgumentType:
failed = True
assert failed
# Wrong key and order being passed in
failed = False
try:
_ = self.dal.user.query({},
sort_key='wrong_key',
sort_order='wrong_order')
except InvalidArgumentType:
failed = True
assert failed
# wrong key and right order being passed in
expected_items = self.dal.user.query({},
sort_key='created_at',
sort_order='ascending')
items = self.dal.user.query({},
sort_key='wrong_key',
sort_order='ascending')
expected_ids = [item.id for item in expected_items]
ids = [item.id for item in items]
assert set(expected_ids) == set(ids)
def test_init_with_id(self):
self.input_dict['id'] = "test"
user_entity = User(self.input_dict)
for k, v in self.input_dict.items():
assert getattr(user_entity, k) == v
assert user_entity.created_at
assert user_entity.updated_at
def test_init_no_id(self):
user_entity = User(self.input_dict)
for k, v in self.input_dict.items():
assert getattr(user_entity, k) == v
assert user_entity.id == None
assert user_entity.created_at
assert user_entity.updated_at
def test_delete_user(self):
user = self.dal.user.create(User(self.user_input_dict))
self.dal.user.delete(user.id)
deleted = False
try:
self.dal.user.get_by_id(user.id)
except EntityNotFound:
deleted = True
assert deleted
def test_get_by_id_user_new_driver_instance(self):
user = self.dal.user.create(User(self.user_input_dict))
# create new dal with new driver instance (fails)
new_driver_instance = BlitzDBDALDriver("file", self.temp_dir)
new_dal_instance = LocalDAL(new_driver_instance)
new_user_1 = new_dal_instance.user.get_by_id(user.id)
assert new_user_1.id == user.id
# create new dal instance with same driver (success)
new_dal_instance = LocalDAL(self.datadriver)
new_user_2 = new_dal_instance.user.get_by_id(user.id)
assert new_user_2.id == user.id
def test_update_user(self):
user = self.dal.user.create(User(self.user_input_dict))
# Update required and optional parameters
updated_user_input_dict = self.user_input_dict.copy()
updated_user_input_dict['id'] = user.id
updated_user_input_dict['name'] = "cooldude"
updated_user_input_dict['created_at'] = datetime.utcnow()
updated_user = self.dal.user.update(updated_user_input_dict)
assert user.id == updated_user.id
assert user.updated_at < updated_user.updated_at
assert updated_user.name == updated_user_input_dict['name']
assert updated_user.created_at == updated_user_input_dict['created_at']
def test_get_by_shortened_id_user(self):
user = self.dal.user.create(User(self.user_input_dict))
result = self.dal.user.get_by_shortened_id(user.id[:10])
assert user.id == result.id
def test_to_dictionary(self):
user_entity = User(self.input_dict)
output_dict = user_entity.to_dictionary()
for k, v in output_dict.items():
assert v == getattr(user_entity, k)
def test_eq(self):
user_entity_1 = User(self.input_dict)
user_entity_2 = User(self.input_dict)
assert user_entity_1 == user_entity_2