예제 #1
0
    def test_exception_get_environment(self):
        requests_mock = patch(
            'networkapiclient.ClientFactory.ApiEnvironment.get').start()
        requests_mock.side_effect = NetworkAPIClientError('')

        napi = NetworkAPI()

        data = napi.get_environment(3)
        self.assertEqual(data, [])
예제 #2
0
    def test_get_environment(self):
        """Test assert called get environment"""

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

        napi = NetworkAPI()
        data = napi.get_environment(1)
        requests_mock.assert_called_once_with(
            ids=[1], include=['default_vrf__details',
                              'father_environment__basic'])

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