fce988aef1
Fixes #1622 - split the api into 4 apps: core, history, schedule, storage - exploded the settings into testing/prod
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import os
|
|
|
|
from django.conf import settings
|
|
from model_bakery import baker
|
|
from rest_framework.test import APITestCase
|
|
|
|
from ...._fixtures import AUDIO_FILENAME, fixture_path
|
|
|
|
|
|
class TestFileViewSet(APITestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.path = "/api/v2/files/{id}/download/"
|
|
cls.token = settings.CONFIG.general.api_key
|
|
|
|
def test_invalid(self):
|
|
path = self.path.format(id="a")
|
|
self.client.credentials(HTTP_AUTHORIZATION=f"Api-Key {self.token}")
|
|
response = self.client.get(path)
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_does_not_exist(self):
|
|
path = self.path.format(id="1")
|
|
self.client.credentials(HTTP_AUTHORIZATION=f"Api-Key {self.token}")
|
|
response = self.client.get(path)
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_exists(self):
|
|
music_dir = baker.make(
|
|
"storage.MusicDir",
|
|
directory=str(fixture_path),
|
|
)
|
|
f = baker.make(
|
|
"storage.File",
|
|
directory=music_dir,
|
|
mime="audio/mp3",
|
|
filepath=AUDIO_FILENAME,
|
|
)
|
|
path = self.path.format(id=str(f.pk))
|
|
self.client.credentials(HTTP_AUTHORIZATION=f"Api-Key {self.token}")
|
|
response = self.client.get(path)
|
|
self.assertEqual(response.status_code, 200)
|