49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import datetime
|
|
import os
|
|
import shutil
|
|
from queue import Queue
|
|
|
|
import pytest
|
|
from airtime_analyzer.analyzer_pipeline import AnalyzerPipeline
|
|
|
|
from .conftest import AUDIO_FILENAME, AUDIO_IMPORT_DEST
|
|
|
|
|
|
def test_basic(src_dir, dest_dir):
|
|
queue = Queue()
|
|
AnalyzerPipeline.run_analysis(
|
|
queue,
|
|
os.path.join(src_dir, AUDIO_FILENAME),
|
|
dest_dir,
|
|
AUDIO_FILENAME,
|
|
"file",
|
|
"",
|
|
)
|
|
metadata = queue.get()
|
|
|
|
assert metadata["track_title"] == "Test Title"
|
|
assert metadata["artist_name"] == "Test Artist"
|
|
assert metadata["album_title"] == "Test Album"
|
|
assert metadata["year"] == "1999"
|
|
assert metadata["genre"] == "Test Genre"
|
|
assert metadata["mime"] == "audio/mp3"
|
|
assert abs(metadata["length_seconds"] - 3.9) < 0.1
|
|
assert metadata["length"] == str(
|
|
datetime.timedelta(seconds=metadata["length_seconds"])
|
|
)
|
|
assert os.path.exists(os.path.join(dest_dir, AUDIO_IMPORT_DEST))
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"params,exception",
|
|
[
|
|
((Queue(), u"", u"", u""), TypeError),
|
|
((Queue(), "", u"", u""), TypeError),
|
|
((Queue(), u"", "", u""), TypeError),
|
|
((Queue(), u"", u"", ""), TypeError),
|
|
],
|
|
)
|
|
def test_run_analysis_wrong_params(params, exception):
|
|
with pytest.raises(exception):
|
|
AnalyzerPipeline.run_analysis(*params)
|