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)