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()))