Remove httpretty which does not support Python3.12.

Author: purple4reina

pyproject.toml

@@ -38,8 +38,7 @@ boto3 = { version = "^1.28.0", optional = true }
 typing_extensions = {version = "^4.0", python = "<3.8"}
 requests = { version ="^2.22.0", optional = true }
 nose2 = { version= "^0.9.1", optional = true }
-flake8 = { version = "^3.7.9", optional = true }
-httpretty = {version = "^0.9.7", optional = true }
+flake8 = { version = "^5.0.4", optional = true }
 
 
 [tool.poetry.extras]
@@ -48,7 +47,6 @@ dev = [
     "requests",
     "nose2",
     "flake8",
-    "httpretty",
 ]
 
 [build-system]

tests/test_extension.py

@@ -1,6 +1,7 @@
+import http.server
 import os
+import threading
 import unittest
-import httpretty
 
 from unittest.mock import patch
 
@@ -11,61 +12,62 @@
 )
 
 
-def exceptionCallback(request, uri, headers):
-    raise Exception("oopsy!")
+class MockServer(threading.Thread):
+    def __init__(self):
+        super().__init__()
+        self.daemon = True
+        self.raises = False
+        self.called = False
+
+        class handler(http.server.BaseHTTPRequestHandler):
+            def do_POST(sf):
+                self.called = True
+                sf.send_response(500 if self.raises else 200)
+                sf.end_headers()
+
+            do_GET = do_POST
+
+        self.server = http.server.HTTPServer(("127.0.0.1", 8124), handler)
+
+    def run(self):
+        self.server.serve_forever()
+
+    def stop(self):
+        self.server.shutdown()
+        self.server.server_close()
+        self.join(timeout=0)
 
 
 class TestLambdaExtension(unittest.TestCase):
+    def setUp(self):
+        self.server = MockServer()
+        self.server.start()
+
+    def tearDown(self):
+        self.server.stop()
+
     @patch("datadog_lambda.extension.EXTENSION_PATH", os.path.abspath(__file__))
     def test_is_extension_running_true(self):
-        httpretty.enable()
-        last_request = httpretty.last_request()
-        httpretty.register_uri(httpretty.GET, "http://127.0.0.1:8124/lambda/hello")
-        assert is_extension_running() == True
-        assert httpretty.last_request() != last_request
-        httpretty.disable()
+        assert is_extension_running()
+        assert self.server.called
 
     def test_is_extension_running_file_not_found(self):
-        httpretty.enable()
-        last_request = httpretty.last_request()
-        httpretty.register_uri(httpretty.GET, "http://127.0.0.1:8124/lambda/hello")
-        assert is_extension_running() == False
-        assert httpretty.last_request() == last_request
-        httpretty.disable()
+        assert not is_extension_running()
+        assert not self.server.called
 
     @patch("datadog_lambda.extension.EXTENSION_PATH", os.path.abspath(__file__))
     def test_is_extension_running_http_failure(self):
-        httpretty.enable()
-        last_request = httpretty.last_request()
-        httpretty.register_uri(
-            httpretty.GET,
-            "http://127.0.0.1:8124/lambda/hello",
-            status=503,
-            body=exceptionCallback,
-        )
-        assert is_extension_running() == False
-        assert httpretty.last_request() != last_request
-        httpretty.disable()
+        self.server.raises = True
+        assert not is_extension_running()
+        assert self.server.called
 
     @patch("datadog_lambda.extension.EXTENSION_PATH", os.path.abspath(__file__))
     def test_flush_ok(self):
-        httpretty.enable()
-        last_request = httpretty.last_request()
-        httpretty.register_uri(httpretty.POST, "http://127.0.0.1:8124/lambda/flush")
-        assert flush_extension() == True
-        assert httpretty.last_request() != last_request
-        httpretty.disable()
+        assert flush_extension()
+        assert self.server.called
 
     @patch("datadog_lambda.extension.EXTENSION_PATH", os.path.abspath(__file__))
     def test_flush_not_ok(self):
-        httpretty.enable()
-        last_request = httpretty.last_request()
-        httpretty.register_uri(
-            httpretty.POST,
-            "http://127.0.0.1:8124/lambda/flush",
-            status=503,
-            body=exceptionCallback,
-        )
-        assert flush_extension() == False
-        assert httpretty.last_request() != last_request
-        httpretty.disable()
+        self.server.raises = True
+        assert not flush_extension()
+        assert self.server.called