1 files changed, 33 insertions, 0 deletions
diff --git a/tests/commands/clean_test.py b/tests/commands/clean_test.py
new file mode 100644
index 0000000..955a6bc
--- /dev/null
+++ b/tests/commands/clean_test.py
@@ -0,0 +1,33 @@
+import os.path
+from unittest import mock
+
+import pytest
+
+from pre_commit.commands.clean import clean
+
+
+@pytest.fixture(autouse=True)
+def fake_old_dir(tempdir_factory):
+    fake_old_dir = tempdir_factory.get()
+
+    def _expanduser(path, *args, **kwargs):
+        assert path == '~/.pre-commit'
+        return fake_old_dir
+
+    with mock.patch.object(os.path, 'expanduser', side_effect=_expanduser):
+        yield fake_old_dir
+
+
+def test_clean(store, fake_old_dir):
+    assert os.path.exists(fake_old_dir)
+    assert os.path.exists(store.directory)
+    clean(store)
+    assert not os.path.exists(fake_old_dir)
+    assert not os.path.exists(store.directory)
+
+
+def test_clean_idempotent(store):
+    clean(store)
+    assert not os.path.exists(store.directory)
+    clean(store)
+    assert not os.path.exists(store.directory)