import os

from dvc.main import main

from tests.basic_env import TestDvc


class TestInstall(TestDvc):
    def test(self):
        ret = main(['install'])
        self.assertEqual(ret, 0)

        ret = main(['install'])
        self.assertNotEqual(ret, 0)

        def hook(name):
            return os.path.join('.git', 'hooks', name)

        self.assertTrue(os.path.isfile(hook('post-checkout')))
        self.assertTrue(os.path.isfile(hook('pre-commit')))

        self.dvc.add(self.FOO)
        os.unlink(self.FOO)

        self.dvc.scm.checkout('branch', create_new=True)
        self.assertTrue(os.path.isfile(self.FOO))