import os
import stat

from dvc.main import main

from tests.basic_env import TestDvc


class TestUnprotect(TestDvc):
    def test(self):
        ret = main(["config", "cache.protected", "true"])
        self.assertEqual(ret, 0)

        ret = main(["add", self.FOO])
        self.assertEqual(ret, 0)

        self.assertNotEqual(
            os.stat(self.FOO).st_mode
            & (stat.S_IREAD | stat.S_IRGRP | stat.S_IROTH),
            0,
        )
        self.assertEqual(
            os.stat(self.FOO).st_mode
            & (stat.S_IWRITE | stat.S_IWGRP | stat.S_IWOTH),
            0,
        )

        ret = main(["unprotect", self.FOO])
        self.assertEqual(ret, 0)

        self.assertNotEqual(
            os.stat(self.FOO).st_mode
            & (stat.S_IREAD | stat.S_IRGRP | stat.S_IROTH | stat.S_IWRITE),
            0,
        )