From b99aae582c776e8f5a43991c4b7df075cbce5d54 Mon Sep 17 00:00:00 2001
From: "gitauto-for-dev[bot]"
 <160085510+gitauto-for-dev[bot]@users.noreply.github.com>
Date: Tue, 11 Feb 2025 04:35:16 +0000
Subject: [PATCH] Update tests/services/github/test_create_headers.py

---
 tests/services/github/test_create_headers.py | 25 ++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 tests/services/github/test_create_headers.py

diff --git a/tests/services/github/test_create_headers.py b/tests/services/github/test_create_headers.py
new file mode 100644
index 00000000..ba082267
--- /dev/null
+++ b/tests/services/github/test_create_headers.py
@@ -0,0 +1,25 @@
+"""Unit tests for create_headers function.
+
+Related Documentation: 
+https://docs.github.com/en/rest/using-the-rest-api/getting-started-with-the-rest-api#headers
+"""
+
+from services.github.create_headers import create_headers
+
+
+def test_create_headers_default():
+    # Testing default headers creation
+    token = "testtoken"
+    headers = create_headers(token)
+    assert headers["Authorization"] == f"Bearer {token}"
+    assert headers["Accept"].startswith("application/vnd.github")
+    assert "User-Agent" in headers
+    assert "X-GitHub-Api-Version" in headers
+
+
+def test_create_headers_custom_media():
+    # Testing custom media type in headers creation
+    token = "testtoken"
+    media_type = ".v3+test"
+    headers = create_headers(token, media_type)
+    assert headers["Accept"] == f"application/vnd.github{media_type}+json"