1+ import sys
12from unittest import TestCase
23
34from mock import MagicMock , patch
45
5- from apsconnectcli .hub import osaapi_raise_for_status , Hub , APS
6+ from apsconnectcli .apsconnect import APSConnectUtil
7+ from apsconnectcli .hub import APS , Hub , osaapi_raise_for_status
8+
9+ if sys .version_info >= (3 ,):
10+ _BUILTINS_OPEN = 'builtins.open'
11+ _BUILTINS_PRINT = 'builtins.print'
12+ else :
13+ _BUILTINS_OPEN = 'apsconnectcli.apsconnect.open'
14+ _BUILTINS_PRINT = 'apsconnectcli.apsconnect.print'
615
716
817class OsaApiRaiseForStatusTest (TestCase ):
@@ -79,7 +88,7 @@ def test_get_resclass_name_without_unit(self):
7988
8089 def test_hub_init (self ):
8190 with patch ('apsconnectcli.hub.osaapi' ), \
82- patch ('apsconnectcli.hub.APS' ) as aps_mock , \
91+ patch ('apsconnectcli.hub.APS' ) as aps_mock , \
8392 patch ('apsconnectcli.hub.get_config' ), \
8493 patch ('apsconnectcli.hub.osaapi_raise_for_status' ):
8594 resp_mock = MagicMock ()
@@ -104,7 +113,7 @@ def test_get_hub_version(self):
104113
105114 def test_hub_incorrect_id (self ):
106115 with patch ('apsconnectcli.hub.osaapi' ), \
107- patch ('apsconnectcli.hub.APS' ) as aps_mock , \
116+ patch ('apsconnectcli.hub.APS' ) as aps_mock , \
108117 patch ('apsconnectcli.hub.get_config' ), \
109118 patch ('apsconnectcli.hub.osaapi_raise_for_status' ), \
110119 patch ('apsconnectcli.hub.sys' ) as sys_mock :
@@ -244,3 +253,17 @@ def test_delete(self):
244253 self .assertEqual (requests_mock .delete .call_args [1 ].get ('headers' ), 'token' )
245254 self .assertEqual (requests_mock .delete .call_args [1 ].get ('verify' ), False )
246255 self .assertEqual (requests_mock .delete .call_args [0 ][0 ], 'https://aps_host:aps_port/test' )
256+
257+
258+ class TestApsConnectUtilHubCommands (TestCase ):
259+ @patch ('apsconnectcli.apsconnect.Hub' )
260+ def test_hub_token (self , hub_cls ):
261+ with patch (_BUILTINS_PRINT ) as mock_print :
262+ expected_hub_token = '359b67d3-fdd7-4e90-a891-b909734fb64a'
263+ hub_mock = MagicMock ()
264+ hub_mock .hub_id = expected_hub_token
265+ hub_cls .return_value = hub_mock
266+
267+ util = APSConnectUtil ()
268+ util .hub_token ()
269+ mock_print .assert_called_with (expected_hub_token )
0 commit comments