From 8f87e77ee96d2e8cc6c9850bac33db651f7abf55 Mon Sep 17 00:00:00 2001
From: Rehan <github@rehanzo.com>
Date: Fri, 5 Jan 2024 18:30:44 -0500
Subject: [PATCH] tests

---
 VisioNomicon/__init__.py |   1 -
 assets/test/1.jpg        | Bin 0 -> 33333 bytes
 assets/test/2.jpg        | Bin 0 -> 33331 bytes
 tests/conftest.py        |  29 +++++++++++++++++++++++++
 tests/test_mapping.py    |  45 +++++++++++++++++++++++++++++++++++++++
 5 files changed, 74 insertions(+), 1 deletion(-)
 create mode 100644 assets/test/1.jpg
 create mode 100644 assets/test/2.jpg
 create mode 100644 tests/conftest.py
 create mode 100644 tests/test_mapping.py

diff --git a/VisioNomicon/__init__.py b/VisioNomicon/__init__.py
index e08203e..e9707f0 100644
--- a/VisioNomicon/__init__.py
+++ b/VisioNomicon/__init__.py
@@ -6,4 +6,3 @@
 
 __version__ = "0.1.1"
 __author__ = 'Rehan Rana'
-
diff --git a/assets/test/1.jpg b/assets/test/1.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..ba240875710b7db813a89acd717fbe2262c834d7
GIT binary patch
literal 33333
zcmeIwIZPB$9LMqZW@g!CwP3hB8iz~4sNqssj4^<KY@@7r#%2**A;!opCME)&ba-nA
z9;h8EUX7x7c6b6d;)zz^jRoVZ2&fIl#^(Ef^YSK{fBwnu&ByFCpNSxMR!)|XQV3}+
zVGfFU8J^6vyv3eW!&fk~cvGRzs7#m@Z^TxV6>anxIqq!3y(BZ+s7#5En_-kxcsCT4
zR~z#Um&<8B6&cp3u}e{;v>vGp*XWz#AM(qD-`a-T$2`qzLU(8>_HC++6pAiYU7GD;
ziWSl}G+X>Esi>MQ!0rfij<XtCgM=bgRnb(NP1CIIt=4y;>9*jAgtUN=T(3Q{Bs8(6
zq0teQ-qIH4>FA42$}g=AbcRouIB9as)M>GC)01b<nVaHD%~+V3mAxp(y)<vx@)avr
ztzKJDxX!n}sJN`WqOxjp^_IHrJ9h5cy{CTfzWoOd9y)yFXw$LdCr&n>I(_Esx$~_T
zE?&BP<@$~Go40P?xqI*agNKhEKk4l1?s@*=<*V0kdf&e5fB)g*r_W!$4t(?Hl0qH1
zvY}vq<kGENil(WW-JeS;Rn}H@%@&am5S*54_m+f2Ce}DY(;FIF+5)4JJbhvLr5(=j
z=;WT5et)zd$^I@_?LQ?O2{xLmTZ~tw_48C+ED(dP-llp}T&){e6WdfT*xUgIFn|FJ
zU;qOczyJm?fB_6(00S7n00uCC0SsUO0~o*n1~7mD3}65Q7{CAqFn|FJU;qOczyJm?
zfB_6(00S7n00uCC0SsUO0~o*n1~7mD3}65Q7{CAqFn|FJU;qOczyJm?fB_6(00S7n
z00uCC0SsUO0~o*n1~7mD3}65Q7{CAqFn|FJU;qOczyJm?fB_6(00S7n00uCC0SsUO
S0~o*n1~7mD4E#?6X4iM5h$@c&

literal 0
HcmV?d00001

diff --git a/assets/test/2.jpg b/assets/test/2.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..af5d48ed8c3cf38f52bedebd41678501487837b1
GIT binary patch
literal 33331
zcmeIwIZPB$9LMqZW@g!CHId=+XdD($FlxBe1!D*xAloP_o*f9T5M$&P6BEIk#^9}$
z;BAMBS0#vNhbLeoo@fQ$S}@LvfZAYeY`*_DFK?3h=b!xEe9TVssR;69<zxvdg^<<~
z=75-+;mu6TTj)(Oe1$Vg)))DVs`#04MoeXS@j9Q8<H<HWi!!r~DtBD$bfdH~e{FF^
zjWN$iPIj74M20nL^imWltw$<DHS(tThx{_(x3;17QBU)l&>foFzFCznq3BZ8rP(ef
zTOn<Ov&GMnimKTH?2bU^7^|TrNGMWO6-~9-G|lSXWPKN!ZVQfxPYVdi&9}QsLlbHn
z8Xb}8&8=bHj=reGg0i|mXZZLD6DLJanHm#2Eos*5Iqu|?jQN>a*$Z+!i}RK&UABD1
z%2kC$t9@&VOUf%MtExBDY}~SS+x8thch&FNyKn!2gNF_uX*zcN#K}{q&zwDX{zA*e
zOP8-)ZM)un<L0f~ckbT1|KQ=H$DLi>J<p!Mc=_se@0+*%@7{m-`04YPuiyN+q)>;i
zY%tg#xpXU+qG_sT_veyIwY61Uvqi)Q1gGWN^Gicq3AK*U^oGXf*1*U_Z(mqJS%)(`
zDyb*B-yiKqvcC&f_fN@&gN@|s7GqUu{XA6{sbZkIx2e82b$45R%u$8S9bf<h7{CAq
zFn|FJU;qOczyJm?fB_6(00S7n00uCC0SsUO0~o*n1~7mD3}65Q7{CAqFn|FJU;qOc
zzyJm?fB_6(00S7n00uCC0SsUO0~o*n1~7mD3}65Q7{CAqFn|FJU;qOczyJm?fB_6(
z00S7n00uCC0SsUO0~o*n1~7mD3}65Q7{CAqFn|FJU;qOczyJm?fB_6(00S7n00uCC
S0SsUO0~o--e>5=8?D`I6k14zW

literal 0
HcmV?d00001

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..31f6749
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,29 @@
+import pytest, shutil, json
+from pathlib import Path
+
+@pytest.fixture
+def asset_copy(tmp_path):
+    # Create a subdirectory for our assets.
+    assets_dir = tmp_path / "assets"
+    assets_dir.mkdir()
+
+    # Your original assets directory.
+    original_assets_dir = Path("assets/test")  # Convert string to Path
+
+    # Copy your assets into the temporary directory before each test.
+    for item in original_assets_dir.iterdir():
+        if item.is_dir():
+            shutil.copytree(item, assets_dir / item.name)
+        else:
+            shutil.copy(item, assets_dir / item.name)
+
+    # Create mapping
+    mapping = {
+        str(assets_dir / "1.jpg"): str(assets_dir / "blue.jpg"),
+        str(assets_dir / "2.jpg"): str(assets_dir / "green.jpg")
+    }
+    mapping_file = assets_dir / 'mapping.json'
+    with open(mapping_file, 'w') as f:
+        json.dump(mapping, f, indent=4)
+    
+    return assets_dir
diff --git a/tests/test_mapping.py b/tests/test_mapping.py
new file mode 100644
index 0000000..41a65ab
--- /dev/null
+++ b/tests/test_mapping.py
@@ -0,0 +1,45 @@
+import filecmp, hashlib, os.path, pytest, json
+from types import SimpleNamespace
+from VisioNomicon.main import save_mapping, rename_from_mapping
+
+def test_mapping_save(asset_copy):
+    og_filepaths = [str(asset_copy / '1.jpg'), str(asset_copy / '2.jpg')]
+    new_filepaths = [str(asset_copy / 'blue.jpg'), str(asset_copy / 'green.jpg')]
+
+    args = SimpleNamespace()
+    args.files = og_filepaths
+    args.output = str(asset_copy / 'mapping-test.json')
+
+    save_mapping(args, new_filepaths)
+
+    assert filecmp.cmp(args.output, asset_copy / 'mapping.json')
+
+def test_rename_from_mapping(asset_copy):
+    # test execution
+    checksums = [sha256sum(asset_copy / "1.jpg"), sha256sum(asset_copy / "2.jpg")]
+    rename_from_mapping(str(asset_copy / 'mapping.json'))
+
+    assert os.path.isfile(asset_copy / "blue.jpg")
+    assert os.path.isfile(asset_copy / "green.jpg")
+
+    rename_sums = [sha256sum(asset_copy / "blue.jpg"), sha256sum(asset_copy / "green.jpg")]
+    assert(checksums == rename_sums)
+
+    # test undo
+    rename_from_mapping(str(asset_copy / 'mapping.json'), True)
+    checksums = [sha256sum(asset_copy / "1.jpg"), sha256sum(asset_copy / "2.jpg")]
+
+    assert os.path.isfile(asset_copy / "1.jpg")
+    assert os.path.isfile(asset_copy / "2.jpg")
+
+    assert(checksums == rename_sums)
+    
+def sha256sum(filename):
+    h  = hashlib.sha256()
+    b  = bytearray(128*1024)
+    mv = memoryview(b)
+    with open(filename, 'rb', buffering=0) as f:
+        while n := f.readinto(mv):
+            h.update(mv[:n])
+    return h.hexdigest()
+