Skip to content
Snippets Groups Projects
Commit c2db7446 authored by vlorentz's avatar vlorentz
Browse files

Update CommonContentIndexerTest to work with non-mocked storage.

parent fb34e1aa
No related branches found
No related tags found
1 merge request!433Remove mocks from range tests of mimetype and license indexers.
......@@ -166,8 +166,7 @@ class IndexerStorage:
if conflict_update:
item_ctags = []
else:
# TODO: this merges old ctags with new ctags. This is
# pointless, new ctags should replace the old ones.
# merge old ctags with new ctags
existing = list(self._content_ctags.get([item['id']]))
item_ctags = [
{
......
......@@ -107,8 +107,13 @@ class TestCtagsIndexer(CommonContentIndexerTest, unittest.TestCase):
- Unknown sha1 in the input list are not indexed
"""
def get_indexer_results(self, ids):
yield from self.idx_storage.content_ctags_get(ids)
def setUp(self):
self.indexer = CtagsIndexerTest()
self.idx_storage = self.indexer.idx_storage
# Prepare test input
self.id0 = '01c9379dfc33803963d07c1ccc748d3fe4c96bb5'
......
......@@ -659,7 +659,13 @@ class CommonIndexerWithErrorsTest:
class CommonContentIndexerTest:
def assert_results_ok(self, actual_results, expected_results=None):
def get_indexer_results(self, ids):
"""Override this for indexers that don't have a mock storage."""
return self.indexer.idx_storage.state
def assert_results_ok(self, sha1s, expected_results=None):
actual_results = self.get_indexer_results(sha1s)
if expected_results is None:
expected_results = self.expected_results
......@@ -678,15 +684,12 @@ class CommonContentIndexerTest:
# when
self.indexer.run(sha1s, policy_update='update-dups')
actual_results = self.indexer.idx_storage.state
self.assertTrue(self.indexer.idx_storage.conflict_update)
self.assert_results_ok(actual_results)
self.assert_results_ok(sha1s)
# 2nd pass
self.indexer.run(sha1s, policy_update='ignore-dups')
self.assertFalse(self.indexer.idx_storage.conflict_update)
self.assert_results_ok(actual_results)
self.assert_results_ok(sha1s)
def test_index_one_unknown_sha1(self):
"""Unknown sha1 are not indexed"""
......@@ -696,14 +699,13 @@ class CommonContentIndexerTest:
# when
self.indexer.run(sha1s, policy_update='update-dups')
actual_results = self.indexer.idx_storage.state
# then
expected_results = {
k: v for k, v in self.expected_results.items() if k in sha1s
}
self.assert_results_ok(actual_results, expected_results)
self.assert_results_ok(sha1s, expected_results)
class CommonContentIndexerRangeTest:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment