Beispiel #1
0
    def test_exception_get_vip(self):
        requests_mock = patch(
            'networkapiclient.ClientFactory.ApiVipRequest.get').start()
        requests_mock.side_effect = NetworkAPIClientError('')

        napi = NetworkAPI()
        data = napi.get_vip(3)
        self.assertEqual(data, [])
Beispiel #2
0
    def test_get_vip(self):
        """Test assert called get vip"""

        requests_mock = patch(
            'networkapiclient.ClientFactory.ApiVipRequest.get').start()
        requests_mock.return_value = {'vips': [{'id': 1}]}

        napi = NetworkAPI()
        data = napi.get_vip(1)
        requests_mock.assert_called_once_with(ids=[1], kind='details')

        self.assertDictEqual(data, {'id': 1})