feat(main): initial commit
This commit is contained in:
0
api/tests/__init__.py
Normal file
0
api/tests/__init__.py
Normal file
67
api/tests/test_channel.py
Normal file
67
api/tests/test_channel.py
Normal file
@@ -0,0 +1,67 @@
|
||||
import pytest
|
||||
from django.test import Client
|
||||
from core.models import AppUser, Library, Channel
|
||||
|
||||
@pytest.fixture
|
||||
def client():
|
||||
return Client()
|
||||
|
||||
@pytest.fixture
|
||||
def user(db):
|
||||
return AppUser.objects.create_user(username="apiuser2", email="api2@example.com", password="password")
|
||||
|
||||
@pytest.fixture
|
||||
def library(db, user):
|
||||
return Library.objects.create(owner_user=user, name="Network Lib")
|
||||
|
||||
@pytest.fixture
|
||||
def channel(db, user, library):
|
||||
return Channel.objects.create(
|
||||
owner_user=user,
|
||||
library=library,
|
||||
name="Test Channel",
|
||||
slug="test-ch",
|
||||
channel_number=1,
|
||||
scheduling_mode="template_driven"
|
||||
)
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_list_channels(client, channel):
|
||||
response = client.get("/api/channel/")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert len(data) == 1
|
||||
assert data[0]["name"] == "Test Channel"
|
||||
assert data[0]["slug"] == "test-ch"
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_get_channel(client, channel):
|
||||
response = client.get(f"/api/channel/{channel.id}")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["name"] == "Test Channel"
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_get_channel_not_found(client):
|
||||
response = client.get("/api/channel/999")
|
||||
assert response.status_code == 404
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_create_channel(client, user, library):
|
||||
payload = {
|
||||
"name": "New API Channel",
|
||||
"slug": "new-api-ch",
|
||||
"channel_number": 5,
|
||||
"description": "Created via API",
|
||||
"library_id": library.id,
|
||||
"owner_user_id": user.id
|
||||
}
|
||||
response = client.post("/api/channel/", data=payload, content_type="application/json")
|
||||
assert response.status_code == 201
|
||||
data = response.json()
|
||||
assert data["name"] == "New API Channel"
|
||||
assert data["slug"] == "new-api-ch"
|
||||
|
||||
# Verify it hit the DB
|
||||
assert Channel.objects.count() == 1
|
||||
assert Channel.objects.get(id=data["id"]).name == "New API Channel"
|
||||
54
api/tests/test_library.py
Normal file
54
api/tests/test_library.py
Normal file
@@ -0,0 +1,54 @@
|
||||
import pytest
|
||||
from django.test import Client
|
||||
from core.models import AppUser, Library
|
||||
|
||||
@pytest.fixture
|
||||
def client():
|
||||
return Client()
|
||||
|
||||
@pytest.fixture
|
||||
def user(db):
|
||||
return AppUser.objects.create_user(username="apiuser", email="api@example.com", password="password")
|
||||
|
||||
@pytest.fixture
|
||||
def library(db, user):
|
||||
return Library.objects.create(owner_user=user, name="Test Library", visibility="private")
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_list_libraries(client, library):
|
||||
response = client.get("/api/library/")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert len(data) == 1
|
||||
assert data[0]["name"] == "Test Library"
|
||||
assert data[0]["owner_user_id"] == library.owner_user.id
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_get_library(client, library):
|
||||
response = client.get(f"/api/library/{library.id}")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["name"] == "Test Library"
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_get_library_not_found(client):
|
||||
response = client.get("/api/library/999")
|
||||
assert response.status_code == 404
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_create_library(client, user):
|
||||
payload = {
|
||||
"name": "New API Library",
|
||||
"description": "Created via API",
|
||||
"visibility": "public",
|
||||
"owner_user_id": user.id
|
||||
}
|
||||
response = client.post("/api/library/", data=payload, content_type="application/json")
|
||||
assert response.status_code == 201
|
||||
data = response.json()
|
||||
assert data["name"] == "New API Library"
|
||||
assert data["visibility"] == "public"
|
||||
|
||||
# Verify it hit the DB
|
||||
assert Library.objects.count() == 1
|
||||
assert Library.objects.get(id=data["id"]).name == "New API Library"
|
||||
106
api/tests/test_schedule.py
Normal file
106
api/tests/test_schedule.py
Normal file
@@ -0,0 +1,106 @@
|
||||
import pytest
|
||||
from django.test import Client
|
||||
from core.models import AppUser, Library, Channel, ScheduleTemplate
|
||||
|
||||
@pytest.fixture
|
||||
def client():
|
||||
return Client()
|
||||
|
||||
@pytest.fixture
|
||||
def user(db):
|
||||
return AppUser.objects.create_user(username="apiuser3", email="api3@example.com", password="password")
|
||||
|
||||
@pytest.fixture
|
||||
def library(db, user):
|
||||
return Library.objects.create(owner_user=user, name="Schedule Lib")
|
||||
|
||||
@pytest.fixture
|
||||
def channel(db, user, library):
|
||||
return Channel.objects.create(
|
||||
owner_user=user,
|
||||
library=library,
|
||||
name="Schedule Channel",
|
||||
slug="sch-ch",
|
||||
channel_number=2
|
||||
)
|
||||
|
||||
@pytest.fixture
|
||||
def schedule_template(db, channel):
|
||||
return ScheduleTemplate.objects.create(
|
||||
channel=channel,
|
||||
name="Morning Block",
|
||||
timezone_name="UTC",
|
||||
priority=10
|
||||
)
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_list_schedule_templates(client, schedule_template):
|
||||
response = client.get("/api/schedule/template/")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert len(data) == 1
|
||||
assert data[0]["name"] == "Morning Block"
|
||||
assert data[0]["priority"] == 10
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_get_schedule_template(client, schedule_template):
|
||||
response = client.get(f"/api/schedule/template/{schedule_template.id}")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["name"] == "Morning Block"
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_get_schedule_template_not_found(client):
|
||||
response = client.get("/api/schedule/template/999")
|
||||
assert response.status_code == 404
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_create_schedule_template(client, channel):
|
||||
payload = {
|
||||
"name": "Evening Block",
|
||||
"description": "Late night programming",
|
||||
"timezone_name": "America/New_York",
|
||||
"priority": 5,
|
||||
"is_active": True,
|
||||
"channel_id": channel.id
|
||||
}
|
||||
response = client.post("/api/schedule/template/", data=payload, content_type="application/json")
|
||||
assert response.status_code == 201
|
||||
data = response.json()
|
||||
assert data["name"] == "Evening Block"
|
||||
assert data["timezone_name"] == "America/New_York"
|
||||
|
||||
assert ScheduleTemplate.objects.count() == 1
|
||||
assert ScheduleTemplate.objects.get(id=data["id"]).name == "Evening Block"
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_generate_schedule(client, channel, schedule_template):
|
||||
# Setup some media for it to find
|
||||
from core.models import MediaSource, MediaItem
|
||||
source = MediaSource.objects.create(library=channel.library, name="G", uri="/tmp", source_type="local_directory")
|
||||
MediaItem.objects.create(media_source=source, title="M1", item_kind="movie", runtime_seconds=3600, file_path="1")
|
||||
MediaItem.objects.create(media_source=source, title="M2", item_kind="movie", runtime_seconds=3600, file_path="2")
|
||||
|
||||
# Add a schedule block to the template (created in fixture)
|
||||
from core.models import ScheduleBlock
|
||||
from datetime import time
|
||||
t = schedule_template
|
||||
ScheduleBlock.objects.create(
|
||||
schedule_template=t,
|
||||
name="TestBlock",
|
||||
block_type="programming",
|
||||
start_local_time=time(0, 0),
|
||||
end_local_time=time(23, 59),
|
||||
day_of_week_mask=127 # All days
|
||||
)
|
||||
|
||||
payload = {"target_date": "2026-03-08"}
|
||||
response = client.post(f"/api/schedule/generate/{channel.id}", data=payload, content_type="application/json")
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["status"] == "success"
|
||||
# Should have scheduled both movies back to back until the block ends
|
||||
assert data["airings_created"] > 0
|
||||
from core.models import Airing
|
||||
assert Airing.objects.filter(channel=channel).count() == data["airings_created"]
|
||||
67
api/tests/test_user.py
Normal file
67
api/tests/test_user.py
Normal file
@@ -0,0 +1,67 @@
|
||||
import pytest
|
||||
from django.test import Client
|
||||
from core.models import AppUser
|
||||
|
||||
@pytest.fixture
|
||||
def api_client():
|
||||
return Client()
|
||||
|
||||
@pytest.fixture
|
||||
def test_user():
|
||||
return AppUser.objects.create_user(
|
||||
username="test_api_user",
|
||||
email="test@api.tv",
|
||||
password="testpassword123"
|
||||
)
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_list_users(api_client, test_user):
|
||||
response = api_client.get("/api/user/")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert len(data) >= 1
|
||||
assert any(u["username"] == "test_api_user" for u in data)
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_get_user(api_client, test_user):
|
||||
response = api_client.get(f"/api/user/{test_user.id}")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["username"] == "test_api_user"
|
||||
assert "password" not in data # Ensure we don't leak passwords
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_create_user(api_client):
|
||||
payload = {
|
||||
"username": "new_viewer",
|
||||
"email": "viewer@api.tv",
|
||||
"password": "securepassword",
|
||||
"is_superuser": False
|
||||
}
|
||||
response = api_client.post("/api/user/", data=payload, content_type="application/json")
|
||||
assert response.status_code == 201
|
||||
data = response.json()
|
||||
assert data["username"] == "new_viewer"
|
||||
|
||||
# Verify DB
|
||||
user = AppUser.objects.get(id=data["id"])
|
||||
assert user.username == "new_viewer"
|
||||
assert not user.is_superuser
|
||||
assert user.check_password("securepassword")
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_update_user(api_client, test_user):
|
||||
payload = {
|
||||
"email": "updated@api.tv",
|
||||
"is_active": False
|
||||
}
|
||||
response = api_client.patch(f"/api/user/{test_user.id}", data=payload, content_type="application/json")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["email"] == "updated@api.tv"
|
||||
assert data["is_active"] is False
|
||||
|
||||
# Verify DB
|
||||
test_user.refresh_from_db()
|
||||
assert test_user.email == "updated@api.tv"
|
||||
assert not test_user.is_active
|
||||
Reference in New Issue
Block a user