本文整理汇总了Python中ripe.atlas.tools.commands.probes.Command类的典型用法代码示例。如果您正苦于以下问题：Python Command类的具体用法？Python Command怎么用？Python Command使用的例子？那么, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了Command类的15个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: 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 = 'ripe.atlas.tools.commands.probes.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"
"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)示例2: 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.probes.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)示例3: 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()示例4: 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()示例5: 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(),
{"latitude": "1", "longitude": "2"}
)示例6: 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()示例7: 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()示例8: 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(), {
"latitude": '1', "longitude": '2'})示例9: 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'}
)示例10: 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 = 'ripe.atlas.tools.commands.probes.ProbeRequest'
with mock.patch(path) as mock_get:
mock_get.return_value = FakeGen()
cmd.run()
self.assertEquals(stdout.getvalue(), "1\n2\n3\n4\n5\n")示例11: 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',
'latitude': '1',
'longitude': '2'
})示例12: 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()示例13: 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()示例14: test_sane_args2
def test_sane_args2(self):
"""User passed several arguments (2)"""
cmd = Command()
cmd.init_args([
"--location", "Amsterdam",
"--asn", "3333",
"--prefixv4", "193.0.0.0/21"
])
path = 'ripe.atlas.tools.commands.probes.Command.location2degrees'
with mock.patch(path) as mock_get:
mock_get.return_value = (1, 2)
self.assertEquals(cmd.build_request_args(), {
'asn': 3333,
'prefix_v4': '193.0.0.0/21',
'latitude': 1,
'longitude': 2
})示例15: test_get_aggregators
def test_get_aggregators(self):
"""User passed --aggregate-by args"""
cmd = Command()
cmd.init_args([
"--aggregate-by", "asn_v4",
"--aggregate-by", "country",
"--aggregate-by", "prefix_v4"
])
expected_output = [
ValueKeyAggregator(key="asn_v4"),
ValueKeyAggregator(key="country_code"),
ValueKeyAggregator(key="prefix_v4")
]
cmd.set_aggregators()
for index, v in enumerate(cmd.aggregators):
self.assertTrue(isinstance(v, ValueKeyAggregator))
self.assertEquals(
v.aggregation_keys,
expected_output[index].aggregation_keys
)