Turn off code execution randomness during pytest

Author: rlan

src/beancount_multitool/JABank.py

@@ -2,6 +2,7 @@
 import pandas as pd
 from pathlib import Path
 import uuid
+import sys
 
 from .Institution import Institution
 from .MappingDatabase import MappingDatabase
@@ -108,7 +109,11 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                         if amount > 0:  # a credit
                             metadata["uuid"] = ""
                         else:  # a debit
-                            metadata["uuid"] = str(uuid.uuid4())
+                            if hasattr(sys, "_called_from_pytest"):
+                                # remove randomness during pytest
+                                metadata["uuid"] = "_called_from_pytest"
+                            else:
+                                metadata["uuid"] = str(uuid.uuid4())
 
                     account_metadata = {}
                     for x in range(1, len(accounts)):
@@ -126,6 +131,7 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                     f.write(output)
                 print(f"Written {file_name}")
         except IOError as e:
+            print(f"Error encountered while writing to: {file_name}")
             print(e)
 
     def convert(self, csv_file: str, bean_file: str):

src/beancount_multitool/RakutenBank.py

@@ -1,6 +1,7 @@
 import pandas as pd
 from pathlib import Path
 import uuid
+import sys
 
 from .Institution import Institution
 from .MappingDatabase import MappingDatabase
@@ -93,7 +94,11 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                         if amount > 0:  # a credit
                             metadata["uuid"] = ""
                         else:  # a debit
-                            metadata["uuid"] = str(uuid.uuid4())
+                            if hasattr(sys, "_called_from_pytest"):
+                                # remove randomness during pytest
+                                metadata["uuid"] = "_called_from_pytest"
+                            else:
+                                metadata["uuid"] = str(uuid.uuid4())
 
                     account_metadata = {}
                     for x in range(1, len(accounts)):
@@ -111,6 +116,7 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                     f.write(output)
                 print(f"Written {file_name}")
         except IOError as e:
+            print(f"Error encountered while writing to: {file_name}")
             print(e)
 
     def convert(self, csv_file: str, bean_file: str):

src/beancount_multitool/RakutenCard.py

@@ -121,6 +121,7 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                     f.write(output)
                 print(f"Written {file_name}")
         except IOError as e:
+            print(f"Error encountered while writing to: {file_name}")
             print(e)
 
     def convert(self, csv_file: str, bean_file: str):

src/beancount_multitool/ShinseiBank.py

@@ -1,6 +1,7 @@
 import pandas as pd
 from pathlib import Path
 import uuid
+import sys
 
 from .Institution import Institution
 from .MappingDatabase import MappingDatabase
@@ -112,7 +113,11 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                         if amount > 0:  # a credit
                             metadata["uuid"] = ""
                         else:  # a debit
-                            metadata["uuid"] = str(uuid.uuid4())
+                            if hasattr(sys, "_called_from_pytest"):
+                                # remove randomness during pytest
+                                metadata["uuid"] = "_called_from_pytest"
+                            else:
+                                metadata["uuid"] = str(uuid.uuid4())
 
                     account_metadata = {}
                     for x in range(1, len(accounts)):
@@ -130,6 +135,7 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                     f.write(output)
                 print(f"Written {file_name}")
         except IOError as e:
+            print(f"Error encountered while writing to: {file_name}")
             print(e)
 
     def convert(self, csv_file: str, bean_file: str):

src/beancount_multitool/SumishinNetBank.py

@@ -1,6 +1,7 @@
 from decimal import Decimal
 from pathlib import Path
 import uuid
+import sys
 
 import pandas as pd
 
@@ -113,7 +114,11 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                         if amount > 0:  # a credit
                             metadata["uuid"] = ""
                         else:  # a debit
-                            metadata["uuid"] = str(uuid.uuid4())
+                            if hasattr(sys, "_called_from_pytest"):
+                                # remove randomness during pytest
+                                metadata["uuid"] = "_called_from_pytest"
+                            else:
+                                metadata["uuid"] = str(uuid.uuid4())
 
                     account_metadata = {}
                     for x in range(1, len(accounts)):
@@ -131,6 +136,7 @@ def write_bean(self, df: pd.DataFrame, file_name: str) -> None:
                     f.write(output)
                 print(f"Written {file_name}")
         except IOError as e:
+            print(f"Error encountered while writing to: {file_name}")
             print(e)
 
     def convert(self, csv_file: str, bean_file: str):

tests/conftest.py

@@ -0,0 +1,10 @@
+# content of conftest.py
+
+# Ref: https://stackoverflow.com/a/25188424
+def pytest_configure(config):
+    import sys
+    sys._called_from_pytest = True
+
+def pytest_unconfigure(config):
+    import sys
+    del sys._called_from_pytest