def test_arg_with_no_value(self):
"""User passed not boolean arg but no value"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asn"])
cmd.run()
def test_arg_with_no_value(self):
"""User passed not boolean arg but no value"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asn"])
cmd.run()
def test_with_random_args(self):
"""User passes random args, should fail with SystemExit"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["blaaaaaaa"])
cmd.run()
def test_render_with_aggregation(self):
"""Tests rendering of results with aggregation"""
cmd = Command()
cmd.init_args([
"--country", "GR", "--aggregate-by", "country", "--aggregate-by",
"asn_v4", "--aggregate-by", "prefix_v4"
])
with capture_sys_output() as (stdout, stderr):
path = '{}.ProbeRequest'.format(COMMAND_MODULE)
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
expected_blob = [
"", "Filters:", " Country: GR", "",
" ID Asn_v4 Asn_v6 Country Status ",
"==============================================",
"Country: DE", " ASN_V4: 3332", " PREFIX_V4: 193.0/22",
str(b" 3 3332 de None "),
" ASN_V4: 3333", " PREFIX_V4: 193.0/22",
str(b" 2 3333 de None "), "",
"Country: GR", " ASN_V4: 3333", " PREFIX_V4: 193.0/22",
str(b" 1 3333 gr None "),
str(b" 5 3333 gr None "), "",
"Country: NL", " ASN_V4: 3333", " PREFIX_V4: 193.0/22",
str(b" 4 3333 nl None "),
"==============================================",
" Showing 4 of 4 total probes", ""
]
self.maxDiff = None
out = stdout.getvalue()
expected_set = set(expected_blob)
returned_set = set(out.split("\n"))
self.assertEquals(returned_set, expected_set)
def test_render_with_aggregation_with_limit(self):
"""Tests rendering of results with aggregation with limit"""
cmd = Command()
cmd.init_args([
"--country", "GR",
"--aggregate-by", "country",
"--aggregate-by", "asn_v4",
"--aggregate-by", "prefix_v4",
"--limit", "1"
])
with capture_sys_output() as (stdout, stderr):
path = '{}.ProbeRequest'.format(COMMAND_MODULE)
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
expected_output = (
"\n"
"Filters:\n"
" Country: GR\n"
"\n"
" ID Asn_v4 Asn_v6 Country Status \n"
"==============================================\n"
"Country: GR\n"
" ASN_V4: 3333\n"
" PREFIX_V4: 193.0/22\n"
" 1 3333 gr None \n"
"==============================================\n"
" Showing 1 of 4 total probes\n"
"\n"
)
expected_set = set(expected_output.split("\n"))
returned_set = set(stdout.getvalue().split("\n"))
self.assertEquals(returned_set, expected_set)
def test_status_arg_wrong_value(self):
"""User passed status arg with wrong value"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--status", "4"])
cmd.run()
def test_render_without_aggregation(self):
"""Tests rendering of results without aggregation"""
cmd = Command()
cmd.init_args([
"--country", "GR"
])
with capture_sys_output() as (stdout, stderr):
path = '{}.ProbeRequest'.format(COMMAND_MODULE)
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
expected_output = (
"\n"
"Filters:\n"
" Country: GR\n"
"\n"
"ID Asn_v4 Asn_v6 Country Status \n"
"===========================================\n"
"1 3333 gr None \n"
"2 3333 de None \n"
"3 3332 de None \n"
"4 3333 nl None \n"
"5 3333 gr None \n"
"===========================================\n"
" Showing 4 of 4 total probes\n"
"\n"
)
self.assertEquals(stdout.getvalue(), expected_output)
def test_render_without_aggregation_with_limit(self):
"""Tests rendering of results without aggregation but with limit"""
cmd = Command()
cmd.init_args([
"--country", "GR",
"--limit", "2"
])
with capture_sys_output() as (stdout, stderr):
path = 'ripe.atlas.tools.commands.probe_search.ProbeRequest'
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
expected_output = (
"\n"
"Filters:\n"
" Country: GR\n"
"\n"
"ID Asn_v4 Asn_v6 Country Status \n"
"===========================================\n"
"1 3333 gr None \n"
"2 3333 de None \n"
"===========================================\n"
" Showing 2 of 4 total probes\n"
"\n"
)
self.assertEquals(stdout.getvalue(), expected_output)
def test_status_arg_wrong_value(self):
"""User passed status arg with wrong value"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--status", "4"])
cmd.run()
def test_render_without_aggregation(self):
"""Tests rendering of results without aggregation"""
cmd = Command()
cmd.init_args(["--country", "GR"])
with capture_sys_output() as (stdout, stderr):
path = '{}.ProbeRequest'.format(COMMAND_MODULE)
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
expected_output = (
"\n"
"Filters:\n"
" Country: GR\n"
"\n"
"ID Asn_v4 Asn_v6 Country Status \n"
"===========================================\n"
"1 3333 gr None \n"
"2 3333 de None \n"
"3 3332 de None \n"
"4 3333 nl None \n"
"5 3333 gr None \n"
"===========================================\n"
" Showing 4 of 4 total probes\n"
"\n")
self.assertEquals(stdout.getvalue(), expected_output)
def test_with_random_args(self):
"""User passes random args, should fail with SystemExit"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["blaaaaaaa"])
cmd.run()
def test_render_with_aggregation_with_limit(self):
"""Tests rendering of results with aggregation with limit"""
cmd = Command()
cmd.init_args([
"--country", "GR", "--aggregate-by", "country", "--aggregate-by",
"asn_v4", "--aggregate-by", "prefix_v4", "--limit", "1"
])
with capture_sys_output() as (stdout, stderr):
path = '{}.ProbeRequest'.format(COMMAND_MODULE)
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
expected_output = (
"\n"
"Filters:\n"
" Country: GR\n"
"\n"
" ID Asn_v4 Asn_v6 Country Status \n"
"==============================================\n"
"Country: GR\n"
" ASN_V4: 3333\n"
" PREFIX_V4: 193.0/22\n"
" 1 3333 gr None \n"
"==============================================\n"
" Showing 1 of 4 total probes\n"
"\n")
expected_set = set(expected_output.split("\n"))
returned_set = set(stdout.getvalue().split("\n"))
self.assertEquals(returned_set, expected_set)
def test_location_google_wrong_output(self):
"""User passed location arg but google api gave not expected format"""
with mock.patch('requests.get') as mock_get:
mock_get.return_value = requests.Response()
with mock.patch('requests.Response.json') as mock_json:
mock_json.return_value = {"blaaa": "bla"}
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--location", "blaaaa"])
cmd.run()
def test_location_google_wrong_output(self):
"""User passed location arg but google api gave not expected format"""
with mock.patch('requests.get') as mock_get:
mock_get.return_value = requests.Response()
with mock.patch('requests.Response.json') as mock_json:
mock_json.return_value = {"blaaa": "bla"}
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--location", "blaaaa"])
cmd.run()
def test_asn_args(self):
"""User passed asn arg together with asnv4 or asnv6"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--asn", "3333", "--asnv4", "3333"])
cmd.run()
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--asn", "3333", "--asnv6", "3333"])
cmd.run()
def test_render_ids_only_with_limit(self):
"""User passed ids_only arg together with limit, testing rendering"""
cmd = Command()
cmd.init_args(["--ids-only", "--country", "GR", "--limit", "2"])
with capture_sys_output() as (stdout, stderr):
path = '{}.ProbeRequest'.format(COMMAND_MODULE)
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
self.assertEquals(stdout.getvalue(), "1\n2\n")
def test_asn_args(self):
"""User passed asn arg together with asnv4 or asnv6"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--asn", "3333", "--asnv4", "3333"])
cmd.run()
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--asn", "3333", "--asnv6", "3333"])
cmd.run()
def test_render_ids_only(self):
"""User passed ids_only arg, testing rendiring"""
cmd = Command()
cmd.init_args([
"--ids-only", "--country", "GR"
])
with capture_sys_output() as (stdout, stderr):
path = '{}.ProbeRequest'.format(COMMAND_MODULE)
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
self.assertEquals(stdout.getvalue(), "1\n2\n3\n4\n5\n")
def test_prefix_args(self):
"""User passed prefix arg together with prefixv4 or prefixv6"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(
["--prefix", "193.0.0.0/21", "--prefixv4", "193.0.0.0/21"])
cmd.run()
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args([
"--prefix", "2001:67c:2e8::/48", "--prefixv6",
"2001:67c:2e8::/48"
])
cmd.run()
def test_location_google_breaks(self):
"""User passed location arg but google api gave error"""
caught_exceptions = [
requests.ConnectionError, requests.HTTPError, requests.Timeout]
with mock.patch('requests.get') as mock_get:
for exception in caught_exceptions:
mock_get.side_effect = exception
with capture_sys_output():
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--location", "blaaaa"])
cmd.run()
mock_get.side_effect = Exception()
with self.assertRaises(Exception):
cmd = Command()
cmd.init_args(["--location", "blaaaa"])
cmd.run()
def test_prefix_args(self):
"""User passed prefix arg together with prefixv4 or prefixv6"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args([
"--prefix", "193.0.0.0/21",
"--prefixv4", "193.0.0.0/21"
])
cmd.run()
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args([
"--prefix", "2001:67c:2e8::/48",
"--prefixv6", "2001:67c:2e8::/48"
])
cmd.run()
def test_location_google_breaks(self):
"""User passed location arg but google api gave error"""
caught_exceptions = [
requests.ConnectionError, requests.HTTPError, requests.Timeout
]
with mock.patch('requests.get') as mock_get:
for exception in caught_exceptions:
mock_get.side_effect = exception
with capture_sys_output():
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--location", "blaaaa"])
cmd.run()
mock_get.side_effect = Exception()
with self.assertRaises(Exception):
cmd = Command()
cmd.init_args(["--location", "blaaaa"])
cmd.run()
def test_render_without_aggregation_with_limit(self):
"""Tests rendering of results without aggregation but with limit"""
cmd = Command()
cmd.init_args(["--country", "GR", "--limit", "2"])
with capture_sys_output() as (stdout, stderr):
path = 'ripe.atlas.tools.commands.probe_search.ProbeRequest'
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
self.maxDiff = None
expected_output = [
"", "Filters:", " Country: GR", "",
"ID Asn_v4 Asn_v6 Country Status ",
"===========================================",
str(b"1 3333 gr None "),
str(b"2 3333 de None "),
"===========================================",
" Showing 2 of 4 total probes", ""
]
self.assertEquals(set(stdout.getvalue().split("\n")),
set(expected_output))
def test_arg_with_wrong_type(self):
"""User passed arg with wrong type. e.g string for asn"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asn", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asnv4", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asnv6", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--limit", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--radius", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--tag", "inv@lid"])
cmd.run()
def test_center_arg_wrong_value(self):
"""User passed center arg with wrong value"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--center", "blaaaa"])
cmd.run()
def test_with_empty_args(self):
"""User passes no args, should fail with RipeAtlasToolsException"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args([])
cmd.run()
def test_with_empty_args(self):
"""User passes no args, should fail with RipeAtlasToolsException"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args([])
cmd.run()
def test_arg_with_wrong_type(self):
"""User passed arg with wrong type. e.g string for asn"""
with capture_sys_output():
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asn", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asnv4", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--asnv6", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--limit", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--radius", "blaaaaa"])
cmd.run()
with self.assertRaises(SystemExit):
cmd = Command()
cmd.init_args(["--tag", "inv@lid"])
cmd.run()
def test_center_arg_wrong_value(self):
"""User passed center arg with wrong value"""
with self.assertRaises(RipeAtlasToolsException):
cmd = Command()
cmd.init_args(["--center", "blaaaa"])
cmd.run()
