def test_broken_configuration(self):
fd, fn = tempfile.mkstemp()
with open(fn, 'w') as f:
f.write('[agent]\nupdate_frequency = "qwe"')
sys.argv = ['pyca', '-c', fn, 'fail']
try:
__main__.main()
except BaseException as e:
assert e.code == 4
os.close(fd)
os.remove(fn)
def test_broken_configuration_type(self):
configs = ('[capture]\nflavors = "x/source"\nfiles = a, b',
'[agent]\nupdate_frequency = "nan"')
for config in configs:
fd, fn = tempfile.mkstemp()
with open(fn, 'w') as f:
f.write(config)
sys.argv = ['pyca', '-c', fn, 'fail']
try:
__main__.main()
except BaseException as e:
assert e.code == 4
os.close(fd)
os.remove(fn)
def test_run(self):
for mod in (agentstate, capture, ingest, schedule):
mod.run = should_fail
sys.argv = ['pyca', mod.__name__.split('.')[-1]]
try:
__main__.main()
assert False
except ShouldFailException:
assert True
# Test ui start
ui.app.run = should_fail
sys.argv = ['pyca', 'ui']
try:
__main__.main()
assert False
except ShouldFailException:
assert True
# Test run all
for mod in (agentstate, capture, ingest, schedule):
mod.run = lambda: True
sys.argv = ['pyca', 'run']
try:
__main__.main()
except:
assert False
for mod in (agentstate, capture, ingest, schedule):
reload(mod)
def test_run(self):
for mod in (agentstate, capture, ingest, schedule):
mod.run = should_fail
sys.argv = ['pyca', mod.__name__.split('.')[-1]]
try:
__main__.main()
assert False
except ShouldFailException:
assert True
# Test ui start
ui.app.run = should_fail
sys.argv = ['pyca', 'ui']
try:
__main__.main()
assert False
except ShouldFailException:
assert True
# Test run all
for mod in (agentstate, capture, ingest, schedule):
mod.run = lambda: True
sys.argv = ['pyca', 'run']
try:
__main__.main()
except:
assert False
for mod in (agentstate, capture, ingest, schedule):
reload(mod)
def test_help(self):
# test scenarios which end in the usage being printed
sys.argv = ['pyca', '-h']
try:
__main__.main()
except BaseException as e:
assert e.code == 0
sys.argv = ['pyca', '-x']
try:
__main__.main()
except BaseException as e:
assert e.code == 1
sys.argv = ['pyca', 'too', 'many', 'arguments']
try:
__main__.main()
except BaseException as e:
assert e.code == 2
sys.argv = ['pyca', 'fail']
try:
__main__.main()
except BaseException as e:
assert e.code == 3
def test_help(self):
# test scenarios which end in the usage being printed
sys.argv = ['pyca', '-h']
try:
__main__.main()
except BaseException as e:
self.assertEqual(e.code, 0)
sys.argv = ['pyca', '-x']
try:
__main__.main()
except BaseException as e:
self.assertEqual(e.code, 1)
sys.argv = ['pyca', 'too', 'many', 'arguments']
try:
__main__.main()
except BaseException as e:
self.assertEqual(e.code, 2)
sys.argv = ['pyca', 'fail']
try:
__main__.main()
except BaseException as e:
self.assertEqual(e.code, 3)
def test_help(self):
# test scenarios which end in the usage being printed
sys.argv = ['pyca', '-h']
try:
__main__.main()
except BaseException as e:
assert e.code == 0
sys.argv = ['pyca', '-x']
try:
__main__.main()
except BaseException as e:
assert e.code == 1
sys.argv = ['pyca', 'too', 'many', 'arguments']
try:
__main__.main()
except BaseException as e:
assert e.code == 2
sys.argv = ['pyca', 'fail']
try:
__main__.main()
except BaseException as e:
assert e.code == 3
def test_run(self):
for mod in (agentstate, capture, ingest, schedule):
mod.run = should_fail
sys.argv = ['pyca', mod.__name__.split('.')[-1]]
with self.assertRaises(ShouldFailException):
__main__.main()
# Test ui start
ui.app.run = should_fail
sys.argv = ['pyca', 'ui']
with self.assertRaises(ShouldFailException):
__main__.main()
# Test run all
for mod in (agentstate, capture, ingest, schedule):
mod.run = lambda: True
sys.argv = ['pyca', 'run']
try:
__main__.main()
except Exception:
self.fail()
for mod in (agentstate, capture, ingest, schedule):
reload(mod)