From bbebd4a7a1c981f6392226dc4127de28ce9c1ab7 Mon Sep 17 00:00:00 2001
From: Shirshendu R Tripathi <shirsh8924@gmail.com>
Date: Sun, 27 Oct 2024 17:42:33 +0000
Subject: [PATCH] Mock file operations Tests

---
 tests/test_fileMetadata.py | 43 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 42 insertions(+), 1 deletion(-)

diff --git a/tests/test_fileMetadata.py b/tests/test_fileMetadata.py
index dae0a56..d870148 100644
--- a/tests/test_fileMetadata.py
+++ b/tests/test_fileMetadata.py
@@ -1,6 +1,7 @@
 # test_fileMetadata.py
+from unittest.mock import mock_open, patch
 import pytest
-from twinTrim.dataStructures.fileMetadata import FileMetadata
+from twinTrim.dataStructures.fileMetadata import normalStore, add_or_update_normal_file, FileMetadata
 
 def test_insert_new_file():
     # Test inserting a new file into the metadata
@@ -35,3 +36,43 @@ def test_insert_multiple_files():
     assert len(metadata.filepaths) == 2
     assert file_path1 in metadata.filepaths
     assert file_path2 in metadata.filepaths
+
+def test_add_or_update_new_file():
+    file_path = "C:\\Users\\2004s\\Desktop\\dummy\\dummy_5.txt"  # Define the mock file path
+    expected_file_hash = "b1295d8ebb927df19ad74eec6aea72e3"  # Use the actual hash computed from the file
+
+    # Mock the get_file_hash function to return the expected hash
+    with patch("twinTrim.utils.get_file_hash", return_value=expected_file_hash), \
+         patch("builtins.open", mock_open(read_data=b"some binary content")):
+
+        # Clear normalStore before the test to avoid conflicts
+        normalStore.clear()
+
+        # Call the function to add a new file
+        add_or_update_normal_file(file_path)
+
+        # Check that the expected file hash is in normalStore
+        assert expected_file_hash in normalStore.keys(), f"Expected hash '{expected_file_hash}' not found in normalStore"
+
+        # Check that the file path was added correctly
+        assert normalStore[expected_file_hash].filepaths == [file_path], "File path not added correctly"
+
+def test_add_or_update_existing_file():
+    file_path1 = "C:\\Users\\2004s\\Desktop\\dummy\\dummy_5.txt"
+    file_path2 = "C:\\Users\\2004s\\Desktop\\dummy\\dummy_5_v2.txt"
+    expected_file_hash = "b1295d8ebb927df19ad74eec6aea72e3"  # Use the actual hash computed from the file
+
+    # First add a file
+    with patch("twinTrim.utils.get_file_hash", return_value=expected_file_hash), \
+         patch("builtins.open", mock_open(read_data=b"some binary content")):
+        normalStore.clear()
+        add_or_update_normal_file(file_path1)
+
+    # Then update it
+    with patch("twinTrim.utils.get_file_hash", return_value=expected_file_hash), \
+         patch("builtins.open", mock_open(read_data=b"some binary content")):
+        add_or_update_normal_file(file_path2)
+
+    # Check that the file paths were updated correctly
+    assert expected_file_hash in normalStore.keys(), f"Expected hash '{expected_file_hash}' not found in normalStore"
+    assert normalStore[expected_file_hash].filepaths == [file_path1, file_path2], "File paths not updated correctly"
\ No newline at end of file