def test_unauthenticated(self):
args = ["--provider", "aws"]
self.mocked_provider.provider_code = "aws"
self.mocked_provider.authenticate = MagicMock(return_value=False)
code = main(passed_args=args)
unauthenticated_code = 42
assert (code == unauthenticated_code)
def test_empty(self):
args = None
code = None
with patch("sys.stderr", return_value=MagicMock()):
with self.assertRaises(SystemExit):
code = main(passed_args=args)
assert (code is None)
def test_unauthenticated(self):
args = ["aws"]
self.mocked_provider.provider_code = "aws"
self.mocked_provider.authenticate = MagicMock(return_value=False)
args = ScoutSuiteArgumentParser().parse_args(args)
code = main(args)
unauthenticated_code = 42
assert (code == unauthenticated_code)
def test_empty(self):
args = None
code = None
with patch("sys.stderr", return_value=MagicMock()):
with self.assertRaises(SystemExit):
args = ScoutSuiteArgumentParser().parse_args(args)
code = main(args)
assert (code is None)
def test_azure_provider(self):
args = ["azure", "--cli"]
self.mocked_provider.provider_code = "azure"
args = ScoutSuiteArgumentParser().parse_args(args)
code = main(args)
success_code = 0
assert (code == success_code)
report_init_args = self.constructor[self.mocked_report].call_args_list[0][0]
assert (report_init_args[0] == "azure") # provider
assert (report_init_args[1] == "azure") # report_file_name
assert (report_init_args[2] == "scoutsuite-report") # report_dir
def test_gcp_provider(self):
args = ["gcp", "--service-account", "fakecredentials"]
self.mocked_provider.provider_code = "gcp"
args = ScoutSuiteArgumentParser().parse_args(args)
code = main(args)
success_code = 0
assert (code == success_code)
report_init_args = self.constructor[self.mocked_report].call_args_list[0][0]
assert (report_init_args[0] == "gcp") # provider
assert (report_init_args[1] == "gcp") # report_file_name
assert (report_init_args[2] == "scoutsuite-report") # report_dir
def test_keyboardinterrupted(self):
args = ["aws"]
self.mocked_provider.provider_code = "aws"
def _raise(e):
raise e
self.mocked_provider.fetch = MagicMock(side_effect=_raise(KeyboardInterrupt))
args = ScoutSuiteArgumentParser().parse_args(args)
code = main(args)
keyboardinterrupted_code = 130
assert (code == keyboardinterrupted_code)
def test_azure_provider(self):
args = ["--provider", "azure"]
self.mocked_provider.provider_code = "azure"
code = main(passed_args=args)
success_code = 0
assert (code == success_code)
report_init_args = self.constructor[
self.mocked_report].call_args_list[0][0]
assert (report_init_args[0] == "azure") # provider
assert (report_init_args[1] == "azure") # report_file_name
assert (report_init_args[2] == "scoutsuite-report") # report_dir
def test_keyboardinterrupted(self):
args = ["--provider", "aws"]
self.mocked_provider.provider_code = "aws"
def _raise(e):
raise e
self.mocked_provider.fetch = MagicMock(
side_effect=_raise(KeyboardInterrupt))
code = main(passed_args=args)
keyboardinterrupted_code = 130
assert (code == keyboardinterrupted_code)
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
from ScoutSuite.__main__ import main
if __name__ == "__main__":
main()
sys.exit()
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
from ScoutSuite.__main__ import main
if __name__ == "__main__":
sys.exit(main())