-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_headers.py
39 lines (27 loc) · 1.55 KB
/
test_headers.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from __future__ import absolute_import, division
import unittest
from six.moves import mock
from aspose_barcode_cloud import Configuration, ApiClient, BarcodeApi, EncodeBarcodeType
from aspose_barcode_cloud.rest import RESTClientObject
class TestHeaders(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.local_config = Configuration(access_token="fake token", host="localhost")
def setUp(self):
self.rest_client_mock = mock.Mock(spec_set=RESTClientObject)
def test_default_headers(self):
api_client = ApiClient(self.local_config)
api_client.rest_client = self.rest_client_mock
BarcodeApi(api_client).get_barcode_generate(EncodeBarcodeType.CODE128, "Test")
self.assertEqual(1, self.rest_client_mock.GET.call_count)
headers = self.rest_client_mock.GET.call_args[1]["headers"]
self.assertEqual("Aspose-Barcode-SDK/22.11.0/python", headers["User-Agent"])
self.assertEqual("python sdk", headers["x-aspose-client"])
self.assertEqual("22.11.0", headers["x-aspose-client-version"])
def test_header_override(self):
api_client = ApiClient(self.local_config, header_name="x-aspose-client", header_value="some custom sdk")
api_client.rest_client = self.rest_client_mock
BarcodeApi(api_client).get_barcode_generate(EncodeBarcodeType.CODE128, "Test")
self.assertEqual(1, self.rest_client_mock.GET.call_count)
headers = self.rest_client_mock.GET.call_args[1]["headers"]
self.assertEqual("some custom sdk", headers["x-aspose-client"])