sonosify/mp3ify_test.py at main · simojenki/sonosify · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import pytest
import os
from uuid import uuid4 as uuid

from test_utils import docker, run, AudioFile


MP3IFY = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'mp3ify')
JSON_SH = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'JSON.sh')


def mp3ify(i):
    new_file_name = "{name}.mp3".format(name = uuid())
    print(run("""docker run \
        --rm \
        -t \
        -v {sounds}:/sounds \
        -v {mp3ify}:/bin/mp3ify \
        -u {uid}:{gid} \
        -e MP3IFY_LOG=true \
        --entrypoint /bin/mp3ify \
        deluan/navidrome:latest \
            /sounds/{i} \
            /sounds/{o}""".format(
                sounds=i.path.dirname,
                mp3ify=MP3IFY,
                uid=os.getuid(),
                gid=os.getgid(),
                i=i.path.basename,
                o=new_file_name
            )))
    return AudioFile(i.path.dirpath(new_file_name))


@pytest.mark.parametrize("input_type",
    [
        ("flac"),
        ("alac"),
        ("aac")
    ]
)
def test_non_mp3_are_transcoded_to_mp3_removing_tags(wav, input_type):
    if input_type == "flac":
        input = wav.to_flac()
    elif input_type == "alac":
        input = wav.to_alac()
    elif input_type == "aac":
        input = wav.to_aac()

    input_with_tags = input.with_tags(
        artist="a1",
        title="t1",
        genre="g1",
        album="b1"
    )

    original_tags = input_with_tags.tags()
    assert original_tags["artist"] == "a1"
    assert original_tags["title"] == "t1"
    assert original_tags["genre"] == "g1"
    assert original_tags["album"] == "b1"

    mp3 = mp3ify(input_with_tags)

    result_stream0 = mp3.stream0()

    assert result_stream0["codec_name"] == "mp3"
    assert result_stream0["sample_fmt"] == "fltp"
    assert result_stream0["sample_rate"] == "44100"

    assert list(mp3.tags().keys()) == ["encoder"]


def test_mp3_file_should_have_tags_removed(wav):
    mp3 = wav.to_mp3()

    with_tags = mp3.with_tags(
        artist="sonosify-artist",
        title="sonosify-title",
        genre="sonosify-genre",
        album="sonosify-album"
    )

    original_tags = with_tags.tags()
    assert original_tags["artist"] == "sonosify-artist"
    assert original_tags["title"] == "sonosify-title"
    assert original_tags["genre"] == "sonosify-genre"
    assert original_tags["album"] == "sonosify-album"

    result = mp3ify(with_tags)

    assert list(result.tags().keys()) == ["encoder"]


def test_mp3_file_should_have_same_audio_as_original_however_tags_removed(wav):
    mp3 = wav.to_mp3()
    original_md5 = mp3.stream_md5()

    with_tags = mp3.with_tags(
        artist="sonosify-artist",
        title="sonosify-title",
        genre="sonosify-genre",
        album="sonosify-album"
    )

    original_tags = with_tags.tags()
    assert original_tags["artist"] == "sonosify-artist"
    assert original_tags["title"] == "sonosify-title"
    assert original_tags["genre"] == "sonosify-genre"
    assert original_tags["album"] == "sonosify-album"

    result = mp3ify(with_tags)

    assert result.stream_md5() == original_md5

    assert list(result.tags().keys()) == ["encoder"]