def test_update_contract(db: Session) -> None:
user = create_random_user(db)
date_today = date.today()
start = str(date_today)
end = str(date_today + relativedelta(months=+12, days=-1))
created_on = datetime.now()
updated_on = datetime.now()
contract_in = ContractCreate(user_id=user.id, start=start, end=end)
contract = crud.contract.create_with_owner(db=db,
obj_in=contract_in,
user_id=user.id)
contract_update = ContractUpdate(start=start, end=end)
contract2 = crud.contract.update(db=db,
db_obj=contract,
obj_in=contract_update)
assert contract.id == contract2.id
assert contract.user_id == contract2.user_id
assert contract.user.id == contract2.user.id
assert contract.user.firstname == contract2.user.firstname
assert contract.user.email == contract2.user.email
assert contract.start == contract2.start
assert contract.end == contract2.end
assert isinstance(contract2.created_on, datetime)
assert isinstance(contract2.updated_on, datetime)
def test_update_contract(db: Session) -> None:
user = create_random_user(db)
nanny = create_random_user(db)
date_today = date.today()
weekdays = random_weekdays()
weeks = random_int_range(20, 47)
hours = random_int_range(10, 50)
price_hour_standard = random_float_range(2.5, 4, 1)
price_hour_extra = random_float_range(2.5, 4, 1)
price_fees = random_float_range(3.08, 5, 2)
price_meals = random_float_range(2, 6, 1)
start = str(date_today)
end = str(date_today + relativedelta(months=+12, days=-1))
created_on = datetime.now()
updated_on = datetime.now()
contract_in = ContractCreate(user_id=user.id,
nanny_id=nanny.id,
weekdays=weekdays,
weeks=weeks,
hours=hours,
price_hour_standard=price_hour_standard,
price_hour_extra=price_hour_extra,
price_fees=price_fees,
price_meals=price_meals,
start=start,
end=end)
contract = crud.contract.create_with_owner(db=db,
obj_in=contract_in,
user_id=user.id,
nanny_id=nanny.id)
contract_update = ContractUpdate(weekdays=weekdays,
weeks=weeks,
hours=hours,
price_hour_standard=price_hour_standard,
price_hour_extra=price_hour_extra,
price_fees=price_fees,
price_meals=price_meals,
start=start,
end=end)
contract2 = crud.contract.update(db=db,
db_obj=contract,
obj_in=contract_update)
assert contract.id == contract2.id
assert contract.user_id == contract2.user_id
assert contract.nanny_id == contract2.nanny_id
assert contract.weekdays == contract2.weekdays
assert contract.weeks == contract2.weeks
assert contract.hours == contract2.hours
assert contract.price_hour_standard == contract2.price_hour_standard
assert contract.price_hour_extra == contract2.price_hour_extra
assert contract.price_fees == contract2.price_fees
assert contract.price_meals == contract2.price_meals
assert contract.start == contract2.start
assert contract.end == contract2.end
assert isinstance(contract2.created_on, datetime)
assert isinstance(contract2.updated_on, datetime)
def test_get_contract_by_nanny(db: Session) -> None:
user = create_random_user(db)
nanny = create_random_user(db)
date_today = date.today()
weekdays = random_weekdays()
weeks = random_int_range(20, 47)
hours = random_int_range(10, 50)
price_hour_standard = random_float_range(2.5, 4, 1)
price_hour_extra = random_float_range(2.5, 4, 1)
price_fees = random_float_range(3.08, 5, 2)
price_meals = random_float_range(2, 6, 1)
start = str(date_today)
end = str(date_today + relativedelta(months=+12, days=-1))
contract_in = ContractCreate(user_id=user.id,
nanny_id=nanny.id,
weekdays=weekdays,
weeks=weeks,
hours=hours,
price_hour_standard=price_hour_standard,
price_hour_extra=price_hour_extra,
price_fees=price_fees,
price_meals=price_meals,
start=start,
end=end)
contract = crud.contract.create_with_owner(db=db,
obj_in=contract_in,
user_id=user.id,
nanny_id=nanny.id)
stored_contracts = crud.contract.get_multi_by_nanny(
db=db, nanny_id=contract.nanny_id)
stored_contract = [x for x in stored_contracts if x.id == contract.id][0]
assert isinstance(stored_contracts, list)
assert stored_contract
assert contract.id == stored_contract.id
assert contract.user_id == stored_contract.user_id
assert contract.nanny_id == stored_contract.nanny_id
assert contract.weekdays == stored_contract.weekdays
assert contract.weeks == stored_contract.weeks
assert contract.hours == stored_contract.hours
assert contract.price_hour_standard == stored_contract.price_hour_standard
assert contract.price_hour_extra == stored_contract.price_hour_extra
assert contract.price_fees == stored_contract.price_fees
assert contract.price_meals == stored_contract.price_meals
assert contract.start == stored_contract.start
assert contract.end == stored_contract.end
def create_random_contract(
db: Session,
user_id: int = None,
) -> models.Contract:
if not user_id:
user = create_random_user(db)
user_id = user.id
today_date = date.today()
first_day_previous_month_date = datetime.strptime(str(today_date)[:7]+"-01", "%Y-%m-%d").date() + relativedelta(months=-1)
start = str(first_day_previous_month_date)
end = str(first_day_previous_month_date + relativedelta(months=+12, days=-1))
contract_in = ContractCreate(
start=start, end=end)
return crud.contract.create_with_owner(
db=db, obj_in=contract_in, user_id=user_id)
def test_delete_contract(db: Session) -> None:
user = create_random_user(db)
date_today = date.today()
start = str(date_today)
end = str(date_today + relativedelta(months=+12, days=-1))
contract_in = ContractCreate(user_id=user.id, start=start, end=end)
contract = crud.contract.create_with_owner(db=db,
obj_in=contract_in,
user_id=user.id)
contract2 = crud.contract.remove(db=db, id=contract.id)
contract3 = crud.contract.get(db=db, id=contract.id)
assert contract3 is None
assert contract.id == contract.id
assert contract2.user_id == contract.user_id
assert contract2.user.id == contract.user.id
assert contract2.user.firstname == contract.user.firstname
assert contract2.user.email == contract.user.email
assert contract2.start == contract.start
assert contract2.end == contract.end
def test_get_contract_by_user(db: Session) -> None:
user = create_random_user(db)
date_today = date.today()
start = str(date_today)
end = str(date_today + relativedelta(months=+12, days=-1))
contract_in = ContractCreate(user_id=user.id, start=start, end=end)
contract = crud.contract.create_with_owner(db=db,
obj_in=contract_in,
user_id=user.id)
stored_contracts = crud.contract.get_multi_by_user(
db=db, user_id=contract.user_id)
stored_contract = [x for x in stored_contracts if x.id == contract.id][0]
assert isinstance(stored_contracts, list)
assert stored_contract
assert contract.id == stored_contract.id
assert contract.user_id == stored_contract.user_id
assert contract.user.id == stored_contract.user.id
assert contract.user.firstname == stored_contract.user.firstname
assert contract.user.email == stored_contract.user.email
assert contract.start == stored_contract.start
assert contract.end == stored_contract.end
def test_get_contract(db: Session) -> None:
user = create_random_user(db)
date_today = date.today()
start = str(date_today)
end = str(date_today + relativedelta(months=+12, days=-1))
contract_in = ContractCreate(user_id=user.id, start=start, end=end)
contract = crud.contract.create_with_owner(db=db,
obj_in=contract_in,
user_id=user.id)
stored_contract = crud.contract.get(db=db, id=contract.id)
assert stored_contract
assert contract.id == stored_contract.id
assert contract.user_id == stored_contract.user_id
assert contract.user.id == stored_contract.user.id
assert contract.user.firstname == stored_contract.user.firstname
assert contract.user.email == stored_contract.user.email
assert contract.start == stored_contract.start
assert contract.end == stored_contract.end
assert isinstance(stored_contract.created_on, datetime)
assert stored_contract.updated_on == None
def create_random_contract(
db: Session,
user_id: int = None,
nanny_id: int = None,
has_nanny: bool = True
) -> models.Contract:
if not user_id:
user = create_random_user(db)
user_id = user.id
if not has_nanny:
nanny_id = None
if has_nanny and not nanny_id:
nanny = create_random_user(db)
nanny_id = nanny.id
today_date = date.today()
first_day_previous_month_date = datetime.strptime(str(today_date)[:7]+"-01", "%Y-%m-%d").date() + relativedelta(months=-1)
weekdays_list = ["Mon", "Tue", "Wed", "Thu", "Fri"]
weekdays = " ".join(random.choices(weekdays_list, k=random_int_range(1, 5)))
weeks = random_int_range(20, 47)
hours = random_int_range(10, 50)
price_hour_standard = random_float_range(2.5, 4, 1)
price_hour_extra = random_float_range(2.5, 4, 1)
price_fees = random_float_range(3.08, 5, 2)
price_meals = random_float_range(2, 6, 1)
start = str(first_day_previous_month_date)
end = str(first_day_previous_month_date + relativedelta(months=+12, days=-1))
contract_in = ContractCreate(
weekdays=weekdays, weeks=weeks, hours=hours,
price_hour_standard=price_hour_standard, price_hour_extra=price_hour_extra,
price_fees=price_fees, price_meals=price_meals, start=start, end=end,
)
return crud.contract.create_with_owner(
db=db, obj_in=contract_in, user_id=user_id, nanny_id=nanny_id)