#!/usr/bin/python3
# -*- encoding: utf-8 -*-

import pytest
import os
import pyglet_ffmpeg2


def test_ffmpeg_unavailable():
    from pyglet.media import have_ffmpeg
    assert not have_ffmpeg()


def test_binaries_not_found(mocker, capsys):
    mocker.patch("pyglet_ffmpeg.loader._locate_binaries")
    pyglet_ffmpeg2.load_ffmpeg()
    out, err = capsys.readouterr()
    assert out == "FFmpeg binaries could not be loaded.\n"


@pytest.mark.parametrize("platform, arch, expected", [
    ("win32", 2 ** 32, "Win32"),
    ("win32", 2 ** 64, "Win64"),
    ("darwin", 2 ** 64, "MacOS"),
    ("linux", 2 ** 64, "linux_x86_64"),
    ("linux-armv7l", 2 ** 32, "RPi"),
])
def test_locate_binaries(platform, arch, expected, mocker):
    mocker.patch("sys.platform", platform)
    mocker.patch("sys.maxsize", arch)
    mocksym = mocker.patch("pyglet_ffmpeg.loader._ensure_linux_symlinks")
    mockloader = mocker.patch("pyglet.lib.loader")

    pyglet_ffmpeg2.loader._locate_binaries()

    env_var = "PATH" if platform == "win32" else "LD_LIBRARY_PATH"
    expected_path = os.path.join("pyglet-ffmpeg", "pyglet_ffmpeg", expected)
    assert expected_path in os.environ[env_var]

    if platform.startswith("linux"):
        mockloader._create_ld_so_cache.assert_called_once()
        mocksym.assert_called_once()


def test_load_ffmpeg():
    import pyglet
    pyglet_ffmpeg2.load_ffmpeg()
    assert pyglet.media.have_ffmpeg()


def test_ensure_linux_symlinks(mocker):
    def glob_side_effet(filename):
        "Returns a fake so filename based on the argument"
        return [filename[:-1] + "1.2"]

    mocker.patch("glob.glob", side_effect=glob_side_effet)
    mocker.patch("os.path.isfile", return_value=False)
    mocksym = mocker.patch("os.symlink")

    bin_folder = os.path.normpath("path/to/lib")
    pyglet_ffmpeg2.loader._ensure_linux_symlinks(bin_folder)

    sofiles = (
        "libavcodec.so.58.1.2", "libavformat.so.58.1.2", "libswresample.so.3.1.2",
        "libavfilter.so.7.1.2", "libavutil.so.56.1.2", "libswscale.so.5.1.2"
    )
    expected = list()
    for sofile in sofiles:
        parts = sofile.split(".")
        for length in (3, 2):
            link = ".".join(parts[:length])
            call = mocker.call(
                os.path.join(bin_folder, sofile),
                os.path.join(bin_folder, link)
            )
            expected.append(call)

    assert mocksym.call_args_list == expected