def test_create_settings(salary_test_client: TestClient, salary_session: Session, salary_user: User) -> None: category_id = conftest.CATEGORY_ID assert utils.get_settings(salary_session, salary_user.id, category_id) is None data = { 'category_id': category_id, 'wage': utils.DEFAULT_SETTINGS.wage, 'off_day': utils.DEFAULT_SETTINGS.off_day, 'holiday_category_id': utils.DEFAULT_SETTINGS.holiday_category_id, 'regular_hour_basis': utils.DEFAULT_SETTINGS.regular_hour_basis, 'night_hour_basis': utils.DEFAULT_SETTINGS.night_hour_basis, 'night_start': utils.DEFAULT_SETTINGS.night_start, 'night_end': utils.DEFAULT_SETTINGS.night_end, 'night_min_len': utils.DEFAULT_SETTINGS.night_min_len, 'first_overtime_amount': utils.DEFAULT_SETTINGS.first_overtime_amount, 'first_overtime_pay': utils.DEFAULT_SETTINGS.first_overtime_pay, 'second_overtime_pay': utils.DEFAULT_SETTINGS.second_overtime_pay, 'week_working_hours': utils.DEFAULT_SETTINGS.week_working_hours, 'daily_transport': utils.DEFAULT_SETTINGS.daily_transport, } response = salary_test_client.post( conftest.ROUTES['new'], data=data, allow_redirects=True) assert response.ok assert conftest.MESSAGES['view_salary'] in response.text settings = utils.get_settings(salary_session, salary_user.id, category_id) assert settings delete_instance(salary_session, settings)
def salary_user(salary_session: Session): test_user = create_model( salary_session, User, username='******', password='******', email='*****@*****.**', ) yield test_user delete_instance(salary_session, test_user)
def sender(session: Session) -> Generator[User, None, None]: mock_user = create_model( session, User, username='******', password='******', email='*****@*****.**', language_id=1, ) yield mock_user delete_instance(session, mock_user)
def comment(session: Session, event: Event, user: User) -> Iterator[Comment]: data = { 'user': user, 'event': event, 'content': 'test comment', 'time': datetime(2021, 1, 1, 0, 1), } create_model(session, Comment, **data) comment = session.query(Comment).first() yield comment delete_instance(session, comment)
def add_quote(session: Session, id_quote: int, text: str, author: str) -> Quote: quote = create_model( session, Quote, id=id_quote, text=text, author=author, ) yield quote delete_instance(session, quote)
def bad_user(session: Session) -> User: test_user = create_model( session, User, username='******', password='******', email='test.email#gmail.com', language_id=1, ) yield test_user delete_instance(session, test_user)
def test_add_comment( event_test_client: TestClient, session: Session, event: Event, ) -> None: assert session.query(Comment).first() is None content = "test comment" path = evt.router.url_path_for("add_comment", event_id=event.id) data = {"comment": content} response = event_test_client.post(path, data=data, allow_redirects=True) assert response.ok assert content in response.text comment = session.query(Comment).first() assert comment delete_instance(session, comment)
def delete_comment(session: Session, comment_id: int) -> bool: """Deletes a comment instance based on `comment_id`. Returns True if successful, False otherwise. Args: session (Session): DB session. comment_id (int): ID of comment instance to delete. Returns: bool: True if successful, False otherwise. """ comment = session.query(Comment).filter_by(id=comment_id).first() if comment: utils.delete_instance(session, comment) return True return False
def wage(salary_session: Session, salary_user: User) -> Iterator[SalarySettings]: wage = create_model( salary_session, SalarySettings, user_id=salary_user.id, category_id=CATEGORY_ID, wage=30, off_day=config.SATURDAY, holiday_category_id=config.ISRAELI_JEWISH, regular_hour_basis=config.REGULAR_HOUR_BASIS, night_hour_basis=config.NIGHT_HOUR_BASIS, night_start=config.NIGHT_START, night_end=config.NIGHT_END, night_min_len=config.NIGHT_MIN_LEN, first_overtime_amount=config.FIRST_OVERTIME_AMOUNT, first_overtime_pay=config.FIRST_OVERTIME_PAY, second_overtime_pay=config.SECOND_OVERTIME_PAY, week_working_hours=config.WEEK_WORKING_HOURS, daily_transport=config.STANDARD_TRANSPORT, ) yield wage delete_instance(salary_session, wage)
def invitation( event: Event, user: User, session: Session ) -> Generator[Invitation, None, None]: """Returns an Invitation object after being created in the database. Args: event: An Event instance. user: A user instance. session: A database connection. Returns: An Invitation object. """ invitation = create_model( session, Invitation, creation=datetime.now(), recipient=user, event=event, event_id=event.id, recipient_id=user.id, ) yield invitation delete_instance(session, invitation)
def add_joke(session: Session, id_joke: int, text: str) -> Joke: joke = create_model(session, Joke, id=id_joke, text=text) yield joke delete_instance(session, joke)
def test_create_comment(session: Session, event: Event, user: User) -> None: assert session.query(Comment).first() is None cmt.create_comment(session, event, 'test content') comment = session.query(Comment).first() assert comment delete_instance(session, comment)
def test_delete_instance(self, session: Session, user: User): assert session.query(User).first() utils.delete_instance(session, user) assert session.query(User).first() is None