update test cases

Author: codereverser

tests/base.py

@@ -1,4 +1,5 @@
 import io
+import json
 import os
 import re
 
@@ -31,9 +32,25 @@ def read_pdf(self, filename, password, output="dict"):
         use_pdfminer = self.mode == "pdfminer"
         return read_cas_pdf(filename, password, output=output, force_pdfminer=use_pdfminer)
 
+    def test_output_json(self):
+        for filename, password in self.pdf_files:
+            json_data = self.read_pdf(filename, password, output="json")
+            data = json.loads(json_data)
+            assert len(data.get("folios", [])) == 10
+            for folio in data["folios"]:
+                for scheme in folio.get("schemes", []):
+                    assert scheme["isin"] is not None
+                    assert scheme["amfi"] is not None
+            assert data.get("investor_info", {}).get("mobile") not in (None, "")
+            assert data["cas_type"] == "DETAILED"
+
     def test_read_summary(self):
         data = self.read_pdf(self.cams_summary_file_name, self.cams_password)
         assert len(data.get("folios", [])) == 4
+        for folio in data["folios"]:
+            for scheme in folio.get("schemes", []):
+                assert scheme["isin"] is not None
+                assert scheme["amfi"] is not None
         assert data.get("investor_info", {}).get("mobile") not in (None, "")
         assert data["cas_type"] == "SUMMARY"
 

tests/test_mupdf.py

@@ -1,4 +1,3 @@
-import json
 import re
 
 from click.testing import CliRunner
@@ -12,14 +11,6 @@
 class TestMuPDF(BaseTestClass):
     """Test PyMuPDF parser."""
 
-    def test_output_json(self):
-        for filename, password in self.pdf_files:
-            json_data = self.read_pdf(filename, password, output="json")
-            data = json.loads(json_data)
-            assert len(data.get("folios", [])) == 10
-            assert data.get("investor_info", {}).get("mobile") not in (None, "")
-            assert data["cas_type"] == "DETAILED"
-
     def test_output_csv(self):
         output = self.read_pdf(self.cams_file_name, self.cams_password, output="csv")
         assert isinstance(output, str)

tests/test_process.py

@@ -6,6 +6,7 @@
 from casparser.process import process_cas_text
 from casparser.process.cas_detailed import parse_header, get_transaction_type
 from casparser.process.cas_summary import parse_header as parse_summary_header
+from casparser.process.utils import isin_search
 from casparser.enums import TransactionType
 
 
@@ -45,3 +46,12 @@ def test_transaction_type(self):
             TransactionType.UNKNOWN,
             None,
         )
+
+    def test_isin_search(self):
+        isin, amfi = isin_search("Axis Long Term Equity Fund - Direct Growth", "KFINTECH" ,"128TSDGG")
+        assert isin == "INF846K01EW2"
+        assert amfi == "120503"
+
+        isin, amfi = isin_search("", "KARVY", "")
+        assert isin is None
+        assert amfi is None