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())