from unittest import TestCase

from dvc.graph.workflow import Workflow,\
    CollapseDvcReproCommitsStrategy, CollapseNotMeticsCommitsStrategy
from dvc.graph.commit import Commit


class TestWorkflow(TestCase):
    def setUp(self):
        self._commit4 = Commit('4', '3', 'name1', 'today', 'comment4')
        self._commit3 = Commit('3', '2', 'name1', 'today', 'DVC repro-run ...')
        self._commit2 = Commit('2', '1', 'name1', 'today', 'DVC repro-run ...')
        self._commit1 = Commit('1', '', 'name1', 'today', 'comment1')

        self._showed_text1 = '[' + self._commit1.hash + '] ' + self._commit1._comment
        self._showed_text2 = '[' + self._commit2.hash + '] ' + self._commit2._comment
        self._showed_text3 = '[' + self._commit3.hash + '] ' + self._commit3._comment
        self._showed_text4 = '[' + self._commit4.hash + '] ' + self._commit4._comment

        self._showed_text3_collapsed = '[' + self._commit3.hash + '] ' + Commit.COLLAPSED_TEXT

    def commits_basic_test(self):
        self.assertFalse(self._commit1.is_repro)
        self.assertTrue(self._commit2.is_repro)
        self.assertTrue(self._commit3.is_repro)
        self.assertFalse(self._commit4.is_repro)
        pass

    def workflow_basic_test(self):
        wf = Workflow('', '')
        wf.add_commit(self._commit1)
        wf.add_commit(self._commit2)
        wf.add_commit(self._commit3)
        wf.add_commit(self._commit4)

        self.assertEqual(len(wf._commits), 4)

        self.assertEqual(wf._commits['1'].text(), self._showed_text1)
        self.assertEqual(wf._commits['2'].text(), self._showed_text2)
        self.assertEqual(wf._commits['3'].text(), self._showed_text3)
        self.assertEqual(wf._commits['4'].text(), self._showed_text4)
        pass

    def collapse_test(self):
        wf = Workflow('', '')
        wf.add_commit(self._commit1)
        wf.add_commit(self._commit2)
        wf.add_commit(self._commit3)
        wf.add_commit(self._commit4)

        wf.collapse_commits(CollapseDvcReproCommitsStrategy(wf))

        self.assertEqual(len(wf._commits), 3)
        self.assertEqual(wf._commits[self._commit1.hash].text(), self._showed_text1)
        self.assertEqual(wf._commits[self._commit3.hash].text(), self._showed_text3_collapsed)
        self.assertTrue('2' not in wf._commits)

        self.assertFalse('2' in wf._edges)
        self.assertFalse('2' in wf._back_edges)
        pass

    def collapse_at_dead_end_test(self):
        wf = Workflow('', '')
        wf.add_commit(self._commit1)
        wf.add_commit(self._commit2)
        wf.add_commit(self._commit3) # Dead end which cannot be collapsed

        self.assertEqual(len(wf._commits), 3)
        wf.collapse_commits(CollapseDvcReproCommitsStrategy(wf))
        self.assertEqual(len(wf._commits), 2)

        self.assertTrue('1' in wf._commits)
        self.assertTrue('3' in wf._commits)
        self.assertFalse('2' in wf._commits)

        self.assertEqual(wf._commits[self._commit1.hash].text(), self._showed_text1)
        self.assertEqual(wf._commits[self._commit3.hash].text(), self._showed_text3)
        self.assertTrue('2' not in wf._commits)
        pass

    def collapse_metric_commit_test(self):
        value = 0.812345
        branches = ['master', 'try_smth']
        metric_commit2 = Commit('2', '1', 'name1', 'today', 'DVC repro-run ...',
                                True, value, branch_tips=branches)

        wf = Workflow('', '')
        wf.add_commit(self._commit1)
        wf.add_commit(metric_commit2)
        wf.add_commit(self._commit3)

        self.assertEqual(len(wf._commits), 3)
        wf.collapse_commits(CollapseDvcReproCommitsStrategy(wf))
        self.assertEqual(len(wf._commits), 2)

        self.assertEqual(wf._commits['3']._target_metric, value)
        self.assertEqual(wf._commits['3'].branch_tips, branches)
        pass

    def collapse_all_commits_test(self):
        value = 0.812345
        branches = ['master', 'try_smth']
        not_metric_commit2 = Commit('2', '1', 'name1', 'today', 'change smth 2...',
                                    True, branch_tips=branches)
        metric_commit3 = Commit('3', '2', 'name3', 'today3', 'hello3',
                                True, value)

        wf = Workflow('', '')
        wf.add_commit(self._commit1)
        wf.add_commit(not_metric_commit2)
        wf.add_commit(metric_commit3)
        wf.add_commit(self._commit4)

        self.assertEqual(len(wf._commits), 4)
        wf.collapse_commits(CollapseDvcReproCommitsStrategy(wf))
        self.assertEqual(len(wf._commits), 4)
        wf.collapse_commits(CollapseNotMeticsCommitsStrategy(wf))
        self.assertEqual(len(wf._commits), 2)

        self.assertEqual(wf._commits['3']._target_metric, value)
        self.assertEqual(wf._commits['3'].branch_tips, branches)
        self.assertTrue('4' in wf._commits)
        pass

    def collapsed_all_commits_comments_test(self):
        value = 0.812345
        branches = ['master', 'try_smth']
        not_metric_commit2 = Commit('2', '1', 'name1', 'today', 'change smth 2...',
                                    True, branch_tips=branches)
        metric_commit3 = Commit('3', '2', 'name3', 'today3', 'hello3',
                                True, value)

        wf = Workflow('', '')
        wf.add_commit(self._commit1)
        wf.add_commit(not_metric_commit2)
        wf.add_commit(metric_commit3)
        wf.add_commit(self._commit4)

        wf.collapse_commits(CollapseDvcReproCommitsStrategy(wf))
        wf.collapse_commits(CollapseNotMeticsCommitsStrategy(wf))
        self.assertEqual(len(wf._commits), 2)

        def is_collapsed_commit_presented(self, commit, collapsed_commit):
            self.assertTrue('[' + collapsed_commit.hash + ']' in commit.text())
            self.assertTrue(collapsed_commit._comment in commit.text())

        is_collapsed_commit_presented(self, wf._commits['3'], metric_commit3)
        is_collapsed_commit_presented(self, wf._commits['3'], not_metric_commit2)
        is_collapsed_commit_presented(self, wf._commits['3'], self._commit1)
        pass