Merge pull request #1418 from mind04/bind

[thirdparty/pdns.git] / regression-tests.api / test_RecursorConfig.py

import json
import requests
import unittest
from test_helper import ApiTestCase, isRecursor


@unittest.skipIf(not isRecursor(), "Only applicable to recursors")
class RecursorConfig(ApiTestCase):

    def test_ConfigAllowFromGet(self):
        r = self.session.get(self.url("/servers/localhost/config/allow-from"))
        self.assertSuccessJson(r)

    def test_ConfigAllowFromReplace(self):
        payload = {'value': ["127.0.0.1"]}
        r = self.session.put(
            self.url("/servers/localhost/config/allow-from"),
            data=json.dumps(payload),
            headers={'content-type': 'application/json'})
        self.assertSuccessJson(r)
        data = r.json()
        self.assertEquals("127.0.0.1/32", data["value"][0])

    def test_ConfigAllowFromReplaceError(self):
        """Test the error case, should return 422."""
        payload = {'value': ["abcdefgh"]}
        r = self.session.put(
            self.url("/servers/localhost/config/allow-from"),
            data=json.dumps(payload),
            headers={'content-type': 'application/json'})
        self.assertEquals(r.status_code, 422)
        data = r.json()
        self.assertIn('Unable to convert', data['error'])