Page MenuHomeSoftware Heritage
Files F7163696

D877.id2806.diff
No OneTemporary
Actions

D877.id2806.diff
View Options

diff --git a/swh/indexer/tests/test_ctags.py b/swh/indexer/tests/test_ctags.py
--- a/swh/indexer/tests/test_ctags.py
+++ b/swh/indexer/tests/test_ctags.py
@@ -3,9 +3,11 @@
# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information
+import json
import unittest
-
from unittest.mock import patch
+
+import swh.indexer.ctags
from swh.indexer.ctags import (
CtagsIndexer, run_ctags
)
@@ -14,7 +16,7 @@
CommonContentIndexerTest,
CommonIndexerWithErrorsTest, CommonIndexerNoTool,
SHA1_TO_CTAGS, NoDiskIndexer, BASE_TEST_CONFIG,
- fill_storage, fill_obj_storage
+ OBJ_STORAGE_DATA, fill_storage, fill_obj_storage
)
@@ -115,6 +117,7 @@
yield from self.idx_storage.content_ctags_get(ids)
def setUp(self):
+ super().setUp()
self.indexer = CtagsIndexerTest()
self.idx_storage = self.indexer.idx_storage
fill_storage(self.indexer.storage)
@@ -132,20 +135,51 @@
self.id0: {
'id': self.id0,
'tool': tool,
- 'ctags': SHA1_TO_CTAGS[self.id0],
+ **SHA1_TO_CTAGS[self.id0][0],
},
self.id1: {
'id': self.id1,
'tool': tool,
- 'ctags': SHA1_TO_CTAGS[self.id1],
+ **SHA1_TO_CTAGS[self.id1][0],
},
self.id2: {
'id': self.id2,
'tool': tool,
- 'ctags': SHA1_TO_CTAGS[self.id2],
+ **SHA1_TO_CTAGS[self.id2][0],
}
}
+ self._set_mocks()
+
+ def _set_mocks(self):
+ def find_ctags_for_content(raw_content):
+ for (sha1, ctags) in SHA1_TO_CTAGS.items():
+ if OBJ_STORAGE_DATA[sha1] == raw_content:
+ return ctags
+ else:
+ raise ValueError(('%r not found in objstorage, can\'t mock '
+ 'its ctags.') % raw_content)
+
+ def fake_language(raw_content, *args, **kwargs):
+ ctags = find_ctags_for_content(raw_content)
+ return {'lang': ctags[0]['lang']}
+ self._real_compute_language = swh.indexer.ctags.compute_language
+ swh.indexer.ctags.compute_language = fake_language
+
+ def fake_subprocess(cmd, *args, **kwargs):
+ print(cmd)
+ id_ = cmd[-1] # when using NoDiskIndexer, path is replaced by id
+ return '\n'.join(
+ json.dumps({'language': ctag['lang'], **ctag})
+ for ctag in SHA1_TO_CTAGS[id_])
+ self._real_subprocess = swh.indexer.ctags.subprocess
+ swh.indexer.ctags.subprocess = fake_subprocess
+
+ def tearDown(self):
+ swh.indexer.ctags.compute_language = self._real_compute_language
+ swh.indexer.ctags.subprocess = self._real_subprocess
+ super().tearDown()
+
class CtagsIndexerUnknownToolTestStorage(
CommonIndexerNoTool, CtagsIndexerTest):

File Metadata

Mime Type
text/plain
Expires
Thu, Jan 30, 1:28 PM (9 h, 3 m ago)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3230331

Event Timeline

About · Developer information · Legal