# Copyright (C) 2010-2024 RhodeCode GmbH # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License, version 3 # (only), as published by the Free Software Foundation. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . # # This program is dual-licensed. If you wish to learn more about the # RhodeCode Enterprise Edition, including its added features, Support services, # and proprietary license terms, please see https://rhodecode.com/licenses/ import os import mock import pytest from whoosh import query from rhodecode.tests import ( TestController, HG_REPO, TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR_PASS) from rhodecode.tests.routes import route_path class TestSearchController(TestController): def test_index(self): self.log_user() response = self.app.get(route_path('search')) assert_response = response.assert_response() assert_response.one_element_exists('input#q') def test_search_files_empty_search(self): if os.path.isdir(self.index_location): pytest.skip('skipped due to existing index') else: self.log_user() response = self.app.get(route_path('search'), {'q': HG_REPO}) response.mustcontain('There is no index to search in. ' 'Please run whoosh indexer') def test_search_validation(self): self.log_user() response = self.app.get(route_path('search'), {'q': query, 'type': 'content', 'page_limit': 1000}) response.mustcontain( 'page_limit - 1000 is greater than maximum value 500') @pytest.mark.parametrize("query, expected_hits, expected_paths", [ ('todo', 23, [ 'vcs/backends/hg/inmemory.py', 'vcs/tests/test_git.py']), ('extension:rst installation', 6, [ 'docs/index.rst', 'docs/installation.rst']), ('def repo', 87, [ 'vcs/tests/test_git.py', 'vcs/tests/test_changesets.py']), ('repository:%s def test' % HG_REPO, 18, [ 'vcs/tests/test_git.py', 'vcs/tests/test_changesets.py']), ('"def main"', 9, [ 'vcs/__init__.py', 'vcs/tests/__init__.py', 'vcs/utils/progressbar.py']), ('owner:test_admin', 358, [ 'vcs/tests/base.py', 'MANIFEST.in', 'vcs/utils/termcolors.py', 'docs/theme/ADC/static/documentation.png']), ('owner:test_admin def main', 72, [ 'vcs/__init__.py', 'vcs/tests/test_utils_filesize.py', 'vcs/tests/test_cli.py']), ('owner:michał test', 0, []), ]) def test_search_files(self, query, expected_hits, expected_paths): self.log_user() response = self.app.get(route_path('search'), {'q': query, 'type': 'content', 'page_limit': 500}) response.mustcontain('%s results' % expected_hits) for path in expected_paths: response.mustcontain(path) @pytest.mark.parametrize("query, expected_hits, expected_commits", [ ('bother to ask where to fetch repo during tests', 3, [ ('hg', 'a00c1b6f5d7a6ae678fd553a8b81d92367f7ecf1'), ('git', 'c6eb379775c578a95dad8ddab53f963b80894850'), ('svn', '98')]), ('michał', 0, []), ('changed:tests/utils.py', 36, [ ('hg', 'a00c1b6f5d7a6ae678fd553a8b81d92367f7ecf1')]), ('changed:vcs/utils/archivers.py', 11, [ ('hg', '25213a5fbb048dff8ba65d21e466a835536e5b70'), ('hg', '47aedd538bf616eedcb0e7d630ea476df0e159c7'), ('hg', 'f5d23247fad4856a1dabd5838afade1e0eed24fb'), ('hg', '04ad456aefd6461aea24f90b63954b6b1ce07b3e'), ('git', 'c994f0de03b2a0aa848a04fc2c0d7e737dba31fc'), ('git', 'd1f898326327e20524fe22417c22d71064fe54a1'), ('git', 'fe568b4081755c12abf6ba673ba777fc02a415f3'), ('git', 'bafe786f0d8c2ff7da5c1dcfcfa577de0b5e92f1')]), ('added:README.rst', 3, [ ('hg', '3803844fdbd3b711175fc3da9bdacfcd6d29a6fb'), ('git', 'ff7ca51e58c505fec0dd2491de52c622bb7a806b'), ('svn', '8')]), ('changed:lazy.py', 15, [ ('hg', 'eaa291c5e6ae6126a203059de9854ccf7b5baa12'), ('git', '17438a11f72b93f56d0e08e7d1fa79a378578a82'), ('svn', '82'), ('svn', '262'), ('hg', 'f5d23247fad4856a1dabd5838afade1e0eed24fb'), ('git', '33fa3223355104431402a888fa77a4e9956feb3e') ]), ('author:marcin@python-blog.com ' 'commit_id:b986218ba1c9b0d6a259fac9b050b1724ed8e545', 1, [ ('hg', 'b986218ba1c9b0d6a259fac9b050b1724ed8e545')]), ('b986218ba1c9b0d6a259fac9b050b1724ed8e545', 1, [ ('hg', 'b986218ba1c9b0d6a259fac9b050b1724ed8e545')]), ('b986218b', 1, [ ('hg', 'b986218ba1c9b0d6a259fac9b050b1724ed8e545')]), ]) def test_search_commit_messages( self, query, expected_hits, expected_commits, enabled_backends): self.log_user() response = self.app.get(route_path('search'), {'q': query, 'type': 'commit', 'page_limit': 500}) response.mustcontain('%s results' % expected_hits) for backend, commit_id in expected_commits: if backend in enabled_backends: response.mustcontain(commit_id) @pytest.mark.parametrize("query, expected_hits, expected_paths", [ ('readme.rst', 3, []), ('test*', 75, []), ('*model*', 1, []), ('extension:rst', 48, []), ('extension:rst api', 24, []), ]) def test_search_file_paths(self, query, expected_hits, expected_paths): self.log_user() response = self.app.get(route_path('search'), {'q': query, 'type': 'path', 'page_limit': 500}) response.mustcontain('%s results' % expected_hits) for path in expected_paths: response.mustcontain(path) def test_search_commit_message_specific_repo(self, backend): self.log_user() response = self.app.get( route_path('search_repo',repo_name=backend.repo_name), {'q': 'bother to ask where to fetch repo during tests', 'type': 'commit'}) response.mustcontain('1 results') def test_filters_are_not_applied_for_admin_user(self): self.log_user() with mock.patch('whoosh.searching.Searcher.search') as search_mock: search_mock.return_value = mock.MagicMock( scored_length=lambda: 100, runtime=10 ) self.app.get(route_path('search'), {'q': 'test query', 'type': 'commit'}) assert search_mock.call_count == 1 _, kwargs = search_mock.call_args assert kwargs['filter'] is None def test_filters_are_applied_for_normal_user(self, enabled_backends): self.log_user(TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR_PASS) with mock.patch('whoosh.searching.Searcher.search') as search_mock: search_mock.return_value = mock.MagicMock( scored_length=lambda: 100, runtime=10 ) self.app.get(route_path('search'), {'q': 'test query', 'type': 'commit'}) assert search_mock.call_count == 1 _, kwargs = search_mock.call_args assert isinstance(kwargs['filter'], query.Or) expected_repositories = [ f'vcs_test_{b}' for b in enabled_backends] queried_repositories = [ name for type_, name in kwargs['filter'].all_terms()] for repository in expected_repositories: assert repository in queried_repositories