def setup(self):
self.viptela_device = viptela.Viptela('user',
'pass',
'test',
auto_login=False)
self.viptela_device.put = _mock_http_method
self.viptela_device.post = _mock_http_method
self.viptela_device.delete = _mock_http_method
def test_viptela_login(self):
viptela_device = viptela.Viptela('user',
'pass',
'test',
auto_login=False)
login_result = mock.MagicMock(response=mock.MagicMock(text=''))
post_mock = mock.MagicMock(return_value=login_result)
viptela_device.post = post_mock
assert login_result == viptela_device.login()
def setup(self):
self.session = mock.MagicMock()
self.session.get = mock.MagicMock
self.session.post = mock.MagicMock
self.session.put = mock.MagicMock
self.session.delete = mock.MagicMock
self.viptela_device = viptela.Viptela('user',
'pass',
'test',
auto_login=False)
def test_viptela_disable_warnings_called(self):
viptela.requests.packages.urllib3.disable_warnings = mock.Mock()
# Assigning to a shorter variable for ease of reading.
mock_disable_warnings = viptela.requests.packages.urllib3.disable_warnings
_ = viptela.Viptela('user',
'pass',
'test',
auto_login=False,
disable_warnings=True)
assert mock_disable_warnings.called
def test_viptela_login_bad_creds(self):
viptela_device = viptela.Viptela('user',
'pass',
'test',
auto_login=False)
viptela_device.post = mock.MagicMock(return_value=mock.MagicMock(
response=mock.MagicMock(text='<html></html>')))
with pytest.raises(exceptions.LoginCredentialsError) as excinfo:
viptela_device.login()
assert 'Could not login to device, check user credentials.' == str(
excinfo.value)
def test_viptela_instance_attributes(self):
viptela_device = viptela.Viptela('user',
'pass',
'test',
auto_login=False)
assert viptela_device.user == 'user'
assert viptela_device.user_pass == 'pass'
assert viptela_device.vmanage_server == 'test'
assert viptela_device.vmanage_server_port == 8443
assert viptela_device.verify is False
assert viptela_device.disable_warnings is False
assert viptela_device.timeout == 10
assert viptela_device.auto_login is False
assert viptela_device.base_url == 'https://test:8443/dataservice'
def setup(self):
self.viptela_device = viptela.Viptela('user',
'pass',
'test',
auto_login=False)
self.viptela_device.delete = _mock_http_method
self.viptela_device.get = _mock_http_method
self.mock_get_template_feature = mock.MagicMock(
data={
'data': [{
constants.TEMPLATE_NAME: _MOCK_TEMPLATE_NAME,
constants.TEMPLATE_ID: 0
}, {
constants.TEMPLATE_NAME: _MOCK_TEMPLATE_NAME + '1',
constants.TEMPLATE_ID: 1
}, {
constants.TEMPLATE_NAME: _MOCK_TEMPLATE_NAME + '2',
constants.TEMPLATE_ID: 2
}]
})
self.viptela_device.get_template_feature = mock.MagicMock(
return_value=self.mock_get_template_feature)
def test_viptela_login_with_invalid_vmanage_host_raises_login_timeout_error(
self):
with pytest.raises(LoginTimeoutError) as excinfo:
viptela.Viptela('mock_user', '', 'mock_device', timeout=0.01)
assert 'Could not connect to mock_device.' == str(excinfo.value)
