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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
# Copyright Notice:
# Copyright 2016-2021 DMTF. All rights reserved.
# License: BSD 3-Clause License. For full text see link:
# https://github.com/DMTF/python-redfish-library/blob/main/LICENSE.md
# -*- encoding: utf-8 -*-
import json
import unittest
from unittest import mock
from redfish.rest.v1 import HttpClient, RetriesExhaustedError, redfish_client
class TestRedFishClient(unittest.TestCase):
def setUp(self) -> None:
self.base_url = "http://foo.bar"
self.username = "rstallman"
self.password = "123456"
self.default_prefix = "/custom/redfish/v1/"
self.sessionkey = "fg687glgkf56vlgkf"
self.capath = "/path/to/the/dir"
self.cafile = "filename.test"
self.timeout = 666
self.max_retry = 42
def test_redfish_client(self) -> None:
# NOTE(hberaud) the client try to connect when we initialize the
# http client object so we need to catch the retries exception first.
# In a second time we need to mock the six.http_client to simulate
# server responses and do some other tests
with self.assertRaises(RetriesExhaustedError):
client = redfish_client(base_url=self.base_url)
# Check the object type
self.assertTrue(isinstance(client, HttpClient))
# Check the object attributes values.
# Here we check if the client object is properly initialized
self.assertEqual(client.base_url, self.base_url)
self.assertEqual(client.username, self.username)
self.assertEqual(client.password, self.password)
self.assertEqual(client.default_prefix, self.default_prefix)
self.assertEqual(client.sessionkey, self.sessionkey)
self.assertEqual(client.capath, self.capath)
self.assertEqual(client.cafile, self.cafile)
self.assertEqual(client.timeout, self.timeout)
self.assertEqual(client.max_retry, self.max_retry)
def test_redfish_client_no_root_resp(self) -> None:
client = redfish_client(base_url=self.base_url, check_connectivity=False)
self.assertIsNone(getattr(client, "root_resp", None))
@mock.patch("requests.Session.request")
def test_redfish_client_root_object_initialized_after_login(
self, mocked_request: mock.Mock
) -> None:
dummy_root_data = '{"Links": {"Sessions": {"@data.id": "/redfish/v1/SessionService/Sessions"}}}'
dummy_session_response = (
'{"@odata.type": "#Session.v1_1_2.Session", '
'"@odata.id": "/redfish/v1/SessionService/Sessions/1", '
'"Id": "1", "Name": "User Session", "Description": "Manager User Session", '
'"UserName": "user", "Oem": {}}'
)
root_resp = mock.Mock(content=dummy_root_data, status_code=200)
auth_resp = mock.Mock(
content=dummy_session_response,
status_code=200,
headers={"location": "fake", "x-auth-token": "fake"},
)
mocked_request.side_effect = [
root_resp,
auth_resp,
]
client = redfish_client(base_url=self.base_url, check_connectivity=False)
client.login()
self.assertEqual(client.root, json.loads(dummy_root_data))
if __name__ == "__main__":
unittest.main()
|
