コード例 #1
0
    def test_exception_get_pool_by_member_id(self):
        requests_mock = patch(
            'networkapiclient.ClientFactory.ApiPool.search').start()
        requests_mock.side_effect = NetworkAPIClientError('')

        napi = NetworkAPI()

        data = napi.get_pool_by_member_id(3)
        self.assertEqual(data, [])
コード例 #2
0
    def test_get_pool_by_member_id(self):
        """Test assert called get pool by member"""

        requests_mock = patch(
            'networkapiclient.ClientFactory.ApiPool.search').start()
        requests_mock.return_value = {'server_pools': [{'id': 1}]}

        napi = NetworkAPI()
        data = napi.get_pool_by_member_id(1)
        requests_mock.assert_called_once_with(
            search={'extends_search': [{'serverpoolmember': 1}]}, kind='details')

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