def test_invalid_query_stat2(self):
"""Test case where stat returns exception"""
self.mock_get.side_effect = requests.exceptions.RequestException
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
self.mock_get.return_value = FakeErrorResponse()
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
def test_invalid_query_stat2(self):
"""Test case where stat returns exception"""
self.mock_get.side_effect = requests.exceptions.RequestException
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
self.mock_get.return_value = FakeErrorResponse()
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
def test_invalid_query_stat(self):
"""Test case where stat returns not ok status"""
self.mock_get.return_value = FakeResponse(
json_return={"status": "notok"})
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
self.mock_get.return_value = FakeResponse(json_return={})
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
def test_invalid_query_stat(self):
"""Test case where stat returns not ok status"""
self.mock_get.return_value = FakeResponse(
json_return={"status": "notok"}
)
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
self.mock_get.return_value = FakeResponse(
json_return={}
)
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
def test_invalid_query_stat1(self):
"""Test case where stat returns not valid data structure"""
# no data at all
self.mock_get.return_value = FakeResponse(json_return={"status": "ok"})
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# data is dict but no asns key
self.mock_get.return_value = FakeResponse(json_return={
"status": "ok",
"data": {
"bla": "bla"
}
})
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# data is not dict
self.mock_get.return_value = FakeResponse(json_return={
"status": "ok",
"data": [1, 2]
})
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# asns is not list
self.mock_get.return_value = FakeResponse(json_return={
"status": "ok",
"data": {
"asns": {}
}
})
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# asns is empty
self.mock_get.return_value = FakeResponse(json_return={
"status": "ok",
"data": {
"asns": [{}]
}
})
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
def test_valid_query_stat(self):
"""Test case for valid stat response"""
ip = IP(self.IP)
self.assertEqual(
ip.query_stat(), {
'Prefix': '193.0.0.0/21',
'Holder':
'RIPE-NCC-AS Reseaux IP Europeens Network Coordination Centre (RIPE NCC),NL',
'ASN': '3333'
})
def test_valid_query_stat(self):
"""Test case for valid stat response"""
ip = IP(self.IP)
self.assertEqual(
ip.query_stat(),
{
'Prefix': '193.0.0.0/21',
'Holder': 'RIPE-NCC-AS Reseaux IP Europeens Network Coordination Centre (RIPE NCC),NL',
'ASN': '3333'
}
)
def test_invalid_query_stat1(self):
"""Test case where stat returns not valid data structure"""
# no data at all
self.mock_get.return_value = FakeResponse(
json_return={"status": "ok"}
)
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# data is dict but no asns key
self.mock_get.return_value = FakeResponse(
json_return={"status": "ok", "data": {"bla": "bla"}}
)
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# data is not dict
self.mock_get.return_value = FakeResponse(
json_return={"status": "ok", "data": [1, 2]}
)
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# asns is not list
self.mock_get.return_value = FakeResponse(
json_return={"status": "ok", "data": {"asns": {}}}
)
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
# asns is empty
self.mock_get.return_value = FakeResponse(
json_return={"status": "ok", "data": {"asns": [{}]}}
)
ip = IP(self.IP)
self.assertEqual(ip.query_stat(), {})
