write unit tests for request_item

tests/requests/test_batch_request_item.py

@@ -0,0 +1,77 @@
+import pytest
+from unittest.mock import Mock
+from urllib.request import Request
+from kiota_abstractions.request_information import RequestInformation
+from kiota_abstractions.headers_collection import HeadersCollection as RequestHeaders
+from msgraph_core.requests.batch_request_item import BatchRequestItem, StreamInterface
+from kiota_abstractions.serialization import SerializationWriter
+
+base_url = "https://graph.microsoft.com/v1.0/me"
+
+
+@pytest.fixture
+def request_info():
+    request_info = RequestInformation()
+    request_info.http_method = "GET"
+    request_info.url = "f{base_url}/me"
+    request_info.headers = RequestHeaders()
+    request_info.content = StreamInterface(b'{"key": "value"}')
+    return request_info
+
+
+@pytest.fixture
+def batch_request_item(request_info):
+    return BatchRequestItem(request_information=request_info)
+
+
+def test_initialization(batch_request_item, request_info):
+    assert batch_request_item.method == "GET"
+    assert batch_request_item.url == "f{base_url}/me"
+    assert batch_request_item.body.read() == b'{"key": "value"}'
+
+
+def test_create_with_urllib_request():
+    urllib_request = Request("https://graph.microsoft.com/v1.0/me", method="POST")
+    urllib_request.add_header("Content-Type", "application/json")
+    urllib_request.data = b'{"key": "value"}'
+    batch_request_item = BatchRequestItem.create_with_urllib_request(urllib_request)
+    assert batch_request_item.method == "POST"
+    assert batch_request_item.url == "https://graph.microsoft.com/v1.0/me"
+    assert batch_request_item.body == b'{"key": "value"}'
+
+
+def test_set_depends_on(batch_request_item):
+    batch_request_item.set_depends_on(["request1", "request2"])
+    assert batch_request_item.depends_on == ["request1", "request2"]
+
+
+def test_set_url(batch_request_item):
+    batch_request_item.set_url("https://graph.microsoft.com/v1.0/me")
+    assert batch_request_item.url == "/v1.0/me"
+
+
+def test_id_property(batch_request_item):
+    batch_request_item.id = "new_id"
+    assert batch_request_item.id == "new_id"
+
+
+def test_headers_property(batch_request_item):
+    new_headers = {"Authorization": "Bearer token"}
+    batch_request_item.headers = new_headers
+    assert batch_request_item.headers["Authorization"] == "Bearer token"
+
+
+def test_body_property(batch_request_item):
+    new_body = StreamInterface(b'{"new_key": "new_value"}')
+    batch_request_item.body = new_body
+    assert batch_request_item.body.read() == b'{"new_key": "new_value"}'
+
+
+def test_method_property(batch_request_item):
+    batch_request_item.method = "POST"
+    assert batch_request_item.method == "POST"
+
+
+def test_depends_on_property(batch_request_item):
+    batch_request_item.set_depends_on(["request1", "request2"])
+    assert batch_request_item.depends_on == ["request1", "request2"]