diff --git a/app/schemas.py b/app/schemas.py index 8c6e8d6..941262f 100644 --- a/app/schemas.py +++ b/app/schemas.py @@ -1,5 +1,5 @@ -from pydantic import BaseModel, EmailStr from fastapi import Query +from pydantic import BaseModel, EmailStr class RegisterSchema(BaseModel): diff --git a/database/__init__.py b/database/__init__.py index 6d11c8b..baf448a 100644 --- a/database/__init__.py +++ b/database/__init__.py @@ -1,14 +1,13 @@ -from constants import DB +from databases import Database from sqlalchemy import MetaData, create_engine from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker -from databases import Database + +from constants import DB metadata = MetaData() -engine = create_engine(DB) +engine = create_engine(DB, connect_args={"check_same_thread": False}) metadata.create_all(engine) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) Base = declarative_base() database = Database(DB) - -from database import models diff --git a/database/crud.py b/database/crud.py index 9abf618..021b447 100644 --- a/database/crud.py +++ b/database/crud.py @@ -1,9 +1,9 @@ from datetime import datetime +from pydoc import locate from app import app from app.schemas import * from database import database -from pydoc import locate @app.on_event("startup") diff --git a/shell.nix b/shell.nix index 788b57a..5ea08fa 100644 --- a/shell.nix +++ b/shell.nix @@ -17,9 +17,8 @@ mkShell { python38Packages.twilio sqlite # Development tools - python38Packages.black + black python38Packages.isort python38Packages.pyflakes - python-language-server ]; } diff --git a/tests/queries_test.py b/tests/queries_test.py index 7b9505f..a124804 100644 --- a/tests/queries_test.py +++ b/tests/queries_test.py @@ -1,74 +1,72 @@ -from database.models import * from datetime import datetime +from database import SessionLocal +from database.models import * + +db = SessionLocal() + def test_users(): - deepak = Users.query.filter_by(id=46).first() - assert deepak.email == "deepak@yopmail.com" + test = db.query(Users).filter(Users.id == 260).first() + assert test.email == "testaccountjugador@gmail.com" def test_cities(): - beirut = Cities.query.filter_by(id=12).first() - assert beirut.name == "Beirut" + test = db.query(Cities).filter(Cities.id == 7).first() + assert test.name == "Lyon" def test_games(): - test = Games.query.filter_by(id=63).first() - assert test.user_id == 255 + test = db.query(Games).filter(Games.id == 508).first() + assert test.user_id == 1987 def test_payments(): - test = Payments.query.filter_by(id=295).first() - assert test.user_id == 260 + test = db.query(Payments).filter(Payments.id == 375).first() + assert test.user_id == 88 def test_player_availabilities(): - test = PlayerAvailabilities.query.filter_by(id=256).first() - assert test.game_id == 94 + test = db.query(PlayerAvailabilities).filter(PlayerAvailabilities.id == 248).first() + assert test.game_id == 89 def test_player_cancel_games(): - test = PlayerCancelGames.query.filter_by(id=132).first() - assert test.player_id == 137 + test = db.query(PlayerCancelGames).filter(PlayerCancelGames.id == 67).first() + assert test.player_id == 44 def test_purchase_games(): - test = PurchaseGames.query.filter_by(id=268).first() - assert test.game_id == 64 + test = db.query(PurchaseGames).filter(PurchaseGames.id == 235).first() + assert test.game_id == 50 def test_sports(): - test = Sports.query.filter_by(id=4).first() + test = db.query(Sports).filter(Sports.id == 4).first() assert test.name == "Volleyball" def test_teams(): - test = Teams.query.filter_by(id=306).first() + test = db.query(Teams).filter(Teams.id == 306).first() assert test.game_id == 64 def test_user_ratings(): - test = UserRatings.query.filter_by(id=420).first() - assert test.user_id == 21 + test = db.query(UserRatings).filter(UserRatings.id == 287).first() + assert test.user_id == 60 def test_venue_images(): - test = VenueImages.query.filter_by(id=6).first() - date = datetime(year=2019, month=7, day=28, hour=14, minute=1, second=21) - assert test.updated == date - - -def test_venues(): - test = Venues.query.filter_by(id=28).first() - assert test.name == "Santiago BernabĂ©u" - - -def test_view_news(): - test = ViewNews.query.filter_by(id=57).first() - date = datetime(year=2019, month=7, day=16, hour=11, minute=32, second=16) + test = db.query(VenueImages).filter(VenueImages.id == 3).first() + date = datetime(year=2019, month=6, day=3, hour=5, minute=43, second=22) assert test.created == date +def test_venues(): + test = db.query(Venues).filter(Venues.id == 26).first() + assert test.name == "Le tuto " + + def test_web_bookings(): - test = WebBookings.query.filter_by(id=9).first() - assert test.game == "FĂștbol " + test = db.query(WebBookings).filter(WebBookings.id == 7).first() + assert test.game == "Soccer"