You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
146 lines
3.5 KiB
146 lines
3.5 KiB
import pytest |
|
|
|
|
|
@pytest.fixture() |
|
def example_orders(): |
|
from datetime import datetime |
|
from ordr3.models import OrderItem, OrderCategory, OrderStatus |
|
|
|
return [ |
|
OrderItem( |
|
1, |
|
"Ethanol", |
|
"1-23", |
|
"VWR", |
|
OrderCategory.SOLVENT, |
|
"5 l", |
|
20.0, |
|
2, |
|
"EUR", |
|
"DFG", |
|
"urgent", |
|
datetime(2020, 2, 3, 15, 14, 13), |
|
"me", |
|
OrderStatus.OPEN, |
|
), |
|
OrderItem( |
|
2, |
|
"Gloves", |
|
"12-3", |
|
"Carl Roth", |
|
OrderCategory.DISPOSABLE, |
|
"100 St.", |
|
40.0, |
|
5, |
|
"USD", |
|
"BMBF", |
|
"no comment", |
|
datetime(2020, 2, 4, 15, 14, 13), |
|
"you", |
|
OrderStatus.APPROVAL, |
|
), |
|
] |
|
|
|
|
|
@pytest.fixture() |
|
def example_users(): |
|
from ordr3.models import User, UserRole |
|
|
|
return [ |
|
User(1, "You", "Jim", "Smith", "jim.smith", "abcd", UserRole.ADMIN,), |
|
User(2, "Me", "Jane", "Doe", "jane.doe", "1234", UserRole.USER), |
|
] |
|
|
|
|
|
def test_sql_repo_add_order(session, example_orders): |
|
from ordr3.repo import SqlAlchemyRepository |
|
from ordr3.models import OrderItem |
|
|
|
repo = SqlAlchemyRepository(session) |
|
repo.add_order(example_orders[0]) |
|
session.flush() |
|
|
|
order = session.query(OrderItem).first() |
|
|
|
assert order == example_orders[0] |
|
|
|
|
|
def test_sql_repo_get_order(session, example_orders): |
|
from ordr3.repo import SqlAlchemyRepository |
|
|
|
repo = SqlAlchemyRepository(session) |
|
repo.add_order(example_orders[0]) |
|
repo.add_order(example_orders[1]) |
|
session.flush() |
|
|
|
assert example_orders[1] == repo.get_order(2) |
|
|
|
|
|
def test_sql_repo_list_orders(session, example_orders): |
|
from ordr3.repo import SqlAlchemyRepository |
|
|
|
earlier, later = example_orders |
|
repo = SqlAlchemyRepository(session) |
|
repo.add_order(earlier) |
|
repo.add_order(later) |
|
session.flush() |
|
|
|
assert repo.list_orders() == [later, earlier] |
|
|
|
|
|
def test_sql_repo_add_user(session, example_users): |
|
from ordr3.repo import SqlAlchemyRepository |
|
from ordr3.models import User |
|
|
|
repo = SqlAlchemyRepository(session) |
|
repo.add_user(example_users[0]) |
|
session.flush() |
|
|
|
user = session.query(User).first() |
|
|
|
assert user == example_users[0] |
|
|
|
|
|
def test_sql_repo_get_user(session, example_users): |
|
from ordr3.repo import SqlAlchemyRepository |
|
|
|
repo = SqlAlchemyRepository(session) |
|
repo.add_user(example_users[0]) |
|
repo.add_user(example_users[1]) |
|
session.flush() |
|
|
|
assert example_users[1] == repo.get_user(2) |
|
|
|
|
|
def test_sql_repo_get_user_by_username(session, example_users): |
|
from ordr3.repo import SqlAlchemyRepository |
|
|
|
repo = SqlAlchemyRepository(session) |
|
repo.add_user(example_users[0]) |
|
repo.add_user(example_users[1]) |
|
session.flush() |
|
|
|
assert example_users[1] == repo.get_user_by_username("Me") |
|
|
|
|
|
def test_sql_repo_get_user_by_email(session, example_users): |
|
from ordr3.repo import SqlAlchemyRepository |
|
|
|
repo = SqlAlchemyRepository(session) |
|
repo.add_user(example_users[0]) |
|
repo.add_user(example_users[1]) |
|
session.flush() |
|
|
|
assert example_users[1] == repo.get_user_by_email("jane.doe") |
|
|
|
|
|
def test_sql_repo_list_users(session, example_users): |
|
from ordr3.repo import SqlAlchemyRepository |
|
|
|
later, earlier = example_users |
|
repo = SqlAlchemyRepository(session) |
|
repo.add_user(later) |
|
repo.add_user(earlier) |
|
session.flush() |
|
|
|
assert repo.list_users() == [earlier, later]
|
|
|