Merge pull request #1 from railsware/add-user-agent

Add User-Agent header
railsware · May 5, 2023 · 9af3c71 · 9af3c71
2 parents a48614b + 1883267
commit 9af3c71
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/mailtrap/client.py b/mailtrap/client.py
@@ -43,6 +43,9 @@ def headers(self) -> Dict[str, str]:
         return {
             "Authorization": f"Bearer {self.token}",
             "Content-Type": "application/json",
+            "User-Agent": (
+                "mailtrap-python (https://github.com/railsware/mailtrap-python)"
+            ),
         }
 
     @staticmethod

diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -41,6 +41,9 @@ def test_headers_should_return_appropriate_dict(self) -> None:
         assert client.headers == {
             "Authorization": "Bearer fake_token",
             "Content-Type": "application/json",
+            "User-Agent": (
+                "mailtrap-python (https://github.com/railsware/mailtrap-python)"
+            ),
         }
 
     @responses.activate