import os
from io import StringIO, BytesIO

import pytest
import pyglet

from pyglet.model import ModelDecodeException
from ...annotations import require_python_version


test_data_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'data'))


def get_test_data_file(path, file_name):
    """Get a file from the test data directory in an OS independent way.

    Supply relative file name as you would in os.path.join().
    """
    return os.path.join(test_data_path, path, file_name)


def test_load_from_disk():
    file_path = get_test_data_file('models', 'logo3d.obj')
    model = pyglet.model.load(file_path)
    assert isinstance(model, pyglet.model.Model)


@require_python_version((3, 4))
def test_load_from_object_str():
    file_path = get_test_data_file('models', 'logo3d.obj')
    with open(file_path, 'r') as f:
        file_obj = StringIO(f.read())
    model = pyglet.model.load(file_path, file=file_obj)
    assert isinstance(model, pyglet.model.Model)


def test_load_from_object_bytes():
    file_path = get_test_data_file('models', 'logo3d.obj')
    with open(file_path, 'rb') as f:
        file_obj = BytesIO(f.read())
    model = pyglet.model.load(file_path, file=file_obj)
    assert isinstance(model, pyglet.model.Model)


def test_no_decoders_available():
    # This is NOT a valid model file:
    file_path = get_test_data_file('media', 'alert.wav')

    with pytest.raises(ModelDecodeException) as e:
        model = pyglet.model.load(file_path)


def test_resource_module():
    folder_path = os.path.join(test_data_path, 'models')
    pyglet.resource.path.append(folder_path)
    pyglet.resource.reindex()

    model = pyglet.resource.model('logo3d.obj')
    assert isinstance(model, pyglet.model.Model)
