def test_run(self, mock_requests): action = self.get_action_instance(self.config_good) expected = [{ "id": "023e105f4ecef8ad9ca31a8372d0c353", "name": "example.com", "type": "full" }] response = requests.Response() response.status_code = 200 response.url = "https://api.cloudflare.com/client/v4/zones" response.headers['Content-Type'] = 'application/json' response._content = MOCK_DATA_SUCCESS mock_session = mock.MagicMock() mock_session.get.return_value = response mock_requests.Session.return_value = mock_session result = action.run() self.assertEqual(result, expected) mock_session.get.assert_called_with( "https://api.cloudflare.com/client/v4/zones", data=None, headers={ 'X-Auth-Email': '*****@*****.**', 'X-Auth-Key': 'API-Key', 'Content-Type': 'application/json', 'User-Agent': user_agent() }, params={})
def test_run(self, mock_requests): action = self.get_action_instance(self.config_good) expected = { 'ipv4_cidrs': [u'199.27.128.0/21'], 'ipv6_cidrs': [u'2400:cb00::/32'] } response = requests.Response() response.status_code = 200 response.url = "https://api.cloudflare.com/client/v4/ips" response.headers['Content-Type'] = 'application/json' response._content = MOCK_DATA_SUCCESS mock_session = mock.MagicMock() mock_session.get.return_value = response mock_requests.Session.return_value = mock_session result = action.run() self.assertEqual(result, expected) mock_session.get.assert_called_with( "https://api.cloudflare.com/client/v4/ips", data=None, headers={ 'Content-Type': 'application/json', 'User-Agent': user_agent() }, params={})
def test_run(self, mock_requests): action = self.get_action_instance(self.config_good) expected = [{ "name": "abc.domain.tld", "content": "1.2.3.4", "zone_name": "domain.tld", "type": "A", "id": "41981e0023c8e1e375ffbc9b35a0fb4e", "zone_id": "023e105f4ecef8ad9ca31a8372d0c353" }, { "name": "xyz.domain.tld", "content": "7.8.9.0", "zone_name": "domain.tld", "type": "A", "id": "bd461aba4b76229c9be2b77e4ef369af", "zone_id": "023e105f4ecef8ad9ca31a8372d0c353" }] response = requests.Response() response.status_code = 200 response.url = "https://api.cloudflare.com/client/v4/zones" response.headers['Content-Type'] = 'application/json' response._content = MOCK_DATA_SUCCESS mock_session = mock.MagicMock() mock_session.get.return_value = response mock_requests.Session.return_value = mock_session result = action.run(zone_id="023e105f4ecef8ad9ca31a8372d0c353") self.assertEqual(result, expected) url = ("https://api.cloudflare.com/client/v4/zones/{}/dns_records". format("023e105f4ecef8ad9ca31a8372d0c353")) mock_session.get.assert_called_with(url, data=None, headers={ 'X-Auth-Email': '*****@*****.**', 'X-Auth-Key': 'API-Key', 'Content-Type': 'application/json', 'User-Agent': user_agent() }, params={})
def index(): response.flash = T("Hello World") return dict(message=T(user_agent()))