Running autopep8 against files

KingAkeem · KingAkeem · commit 60f7c6df3fea · 2025-10-29T16:00:46.000-04:00
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -16,16 +16,16 @@ def _install_nlp_stub():
 
     # Create a minimal module with classify function
     stub = types.ModuleType(mod_name)
-    
+
     def classify(data):
         """Lightweight test-only classifier.
-        
+
         Returns a deterministic classification without requiring ML libraries.
         Real implementation uses sklearn pipeline with training data.
         """
         _ = data  # unused in stub
         return ["unknown", 0.0]
-    
+
     # Use setattr to avoid linter complaints about dynamic attributes
     setattr(stub, "classify", classify)
     sys.modules[mod_name] = stub
diff --git a/tests/test_linktree_tree.py b/tests/test_linktree_tree.py
@@ -53,7 +53,7 @@ def test_parse_hostname_raises_on_invalid():
     """Ensure parse_hostname raises when URL has no hostname."""
     with pytest.raises(Exception, match="unable to parse hostname"):
         parse_hostname("not-a-url")
-    
+
     with pytest.raises(Exception, match="unable to parse hostname"):
         parse_hostname("file:///local/path")
 
@@ -69,7 +69,7 @@ def test_linknode_initialization():
         numbers=["+14155551234"],
         emails=["test@example.com"],
     )
-    
+
     assert node.tag == "Test Page"
     assert node.identifier == "https://example.com"
     assert node.status == 200
@@ -89,14 +89,14 @@ def test_linktree_creates_root_node_with_title():
         </body>
     </html>
     """
-    
+
     client = FakeClient({
         "https://example.com": FakeResponse("https://example.com", html, 200),
     })
-    
+
     tree = LinkTree("https://example.com", depth=0, client=client)
     tree.load()
-    
+
     root = tree.get_node("https://example.com")
     assert root is not None
     assert root.tag == "Example Site"
@@ -106,14 +106,14 @@ def test_linktree_creates_root_node_with_title():
 def test_linktree_creates_root_node_without_title():
     """Test that LinkTree falls back to hostname when no <title> tag."""
     html = "<html><body><p>No title here</p></body></html>"
-    
+
     client = FakeClient({
         "https://test.onion": FakeResponse("https://test.onion", html, 200),
     })
-    
+
     tree = LinkTree("https://test.onion", depth=0, client=client)
     tree.load()
-    
+
     root = tree.get_node("https://test.onion")
     assert root is not None
     assert root.tag == "test.onion"
@@ -131,16 +131,16 @@ def test_linktree_extracts_contacts_from_page():
         </body>
     </html>
     """
-    
+
     client = FakeClient({
         "https://example.com/contact": FakeResponse(
             "https://example.com/contact", html, 200
         ),
     })
-    
+
     tree = LinkTree("https://example.com/contact", depth=0, client=client)
     tree.load()
-    
+
     root = tree.get_node("https://example.com/contact")
     assert root is not None
     assert len(root.data.emails) == 2
@@ -160,10 +160,10 @@ def test_linktree_builds_tree_to_depth_1():
         </body>
     </html>
     """
-    
+
     child1_html = "<html><head><title>Child 1</title></head></html>"
     child2_html = "<html><head><title>Child 2</title></head></html>"
-    
+
     client = FakeClient({
         "https://example.com": FakeResponse("https://example.com", root_html, 200),
         "https://example.com/child1": FakeResponse(
@@ -173,22 +173,22 @@ def test_linktree_builds_tree_to_depth_1():
             "https://example.com/child2", child2_html, 200
         ),
     })
-    
+
     tree = LinkTree("https://example.com", depth=1, client=client)
     tree.load()
-    
+
     # Verify root exists
     root = tree.get_node("https://example.com")
     assert root is not None
-    
+
     # Verify children exist
     child1 = tree.get_node("https://example.com/child1")
     child2 = tree.get_node("https://example.com/child2")
     assert child1 is not None
     assert child2 is not None
     assert child1.tag == "Child 1"
     assert child2.tag == "Child 2"
-    
+
     # Verify tree structure
     assert tree.parent("https://example.com/child1").identifier == "https://example.com"
     assert tree.parent("https://example.com/child2").identifier == "https://example.com"
@@ -199,7 +199,7 @@ def test_linktree_respects_depth_limit():
     root_html = '<html><title>Root</title><a href="https://example.com/level1">L1</a></html>'
     level1_html = '<html><title>Level 1</title><a href="https://example.com/level2">L2</a></html>'
     level2_html = '<html><title>Level 2</title><a href="https://example.com/level3">L3</a></html>'
-    
+
     client = FakeClient({
         "https://example.com": FakeResponse("https://example.com", root_html, 200),
         "https://example.com/level1": FakeResponse(
@@ -212,16 +212,16 @@ def test_linktree_respects_depth_limit():
             "https://example.com/level3", '<html><title>Level 3</title></html>', 200
         ),
     })
-    
+
     # Build tree with depth=2 (root + 2 levels)
     tree = LinkTree("https://example.com", depth=2, client=client)
     tree.load()
-    
+
     # Root and level1 and level2 should exist
     assert tree.get_node("https://example.com") is not None
     assert tree.get_node("https://example.com/level1") is not None
     assert tree.get_node("https://example.com/level2") is not None
-    
+
     # Level3 should NOT exist (depth limit)
     assert tree.get_node("https://example.com/level3") is None
 
@@ -236,10 +236,10 @@ def test_linktree_handles_duplicate_links():
         <a href="https://example.com/other">Other</a>
     </html>
     """
