本文整理匯總了Python中ripe.atlas.tools.commands.probe_search.Command類的典型用法代碼示例。如果您正苦於以下問題：Python Command類的具體用法？Python Command怎麽用？Python Command使用的例子？那麽, 這裏精選的類代碼示例或許可以為您提供幫助。
在下文中一共展示了Command類的15個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: test_render_without_aggregation_with_limit
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)示例2: test_render_without_aggregation
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)示例3: test_arg_with_no_value
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()示例4: test_with_random_args
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()示例5: test_status_arg_wrong_value
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()示例6: test_render_with_aggregation_with_limit
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)示例7: test_center_arg
def test_center_arg(self):
"""User passed center arg"""
cmd = Command()
cmd.init_args(["--center", "1,2"])
self.assertEquals(
cmd.build_request_args(),
{"radius": "1,2:15"}
)示例8: test_location_google_wrong_output
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()示例9: test_location_arg
def test_location_arg(self):
"""User passed location arg"""
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 = {"results": [
{"geometry": {"location": {"lat": 1, "lng": 2}}}]}
cmd = Command()
cmd.init_args(["--location", "blaaaa"])
self.assertEquals(cmd.build_request_args(), {'radius': '1,2:15'})示例10: test_asn_args
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()示例11: test_sane_args1
def test_sane_args1(self):
"""User passed several arguments (1)"""
cmd = Command()
cmd.init_args([
"--center", "1,2",
"--radius", "4",
"--asnv4", "3333",
"--prefix", "193.0.0.0/21"
])
self.assertEquals(
cmd.build_request_args(),
{'asn_v4': 3333, 'prefix': '193.0.0.0/21', 'radius': '1,2:4'}
)示例12: test_render_ids_only
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")示例13: test_sane_args3
def test_sane_args3(self):
"""User passed several arguments (3)"""
cmd = Command()
cmd.init_args([
"--center", "1,2",
"--asnv6", "3333",
"--prefixv6", "2001:67c:2e8::/48"
])
self.assertEquals(cmd.build_request_args(), {
'asn_v6': 3333,
'prefix_v6': '2001:67c:2e8::/48',
'radius': '1,2:15'
})示例14: test_location_google_breaks
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()示例15: test_prefix_args
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()