added activity view
This commit is contained in:
88
FitnessSync/backend/tests/functional/test_bike_setups.py
Normal file
88
FitnessSync/backend/tests/functional/test_bike_setups.py
Normal file
@@ -0,0 +1,88 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock
|
||||
import sys
|
||||
|
||||
# Mock scheduler before importing main to prevent it from starting
|
||||
mock_scheduler = MagicMock()
|
||||
mock_scheduler_module = MagicMock()
|
||||
mock_scheduler_module.scheduler = mock_scheduler
|
||||
sys.modules["src.services.scheduler"] = mock_scheduler_module
|
||||
|
||||
from fastapi.testclient import TestClient
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from src.models import Base, BikeSetup
|
||||
from main import app
|
||||
from src.utils.config import config
|
||||
from src.api.bike_setups import get_db
|
||||
|
||||
# Use a separate test database or the existing test.db
|
||||
SQLALCHEMY_DATABASE_URL = "sqlite:///./test_bike_setups.db"
|
||||
|
||||
engine = create_engine(
|
||||
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
|
||||
)
|
||||
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
|
||||
def override_get_db():
|
||||
try:
|
||||
db = TestingSessionLocal()
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
app.dependency_overrides[get_db] = override_get_db
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def test_db():
|
||||
Base.metadata.create_all(bind=engine)
|
||||
yield
|
||||
Base.metadata.drop_all(bind=engine)
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def client(test_db):
|
||||
with TestClient(app) as c:
|
||||
yield c
|
||||
|
||||
def test_create_bike_setup(client):
|
||||
response = client.post(
|
||||
"/api/bike-setups/",
|
||||
json={"frame": "Trek Emonda", "chainring": 50, "rear_cog": 11, "name": "Road Setup"}
|
||||
)
|
||||
assert response.status_code == 201
|
||||
data = response.json()
|
||||
assert data["frame"] == "Trek Emonda"
|
||||
assert data["chainring"] == 50
|
||||
assert "id" in data
|
||||
|
||||
def test_read_bike_setups(client):
|
||||
response = client.get("/api/bike-setups/")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert len(data) >= 1
|
||||
assert data[0]["frame"] == "Trek Emonda"
|
||||
|
||||
def test_update_bike_setup(client):
|
||||
# First get id
|
||||
response = client.get("/api/bike-setups/")
|
||||
setup_id = response.json()[0]["id"]
|
||||
|
||||
response = client.put(
|
||||
f"/api/bike-setups/{setup_id}",
|
||||
json={"chainring": 52}
|
||||
)
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["chainring"] == 52
|
||||
assert data["frame"] == "Trek Emonda"
|
||||
|
||||
def test_delete_bike_setup(client):
|
||||
# First get id
|
||||
response = client.get("/api/bike-setups/")
|
||||
setup_id = response.json()[0]["id"]
|
||||
|
||||
response = client.delete(f"/api/bike-setups/{setup_id}")
|
||||
assert response.status_code == 204
|
||||
|
||||
response = client.get(f"/api/bike-setups/{setup_id}")
|
||||
assert response.status_code == 404
|
||||
Reference in New Issue
Block a user