tmtt/redis-om-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Test for embedded JSON models

Browse source
This commit is contained in:
Andrew Brookins 2021-09-01 13:06:23 -07:00
parent 59dced95a6
commit 87b96ffb20
1 changed files with 51 additions and 45 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

96
tests/test_json_model.py
View file

@ -12,6 +12,7 @@ from redis_developer.orm import (
)
r = redis.Redis()
today = datetime.datetime.today()
class BaseJsonModel(JsonModel):
@ -27,9 +28,14 @@ class Address(BaseJsonModel):
postal_code: str
class Item(BaseJsonModel):
price: decimal.Decimal
name: str
class Order(BaseJsonModel):
items: List[Item]
total: decimal.Decimal
currency: str
created_on: datetime.datetime
@ -46,8 +52,16 @@ class Member(BaseJsonModel):
orders: Optional[List[Order]]
class Meta(BaseJsonModel.Meta):
model_key_prefix = "member"
primary_key_pattern = ""
model_key_prefix = "member" # This is the default
address = Address(
address_line_1="1 Main St.",
city="Happy Town",
state="WY",
postal_code=11111,
country="USA"
)
def test_validates_required_fields():
@ -56,7 +70,7 @@ def test_validates_required_fields():
Member(
first_name="Andrew",
zipcode="97086",
join_date=datetime.date.today()
join_date=today
)
@ -72,58 +86,36 @@ def test_validates_field():
# Passes validation
def test_validation_passes():
address = Address(
address_line_1="1 Main St.",
city="Happy Town",
state="WY",
postal_code=11111,
country="USA"
)
member = Member(
first_name="Andrew",
last_name="Brookins",
email="a@example.com",
address=address,
join_date=datetime.date.today()
join_date=today
)
assert member.first_name == "Andrew"
def test_gets_pk():
address = Address(
new_address = Address(
address_line_1="1 Main St.",
city="Happy Town",
state="WY",
postal_code=11111,
country="USA"
)
assert address.pk is not None
assert new_address.pk is not None
def test_saves_model():
address = Address(
address_line_1="1 Main St.",
city="Happy Town",
state="WY",
postal_code=11111,
country="USA"
)
# Save a model instance to Redis
address.save()
address2 = Address.get(address.pk)
assert address2 == address
# Saves a model with embedded models
def test_saves_with_relationships():
address = Address(
address_line_1="1 Main St.",
city="Happy Town",
state="WY",
postal_code=11111,
country="USA"
)
member = Member(
first_name="Andrew",
last_name="Brookins",
@ -137,17 +129,39 @@ def test_saves_with_relationships():
assert member2.address == address
# Saves a model with deeply embedded models
def test_saves_with_relationships():
hat = Item(
name="Cool hat",
price=2.99
)
shoe = Item(
name="Expensive shoe",
price=299.99
)
order = Order(
total=302.98,
items=[hat, shoe],
created_on=today,
)
member = Member(
first_name="Andrew",
last_name="Brookins",
email="a@example.com",
address=address,
orders=[order],
join_date=today
)
member.save()
member2 = Member.get(member.pk)
assert member2.orders[0] == order
assert member2.orders[0].items[0] == hat
# Save many model instances to Redis
@pytest.mark.skip("Not implemented yet")
def test_saves_many():
address = Address(
address_line_1="1 Main St.",
city="Happy Town",
state="WY",
postal_code=11111,
country="USA"
)
today = datetime.date.today()
members = [
Member(
first_name="Andrew",
@ -169,14 +183,6 @@ def test_saves_many():
@pytest.mark.skip("No implemented yet")
def test_updates_a_model():
address = Address(
address_line_1="1 Main St.",
city="Happy Town",
state="WY",
postal_code=11111,
country="USA"
)
today = datetime.date.today()
member = Member(
first_name="Andrew",
last_name="Brookins",