init

tests/test_ip_lookup.py

@@ -0,0 +1,47 @@
+"""Tests for ip_lookup module."""
+
+from unittest.mock import MagicMock, patch
+
+from ip_lookup import get_ipv4, get_ipv6
+
+
+class TestGetIpv4:
+    def test_returns_stripped_ip(self) -> None:
+        mock_result = MagicMock()
+        mock_result.stdout = "  1.2.3.4  \n"
+
+        with patch("ip_lookup.subprocess.run", return_value=mock_result) as mock_run:
+            ip = get_ipv4()
+
+            assert ip == "1.2.3.4"
+            mock_run.assert_called_once_with(["curl", "-4", "ifconfig.me"], capture_output=True, text=True, check=False)
+
+    def test_empty_response(self) -> None:
+        mock_result = MagicMock()
+        mock_result.stdout = ""
+
+        with patch("ip_lookup.subprocess.run", return_value=mock_result):
+            ip = get_ipv4()
+
+            assert ip == ""
+
+
+class TestGetIpv6:
+    def test_returns_stripped_ip(self) -> None:
+        mock_result = MagicMock()
+        mock_result.stdout = "  2001:db8::1  \n"
+
+        with patch("ip_lookup.subprocess.run", return_value=mock_result) as mock_run:
+            ip = get_ipv6()
+
+            assert ip == "2001:db8::1"
+            mock_run.assert_called_once_with(["curl", "-6", "ifconfig.me"], capture_output=True, text=True, check=False)
+
+    def test_empty_response(self) -> None:
+        mock_result = MagicMock()
+        mock_result.stdout = ""
+
+        with patch("ip_lookup.subprocess.run", return_value=mock_result):
+            ip = get_ipv6()
+
+            assert ip == ""