add test_dirganize.py

Author: aahnik

test_dirganize.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python
+
+# pylint: skip-file
+
+import logging
+import os
+import shutil
+
+import click
+import pytest
+
+from dirganize import cli
+
+
+def create_files(folder, *files):
+    cwd = os.getcwd()
+    os.chdir(folder)
+
+    for file in files:
+        with open(file, "w"):
+            pass
+
+    os.chdir(cwd)
+
+
+def test_dirganize():
+    sb = "sandbox"
+
+    try:
+        shutil.rmtree(sb)
+    except:
+        pass
+    os.makedirs(sb)
+
+    files = ["a.mp4", "b.mp3", "c.png", "d.py", ".dotfile", "what", "e.gif"]
+    create_files(sb, *files)
+
+    with open(os.path.join(sb, ".dirganize.yml"), "w") as file:
+        file.write("Animations: [gif]")
+
+    with pytest.raises(click.exceptions.Exit):
+        cli.version_callback(value=True)
+
+    cli.verbosity_callback(value=True)
+
+    cli.main(sb)
+
+    assert set(os.listdir()) == set(
+        [
+            ".dirganize.yml",
+            ".dotfile",
+            "Images",
+            "Others",
+            "Videos",
+            "Audios",
+            "Texts",
+            "Animations",
+        ]
+    )
+    assert set(os.listdir("Others")) == set(["what"])
+    assert set(os.listdir("Animations")) == set(["e.gif"])
+    assert set(os.listdir("Audios")) == set(["b.mp3"])
+
+
+if __name__ == "__main__":
+    test_dirganize()