-    
+
     page_html = "<html><title>Target Page</title></html>"
     other_html = "<html><title>Other Page</title></html>"
-    
+
     client = FakeClient({
         "https://example.com": FakeResponse("https://example.com", html_with_dup, 200),
         "https://example.com/page": FakeResponse(
@@ -249,14 +249,14 @@ def test_linktree_handles_duplicate_links():
             "https://example.com/other", other_html, 200
         ),
     })
-    
+
     tree = LinkTree("https://example.com", depth=1, client=client)
     tree.load()
-    
+
     # Should have 3 nodes total: root + 2 unique children
     all_nodes = tree.all_nodes()
     assert len(all_nodes) == 3
-    
+
     # Duplicate should have been attempted once (first add), then skipped
     assert tree.get_node("https://example.com/page") is not None
     assert tree.get_node("https://example.com/other") is not None
@@ -265,23 +265,23 @@ def test_linktree_handles_duplicate_links():
 def test_linktree_save_json_creates_file():
     """Test saveJSON writes a valid JSON file with tree structure."""
     html = '<html><title>Test JSON Save</title></html>'
-    
+
     client = FakeClient({
         "https://example.com": FakeResponse("https://example.com", html, 200),
     })
-    
+
     tree = LinkTree("https://example.com", depth=0, client=client)
     tree.load()
-    
+
     with tempfile.TemporaryDirectory() as tmpdir:
         # Patch project_root_directory to use temp dir
         with patch("torbot.modules.linktree.project_root_directory", tmpdir):
             tree.saveJSON()
-            
+
             # Check that JSON file was created
             json_files = list(Path(tmpdir).glob("*.json"))
             assert len(json_files) == 1
-            
+
             # Verify JSON file is not empty and is valid JSON
             with open(json_files[0]) as f:
                 content = f.read()
@@ -295,39 +295,39 @@ def test_linktree_save_json_creates_file():
 def test_linktree_save_text_creates_file():
     """Test save creates a text file representation of the tree."""
     html = '<html><title>Test Text Save</title></html>'
-    
+
     client = FakeClient({
         "https://example.com": FakeResponse("https://example.com", html, 200),
     })
-    
+
     tree = LinkTree("https://example.com", depth=0, client=client)
     tree.load()
-    
+
     with tempfile.TemporaryDirectory() as tmpdir:
         with patch("torbot.modules.linktree.project_root_directory", tmpdir):
             tree.save()
-            
+
             # Check that text file was created
             txt_files = list(Path(tmpdir).glob("*.txt"))
             assert len(txt_files) == 1
-            
+
             # Verify file is not empty
             assert txt_files[0].stat().st_size > 0
 
 
 def test_linktree_handles_non_200_status():
     """Verify LinkTree records non-200 status codes correctly."""
     html = '<html><title>Not Found</title></html>'
-    
+
     client = FakeClient({
         "https://example.com/missing": FakeResponse(
             "https://example.com/missing", html, 404
         ),
     })
-    
+
     tree = LinkTree("https://example.com/missing", depth=0, client=client)
     tree.load()
-    
+
     root = tree.get_node("https://example.com/missing")
     assert root is not None
     assert root.data.status == 404
@@ -345,17 +345,17 @@ def test_linktree_filters_invalid_links():
         <a href="#fragment">Fragment</a>
     </html>
     """
-    
+
     valid_html = '<html><title>Valid</title></html>'
-    
+
     client = FakeClient({
         "https://example.com": FakeResponse("https://example.com", html, 200),
         "https://valid.com": FakeResponse("https://valid.com", valid_html, 200),
     })
-    
+
     tree = LinkTree("https://example.com", depth=1, client=client)
     tree.load()
-    
+
     # Should have 2 nodes: root + 1 valid child
     all_nodes = tree.all_nodes()
     assert len(all_nodes) == 2
