# -*- coding: utf-8 -*-

from __future__ import absolute_import, division

from six.moves.urllib.request import urlretrieve

import os
import os.path

import pytest

from MDAnalysisData import base
from MDAnalysisData import adk_equilibrium

@pytest.fixture(scope="module")
def some_text():
    return (u"Four score and seven years ago \n"
            u"our fathers brought forth on this continent, \n"
            u"a new nation, conceived in Liberty, and dedicated\n"
            u"to the proposition that all men are created equal.")

@pytest.fixture
def bunch():
    return base.Bunch(a=1, b=2)

class TestBunch(object):
    def test_getattr(self, bunch):
        assert bunch.a == 1
        assert bunch['a'] == bunch.a
        assert bunch.b == 2

    def test_missing_attr(self, bunch):
        with pytest.raises(AttributeError):
            bunch.does_not_exist

    @pytest.mark.parametrize("key,value", [
        ("a", "fortytwo"),
        ("c", -1),
        ("_hidden", None),
        ])
    def test_setattr(self, bunch, key, value):
        setattr(bunch, key, value)
        assert bunch[key] == value

    def test_dir(self, bunch):
        assert dir(bunch) == list(bunch.keys())


def test_read_description():
    descr = base._read_description("adk_equilibrium.rst")
    assert len(descr) == 1252
    assert descr.startswith(".. -*- coding: utf-8 -*-\n\n.. _`adk-equilibrium-dataset`:")

def test_sha256(tmpdir, some_text):
    filename = "address.txt"
    with tmpdir.as_cwd():
        with open(filename, "w") as txt:
            txt.write(some_text)
        checksum = base._sha256(filename)
    assert checksum == "4446bfb2ec5dedfbd981d059d6005f5144b067b392a00e3bcf98f8302ec8f765"

@pytest.mark.parametrize('data_home,location', [
    (None, os.path.expanduser("~/MDAnalysis_data")),
    ("/tmp/MDAnalysisData", "/tmp/MDAnalysisData"),
    ])
def test_get_data_home(data_home, location):
    assert base.get_data_home(data_home=data_home) == location

def test_clear_data_home(tmpdir, some_text):
    data_home_path = tmpdir.join("MDAnalysis_data_test")
    data_home = base.get_data_home(data_home=str(data_home_path))
    textfile = data_home_path.join("address.txt")
    textfile.write_text(some_text, encoding="utf-8")

    assert data_home_path.exists()  # get_data_home() should have created it
    assert data_home == str(data_home_path)
    assert textfile.exists()

    base.clear_data_home(data_home=data_home)
    assert not data_home_path.exists()  # should have been deleted
    assert not textfile.exists()        # the directory should be gone...


@pytest.fixture
def remote_topology():
    # small-ish file
    return adk_equilibrium.ARCHIVE['topology']


# Run this test *always* (even though it is an 'online' test) so
# that we run at least one real fetch. (The file is small so it
# will not take long.)
# @pytest.mark.online
def test_fetch_remote(remote_topology, tmpdir):
    filename = base._fetch_remote(remote_topology, dirname=str(tmpdir))
    assert os.path.basename(filename) == remote_topology.filename
    assert os.path.dirname(filename) == str(tmpdir)


def test_fetch_remote_sha_fail(remote_topology, mocker):
    # Mock a SHA fail and check error message
    mocker.patch('MDAnalysisData.base.urlretrieve')
    mock_sha = mocker.patch('MDAnalysisData.base._sha256',
                            return_value='12345678')

    with pytest.raises(IOError,
                       match='.+? file may be corrupted'):
        base._fetch_remote(remote_topology)


@pytest.mark.parametrize('dirname', ['', 'thisplace'])
def test_fetch_remote_sha_success(remote_topology, mocker, dirname):
    # Mocked offline successful remote
    mocker.patch('MDAnalysisData.base.urlretrieve')
    # Mock a SHA success
    mocker.patch('MDAnalysisData.base._sha256',
                 return_value=remote_topology.checksum)
    exp = os.path.join(dirname, remote_topology.filename)
    assert base._fetch_remote(remote_topology, dirname=dirname) == exp

def test_lazy_fetch(tmpdir, mocker):
    mocker.patch('MDAnalysisData.adk_equilibrium.exists', return_value=True)
    fr = mocker.patch('MDAnalysisData.adk_equilibrium._fetch_remote')
    # check the laziness of grabbing a dataset
    # - mock exists to always say true
    # - grab the "dataset" then check no remote calls were done
    stuff = adk_equilibrium.fetch_adk_equilibrium()
    assert not fr.called


def test_tqdm(remote_topology):
    # not sure how to test apart from not getting errors
    remote = remote_topology
    with base.TqdmUpTo(unit='B', unit_scale=True, miniters=1,
                       desc=remote.filename) as t:
        urlretrieve(remote.url, filename=os.devnull,
                    reporthook=t.update_to, data=None)
