Couple of tweaks to improve tests

haralyzer · Sep 6, 2020 · 431ff84 · 431ff84
1 parent cc15335
commit 431ff84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/haralyzer/http.py b/haralyzer/http.py
@@ -22,10 +22,6 @@ def bodySize(self):
     def cookies(self):
         return self.raw_entry["cookies"]
 
-    @cached_property
-    def headers(self):
-        return self.raw_entry["headers"]
-
     @cached_property
     def headersSize(self):
         return self.raw_entry["headersSize"]
@@ -82,10 +78,6 @@ class Response(HttpTransaction):
     def bodySize(self):
         return self.raw_entry["bodySize"]
 
-    @cached_property
-    def headers(self):
-        return self.raw_entry["headers"]
-
     @cached_property
     def headersSize(self):
         return self.raw_entry["headersSize"]

diff --git a/tests/test_page.py b/tests/test_page.py
@@ -19,11 +19,12 @@ def test_init(har_data):
 
     # Throws PageNotFoundException with bad page ID
     with pytest.raises(PageNotFoundError):
-        page = HarPage(BAD_PAGE_ID, har_data=init_data)
+        assert HarPage(BAD_PAGE_ID, har_data=init_data)
 
     # Make sure it can load with either har_data or a parser
     page = HarPage(PAGE_ID, har_data=init_data)
     assert isinstance(page, HarPage)
+    assert repr(page) == "ID: page_3, URL: http://humanssuck.net/"
     parser = HarParser(init_data)
     page = HarPage(PAGE_ID, har_parser=parser)
     assert isinstance(page, HarPage)