from galaxy.managers import hdas
from galaxy.managers.histories import HistoryManager
from galaxy.model.tags import GalaxyTagHandler
from galaxy.util import unicodify
from .base import BaseTestCase

# =============================================================================
default_password = "123456"
user2_data = dict(email="user2@user2.user2", username="user2", password=default_password)


# =============================================================================
class TestTagHandler(BaseTestCase):
    def METHOD_NAME(self):
        super().METHOD_NAME()
        self.app.hda_manager = self.app[hdas.HDAManager]
        self.app.history_manager = self.app[HistoryManager]
        self.tag_handler = self.app[GalaxyTagHandler]
        self.user = self.user_manager.create(**user2_data)

    def _create_vanilla_hda(self, user=None):
        owner = user or self.user
        history1 = self.app.history_manager.create(name="history1", user=owner)
        return self.app.hda_manager.create(history=history1)

    def _check_tag_list(self, tags, expected_tags):
        assert len(tags) == len(expected_tags)
        actual_tags = []
        for tag in tags:
            if tag.user_value:
                tag = f"{tag.user_tname}:{tag.user_value}"
            else:
                tag = tag.user_tname
            actual_tags.append(tag)
        expected = [unicodify(e) for e in expected_tags]
        assert sorted(expected) == sorted(actual_tags), f"{expected} vs {actual_tags}"

    def test_apply_item_tags(self):
        tag_strings = [
            "tag1",
            "tag1:value1",
            "tag1:value1:value11",
            "\x00tag1",
            "tag1:\x00value1",
            "tag1,tag2",
            "...",
            ".test",
            "test.a.b",
        ]
        expected_tags = [
            ["tag1"],
            ["tag1:value1"],
            ["tag1:value1:value11"],
            ["tag1"],
            ["tag1:value1"],
            ["tag1", "tag2"],
            [],
            ["test"],
            ["test.a.b"],
        ]
        for tag_string, expected_tag in zip(tag_strings, expected_tags):
            hda = self._create_vanilla_hda()
            self.tag_handler.apply_item_tags(user=self.user, item=hda, tags_str=tag_string)
            self._check_tag_list(hda.tags, expected_tag)

    def test_set_tag_from_list(self):
        hda = self._create_vanilla_hda()
        tags = ["tag1", "tag2"]
        self.tag_handler.set_tags_from_list(self.user, hda, tags)
        self._check_tag_list(hda.tags, tags)
        # Setting tags should erase previous tags
        self.tag_handler.set_tags_from_list(self.user, hda, ["tag1"])
        self._check_tag_list(hda.tags, expected_tags=["tag1"])

    def test_add_tag_from_list(self):
        hda = self._create_vanilla_hda()
        tags = ["tag1", "tag2"]
        self.tag_handler.add_tags_from_list(self.user, hda, tags)
        self._check_tag_list(tags=hda.tags, expected_tags=tags)
        # Adding tags should keep previous tags
        self.tag_handler.add_tags_from_list(self.user, hda, ["tag3"])
        self._check_tag_list(hda.tags, expected_tags=["tag1", "tag2", "tag3"])

    def test_remove_tag_from_list(self):
        hda = self._create_vanilla_hda()
        tags = ["tag1", "tag2", "tag3"]
        self.tag_handler.set_tags_from_list(self.user, hda, tags)
        self._check_tag_list(hda.tags, tags)
        self.tag_handler.remove_tags_from_list(self.user, hda, ["tag1", "tag3"])
        self._check_tag_list(hda.tags, ["tag2"])

    def test_delete_item_tags(self):
        hda = self._create_vanilla_hda()
        tags = ["tag1"]
        self.tag_handler.set_tags_from_list(self.user, hda, tags)
        self.tag_handler.delete_item_tags(user=self.user, item=hda)
        assert hda.tags == []

    def test_unique_constraint_applied(self):
        tag_name = "abc"
        tag = self.tag_handler._create_tag_instance(tag_name)
        same_tag = self.tag_handler._create_tag_instance(tag_name)
        assert tag.id == same_tag.id

    def test_item_has_tag(self):
        hda = self._create_vanilla_hda()
        tags = ["tag1"]
        self.tag_handler.set_tags_from_list(self.user, hda, tags)
        assert self.tag_handler.item_has_tag(self.user, item=hda, tag="tag1")
        # ItemTagAssociation
        assert self.tag_handler.item_has_tag(self.user, item=hda, tag=hda.tags[0])
        # Tag
        assert self.tag_handler.item_has_tag(self.user, item=hda, tag=hda.tags[0].tag)
        assert not self.tag_handler.item_has_tag(self.user, item=hda, tag="tag2")