def test_first_save_registers_purchase():
for account in Account.get_many():
account.delete()
for article in Article.get_many():
article.delete()
for purchase in Purchase.get_many():
purchase.delete()
score = random.randint(0, 100)
purchaser_id = Account({
'user_id': user_id,
'email': fake.email(),
'name': fake.sentence(),
'score': score
}).save()
article_id = Article({
'name': fake.word(),
'description': fake.sentence(),
'available_units': fake.pyint(),
'price': 20.0,
'latitude': 0.0,
'longitude': 0.0,
'user': '******',
}).save()
purchase_id = Purchase({
'user_id': purchaser_id,
'article_id': article_id,
'units': 1
}).save()
assert Account.get_one(purchaser_id).score() == score + 5
Purchase.get_one(purchase_id).save()
assert Account.get_one(purchaser_id).score() == score + 5
def test_first_save_registers_publication():
for account in Account.get_many():
account.delete()
for article in Article.get_many():
article.delete()
user_id = 'user_id'
score = random.randint(0, 100)
account_id = Account({
'user_id': user_id,
'email': fake.email(),
'name': fake.sentence(),
'score': score
}).save()
article_id = Article({
'name': fake.word(),
'description': fake.sentence(),
'available_units': fake.pyint(),
'price': 20.0,
'latitude': 0.0,
'longitude': 0.0,
'user': user_id,
}).save()
assert Account.get_one(account_id).score() == score + 1
Article.get_one(article_id).save()
assert Account.get_one(account_id).score() == score + 1
def get_by_seller(cls, seller: Account):
article_ids = [
a.get_id() for a in Article.get_many(user=seller['user_id'])
]
if not article_ids:
return []
purchases = cls.get_many(article_id=article_ids)
return cls.expand(purchases)
def get_by_owner(self):
account = Account.get_one(self.owner)
article_ids = [a.get_id()
for a in Article.get_many(user=account['user_id'])]
questions = []
for article_id in article_ids:
questions.extend(
Question.get_many(article_id=article_id)
)
return self.append_article_to_questions(questions)
def teardown_function():
for a in Article.get_many():
a.delete()
