import pytest

from tests import TestSearchResult
from yandex_music import Search, SearchResult


@pytest.fixture(scope='class')
def search_result(results):
    def make_search_result(param):
        return SearchResult(TestSearchResult.total, TestSearchResult.per_page,
                            TestSearchResult.order, [results[param]])

    return make_search_result


@pytest.fixture(scope='class')
def search(best, search_result):
    return Search(TestSearch.search_request_id, TestSearch.text, best,
                  search_result(3), search_result(2), search_result(4),
                  search_result(1), search_result(5), TestSearch.misspell_corrected, TestSearch.nocorrect)


class TestSearch:
    search_request_id = 'myt1-0261-c2e-msk-myt-music-st-e72-18274.gencfg-c.yandex.net-1573323135801461' \
                        '-3742331365077765411-1573323135819 '
    text = 'NCS'
    misspell_corrected = False
    nocorrect = False

    def test_expected_values(self, search, best, search_result):
        assert search.search_request_id == self.search_request_id
        assert search.text == self.text
        assert search.best == best
        assert search.albums == search_result(3)
        assert search.artists == search_result(2)
        assert search.playlists == search_result(4)
        assert search.tracks == search_result(1)
        assert search.videos == search_result(5)
        assert search.misspell_corrected == self.misspell_corrected
        assert search.nocorrect == self.nocorrect

    def test_de_json_none(self, client):
        assert Search.de_json({}, client) is None

    def test_de_json_required(self, client, best, search_result):
        json_dict = {'search_request_id': self.search_request_id, 'text': self.text, 'best': best.to_dict(),
                     'albums': search_result(3).to_dict(), 'artists': search_result(2).to_dict(),
                     'playlists': search_result(4).to_dict(), 'tracks': search_result(1).to_dict(),
                     'videos': search_result(5).to_dict()}
        search = Search.de_json(json_dict, client)

        assert search.search_request_id == self.search_request_id
        assert search.text == self.text
        assert search.best == best
        assert search.albums == search_result(3)
        assert search.artists == search_result(2)
        assert search.playlists == search_result(4)
        assert search.tracks == search_result(1)
        assert search.videos == search_result(5)

    def test_de_json_all(self, client, best, search_result):
        json_dict = {'search_request_id': self.search_request_id, 'text': self.text, 'best': best.to_dict(),
                     'albums': search_result(3).to_dict(), 'artists': search_result(2).to_dict(),
                     'playlists': search_result(4).to_dict(), 'tracks': search_result(1).to_dict(),
                     'videos': search_result(5).to_dict(), 'misspell_corrected': self.misspell_corrected,
                     'nocorrect': self.nocorrect}
        search = Search.de_json(json_dict, client)

        assert search.search_request_id == self.search_request_id
        assert search.text == self.text
        assert search.best == best
        assert search.albums == search_result(3)
        assert search.artists == search_result(2)
        assert search.playlists == search_result(4)
        assert search.tracks == search_result(1)
        assert search.videos == search_result(5)
        assert search.misspell_corrected == self.misspell_corrected
        assert search.nocorrect == self.nocorrect

    def test_equality(self, best, search_result):
        a = Search(self.search_request_id, self.text, best, search_result(3), search_result(2), search_result(4),
                   search_result(1), search_result(5))
        b = Search(self.search_request_id, '', best, search_result(3), None, search_result(4), search_result(1),
                   search_result(5))
        c = Search(self.search_request_id, self.text, best, search_result(3), search_result(2), search_result(4),
                   search_result(1), search_result(5))

        assert a != b
        assert hash(a) != hash(b)
        assert a is not b

        assert a == c