Files
dvc/tests/test_unprotect.py
T
Ruslan Kuprieiev 4896f08ddf dvc: comply with both flake8 and black
flake8 provides some additional features like unused vars check, etc, so
it makes total sense to use it as well.

Signed-off-by: Ruslan Kuprieiev <[email protected]>
2019-02-04 00:10:36 +02:00

36 lines
851 B
Python

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,
)