def test_call_close_success(self, mock_device):
"""
test_call_close_success - Test if the plugin junos plugin close a connection with the expected asserts
"""
# Mocking
mock_device.connected = False
plugin = JUNOS(equipment_access=self.mock_equipment_access)
plugin.remote_conn = mock_device
# Close to a real test:
close_response = plugin.close()
# Asserts
plugin.remote_conn.close.assert_called_once_with()
self.assertTrue(close_response, True)
def test_call_ensure_privilege_level_success(self, mock_start_shell):
"""
test_call_ensure_privilege_level_success
Note: The shell run function expects an array as a return value,
and ensure_privilege_level() parse it to ensure the privilege.
"""
mock_start_shell.return_value.run.return_value = [
False,
u'cli -c "show cli authorization"\r\r\nCurrent user: \'root \' class \'super-user\'\r'
]
plugin = JUNOS(equipment_access=self.mock_equipment_access)
# Mock connection
plugin.remote_conn = MagicMock()
plugin.remote_conn.connected.return_value = True
result = plugin.ensure_privilege_level()
self.assertTrue(result)