diff --git a/manage.py b/manage.py index ace5935..c233703 100755 --- a/manage.py +++ b/manage.py @@ -246,7 +246,7 @@ def configure(self) -> None: # API token already exists - ask if they want to update it. while True: update_api_token = self._get_input( - 'Change the CarbonBlack API token [yes/no]?', 'no' + 'Change the CarbonBlack API token?', 'no' ) if update_api_token in {'yes', 'no'}: break diff --git a/tests/manage_test.py b/tests/manage_test.py index b5ee111..96ea710 100644 --- a/tests/manage_test.py +++ b/tests/manage_test.py @@ -145,9 +145,9 @@ def test_configure_with_defaults( mock_input.assert_has_calls([ mock.call('AWS Region (us-test-1): '), mock.call('Unique name prefix, e.g. "company_team" (test_prefix): '), - mock.call('Enable the CarbonBlack downloader [yes/no]? (yes): '), + mock.call('Enable the CarbonBlack downloader? (yes): '), mock.call('CarbonBlack URL (https://cb-example.com): '), - mock.call('Change the CarbonBlack API token [yes/no]? (no): ') + mock.call('Change the CarbonBlack API token? (no): ') ]) # Verify that the configuration has changed. @@ -171,7 +171,7 @@ def test_configure_with_no_defaults( mock_input.assert_has_calls([ mock.call('AWS Region: '), mock.call('Unique name prefix, e.g. "company_team": '), - mock.call('Enable the CarbonBlack downloader [yes/no]? (no): '), + mock.call('Enable the CarbonBlack downloader? (no): '), mock.call('CarbonBlack URL: '), ]) diff --git a/tests/rules/compile_rules_test.py b/tests/rules/compile_rules_test.py index 5742540..a3c5b73 100644 --- a/tests/rules/compile_rules_test.py +++ b/tests/rules/compile_rules_test.py @@ -39,10 +39,10 @@ def test_find_yara_rules_mixed_case(self): def test_find_yara_rules_skip_other_files(self): """Non-YARA files are skipped during the traversal.""" + self.fs.CreateFile('/rules/clone_rules.py') self.fs.CreateFile('/rules/compile_rules.py') self.fs.CreateFile('/rules/eicar.yar') self.fs.CreateFile('/rules/README.md') - self.fs.CreateFile('/rules/clone_rules.py') self.assertEqual(['eicar.yar'], self._sorted_find())