91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
import pytest
|
|
import os
|
|
|
|
from pathlib import Path
|
|
from dotenv import load_dotenv
|
|
|
|
import groove.db
|
|
from groove.playlist import Playlist
|
|
|
|
from sqlalchemy import create_engine, insert
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
|
|
@pytest.fixture(autouse=True, scope='function')
|
|
def env(monkeypatch):
|
|
root = Path(__file__).parent / Path('fixtures')
|
|
monkeypatch.setattr('groove.path.root', MagicMock(return_value=str(root)))
|
|
load_dotenv(Path('test/fixtures/env'))
|
|
os.environ['MEDIA_ROOT'] = str(root / Path('media'))
|
|
return os.environ
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def auth():
|
|
return (os.environ.get('USERNAME'), os.environ.get('PASSWORD'))
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def in_memory_engine(monkeypatch):
|
|
engine = create_engine('sqlite:///:memory:', future=True)
|
|
monkeypatch.setattr('groove.db.manager.create_engine',
|
|
MagicMock(return_value=engine))
|
|
return engine
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def in_memory_db(in_memory_engine):
|
|
"""
|
|
An (empty) in-memory SQLite3 database
|
|
"""
|
|
Session = sessionmaker(bind=in_memory_engine, future=True)
|
|
session = Session()
|
|
groove.db.metadata.create_all(bind=in_memory_engine)
|
|
yield session
|
|
session.close()
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def db(in_memory_db):
|
|
"""
|
|
Populate the in-memory sqlite database with fixture data.
|
|
"""
|
|
|
|
# create tracks
|
|
query = insert(groove.db.track)
|
|
in_memory_db.execute(query, [
|
|
{'id': 1, 'artist': 'UNKLE', 'title': 'Guns Blazing', 'relpath': 'UNKLE/Psyence Fiction/01 Guns Blazing (Drums of Death, Part 1).flac'},
|
|
{'id': 2, 'artist': 'UNKLE', 'title': 'UNKLE', 'relpath': 'UNKLE/Psyence Fiction/02 UNKLE (Main Title Theme).flac'},
|
|
{'id': 3, 'artist': 'UNKLE', 'title': 'Bloodstain', 'relpath': 'UNKLE/Psyence Fiction/03 Bloodstain.flac'}
|
|
])
|
|
|
|
# create playlists
|
|
query = insert(groove.db.playlist)
|
|
in_memory_db.execute(query, [
|
|
{'id': 1, 'name': 'playlist one', 'description': 'the first one', 'slug': 'playlist-one'},
|
|
{'id': 2, 'name': 'playlist two', 'description': 'the second one', 'slug': 'playlist-two'},
|
|
{'id': 3, 'name': 'playlist three', 'description': 'the threerd one', 'slug': 'playlist-three'},
|
|
{'id': 4, 'name': 'empty playlist', 'description': 'no tracks', 'slug': 'empty-playlist'}
|
|
])
|
|
|
|
# populate the playlists
|
|
query = insert(groove.db.entry)
|
|
in_memory_db.execute(query, [
|
|
{'playlist_id': '1', 'track': '1', 'track_id': '1'},
|
|
{'playlist_id': '1', 'track': '2', 'track_id': '2'},
|
|
{'playlist_id': '1', 'track': '3', 'track_id': '3'},
|
|
|
|
{'playlist_id': '2', 'track': '1', 'track_id': '1'},
|
|
|
|
{'playlist_id': '3', 'track': '6', 'track_id': '2'},
|
|
{'playlist_id': '3', 'track': '2', 'track_id': '3'},
|
|
])
|
|
yield in_memory_db
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def empty_playlist(db):
|
|
return Playlist.by_slug('empty-playlist', session=db)
|