57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
import groove.db
|
||
|
|
|
||
|
|
from sqlalchemy import create_engine, insert
|
||
|
|
from sqlalchemy.orm import sessionmaker
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope='function')
|
||
|
|
def in_memory_db():
|
||
|
|
"""
|
||
|
|
An (empty) in-memory SQLite3 database
|
||
|
|
"""
|
||
|
|
engine = create_engine('sqlite:///:memory:', future=True)
|
||
|
|
Session = sessionmaker(bind=engine, future=True)
|
||
|
|
session = Session()
|
||
|
|
groove.db.metadata.create_all(bind=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, 'relpath': '/UNKLE/Psyence Fiction/01 Guns Blazing (Drums of Death, Part 1).flac'},
|
||
|
|
{'id': 2, 'relpath': '/UNKLE/Psyence Fiction/02 UNKLE (Main Title Theme).flac'},
|
||
|
|
{'id': 3, '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'}
|
||
|
|
])
|
||
|
|
|
||
|
|
# 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
|