def _build_regex(self, parsed_args): regex = '' if parsed_args.smoke: regex = 'smoke' elif parsed_args.regex: regex = parsed_args.regex if parsed_args.whitelist_file or parsed_args.blacklist_file: regex = regex_builder.construct_regex(parsed_args.blacklist_file, parsed_args.whitelist_file, regex, False) return regex
def test_blacklist_regex_without_comments(self): with io.StringIO() as blacklist_file: for i in range(4): blacklist_file.write(u'fake_regex_%s\n' % i) blacklist_file.seek(0) with mock.patch('six.moves.builtins.open', return_value=blacklist_file): result = os_testr.construct_regex( 'fake_path', None, None, False) self.assertEqual(result, "^(?!fake_regex_3|fake_regex_2|" "fake_regex_1|fake_regex_0).*().*$")
def test_whitelist_regex_with_comments(self): with io.StringIO() as whitelist_file: for i in range(4): whitelist_file.write(u'fake_regex_%s # A Comment\n' % i) whitelist_file.seek(0) with mock.patch('six.moves.builtins.open', return_value=whitelist_file): result = os_testr.construct_regex( None, 'fake_path', None, False) self.assertEqual( result, "^.*(fake_regex_0|fake_regex_1|fake_regex_2|fake_regex_3).*$")
def test_whitelist_regex_without_comments_with_regex_passthrough(self): file_contents = u"""regex_a regex_b""" with io.StringIO() as whitelist_file: whitelist_file.write(file_contents) whitelist_file.seek(0) with mock.patch('six.moves.builtins.open', return_value=whitelist_file): result = os_testr.construct_regex(None, 'fake_path', 'fake_regex', False) expected_regex = '^.*(fake_regex|regex_a|regex_b).*$' self.assertEqual(result, expected_regex)
def test_blacklist_whitelist_and_regex_passthrough_at_once(self): with io.StringIO() as blacklist_file, io.StringIO() as whitelist_file: for i in range(4): blacklist_file.write(u'fake_regex_%s\n' % i) blacklist_file.seek(0) whitelist_file.write(u'regex_a\n') whitelist_file.write(u'regex_b\n') whitelist_file.seek(0) with mock.patch('six.moves.builtins.open', side_effect=[blacklist_file, whitelist_file]): result = os_testr.construct_regex('fake_path_1', 'fake_path_2', 'fake_regex', False) expected_regex = ( "^(?!fake_regex_3|fake_regex_2|fake_regex_1|" "fake_regex_0).*(fake_regex|regex_a|regex_b).*$") self.assertEqual(result, expected_regex)
def test_blacklist_regex_without_comment_print_skips(self, print_mock): with io.StringIO() as blacklist_file: for i in range(4): blacklist_file.write(u'fake_regex_%s\n' % i) blacklist_file.seek(0) with mock.patch('six.moves.builtins.open', return_value=blacklist_file): result = os_testr.construct_regex('fake_path', None, None, True) expected_regex = ("^(?!fake_regex_3|fake_regex_2|" "fake_regex_1|fake_regex_0).*().*$") self.assertEqual(result, expected_regex) calls = print_mock.mock_calls self.assertEqual(len(calls), 4) args = list(map(lambda x: x[1], calls)) self.assertIn(('fake_regex_0', ''), args) self.assertIn(('fake_regex_1', ''), args) self.assertIn(('fake_regex_2', ''), args) self.assertIn(('fake_regex_3', ''), args)
def test_regex_passthrough(self): result = os_testr.construct_regex(None, None, 'fake_regex', False) self.assertEqual(result, '^.*(fake_regex).*$')