def test_single(self): map = ResultMap() self.assertEqual( map.add("Magnum", "namespaceMagnum.html", suffix_length=11, flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.NAMESPACE)), 0) for i in type_sizes: with self.subTest(**i): serialized = map.serialize(Serializer(**i)) self.compare( Deserializer(**i), serialized, """ 0: Magnum [suffix_length=11, type=NAMESPACE] -> namespaceMagnum.html """) # Verify just the smallest and largest size, everything else # should fit in between. The `result_id_bytes` don't affect # this case. if i['file_offset_bytes'] == 3 and i['name_size_bytes'] == 1: self.assertEqual(len(serialized), 35) elif i['file_offset_bytes'] == 4 and i['name_size_bytes'] == 2: self.assertEqual(len(serialized), 38) else: self.assertGreater(len(serialized), 35) self.assertLess(len(serialized), 38)
def test_empty(self): map = ResultMap() for i in type_sizes: with self.subTest(**i): serialized = map.serialize(Serializer(**i)) self.compare(Deserializer(**i), serialized, "") self.assertEqual(len(serialized), i['file_offset_bytes'])
def test_multiple(self): map = ResultMap() self.assertEqual( map.add("Math", "namespaceMath.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.NAMESPACE)), 0) self.assertEqual( map.add("Math::Vector", "classMath_1_1Vector.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.CLASS)), 1) self.assertEqual( map.add("Math::Range", "classMath_1_1Range.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.CLASS)), 2) self.assertEqual( map.add("Math::min()", "namespaceMath.html#abcdef2875", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.FUNC)), 3) self.assertEqual( map.add("Math::max(int, int)", "namespaceMath.html#abcdef1234", suffix_length=8, flags=ResultFlag.from_type( ResultFlag.DEPRECATED | ResultFlag.DELETED, EntryType.FUNC)), 4) self.assertEqual(map.add("Rectangle", "", alias=2), 5) self.assertEqual( map.add("Rectangle::Rect()", "", suffix_length=2, alias=2), 6) for i in type_sizes: with self.subTest(**i): serialized = map.serialize(Serializer(**i)) self.compare( Deserializer(**i), serialized, """ 0: Math [type=NAMESPACE] -> namespaceMath.html 1: ::Vector [prefix=0[:0], type=CLASS] -> classMath_1_1Vector.html 2: ::Range [prefix=0[:0], type=CLASS] -> classMath_1_1Range.html 3: ::min() [prefix=0[:18], type=FUNC] -> #abcdef2875 4: ::max(int, int) [prefix=0[:18], suffix_length=8, deprecated, deleted, type=FUNC] -> #abcdef1234 5: Rectangle [alias=2] -> 6: ::Rect() [alias=2, prefix=5[:0], suffix_length=2] -> """) # Verify just the smallest and largest size, everything else # should fit in between if i['file_offset_bytes'] == 3 and i[ 'result_id_bytes'] == 2 and i['name_size_bytes'] == 1: self.assertEqual(len(serialized), 202) elif i['file_offset_bytes'] == 4 and i[ 'result_id_bytes'] == 4 and i['name_size_bytes'] == 2: self.assertEqual(len(serialized), 231) else: self.assertGreater(len(serialized), 202) self.assertLess(len(serialized), 231)
def test_24bit_alias_id_too_small(self): map = ResultMap() # The alias doesn't exist of course, hopefully that's fine in this case map.add("B", "", alias=16 * 1024 * 1024) with self.assertRaisesRegex( OverflowError, "Result map alias ID too large to store in 24 bits, set SEARCH_RESULT_ID_BYTES = 4 in your conf.py." ): map.serialize( Serializer(file_offset_bytes=3, result_id_bytes=3, name_size_bytes=1)) # This should work map.serialize( Serializer(file_offset_bytes=3, result_id_bytes=4, name_size_bytes=1))
def test_8bit_suffix_length_too_small(self): map = ResultMap() map.add("F()" + ';' * 256, "f.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.FUNC), suffix_length=256) with self.assertRaisesRegex( OverflowError, "Result map suffix length too large to store in 8 bits, set SEARCH_NAME_SIZE_BYTES = 2 in your conf.py." ): map.serialize( Serializer(file_offset_bytes=3, result_id_bytes=2, name_size_bytes=1)) # This should work map.serialize( Serializer(file_offset_bytes=3, result_id_bytes=2, name_size_bytes=2))
def test_16bit_prefix_id_too_small(self): map = ResultMap() # Adding A0 to A65535 would be too slow due to the recursive Trie # population during prefix merging (SIGH) so trying this instead. It's # still hella slow, but at least not TWO MINUTES. for i in range(128): for j in range(128): for k in range(4): map.add(bytes([i, j, k]).decode('utf-8'), "a.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.CLASS)) self.assertEqual( map.add("B", "b.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.CLASS)), 65536) map.add("B::foo()", "b.html#foo", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.FUNC)) with self.assertRaisesRegex( OverflowError, "Result map prefix ID too large to store in 16 bits, set SEARCH_RESULT_ID_BYTES = 3 in your conf.py." ): map.serialize( Serializer(file_offset_bytes=3, result_id_bytes=2, name_size_bytes=1)) # This should work map.serialize( Serializer(file_offset_bytes=3, result_id_bytes=3, name_size_bytes=1))
def test_single(self): map = ResultMap() self.assertEqual( map.add("Magnum", "namespaceMagnum.html", suffix_length=11, flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.NAMESPACE)), 0) serialized = map.serialize() self.compare( serialized, """ 0: Magnum [suffix_length=11, type=NAMESPACE] -> namespaceMagnum.html """) self.assertEqual(len(serialized), 36)
def test_24bit_file_offset_too_small(self): map = ResultMap() # 3 bytes for the initial offset, 3 bytes for file size, 1 byte for the # flags, 1 byte for the null terminator, 6 bytes for the URL map.add('F' * (16 * 1024 * 1024 - 14), 'f.html', flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.CLASS)) with self.assertRaisesRegex( OverflowError, "Result map offset too large to store in 24 bits, set SEARCH_FILE_OFFSET_BYTES = 4 in your conf.py." ): # Disabling prefix merging otherwise memory usage goes to hell map.serialize(Serializer(file_offset_bytes=3, result_id_bytes=2, name_size_bytes=1), merge_prefixes=False) # This should work. Disabling prefix merging otherwise memory usage # goes to hell. map.serialize(Serializer(file_offset_bytes=4, result_id_bytes=2, name_size_bytes=1), merge_prefixes=False)
def test_multiple(self): map = ResultMap() self.assertEqual( map.add("Math", "namespaceMath.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.NAMESPACE)), 0) self.assertEqual( map.add("Math::Vector", "classMath_1_1Vector.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.CLASS)), 1) self.assertEqual( map.add("Math::Range", "classMath_1_1Range.html", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.CLASS)), 2) self.assertEqual( map.add("Math::min()", "namespaceMath.html#abcdef2875", flags=ResultFlag.from_type(ResultFlag.NONE, EntryType.FUNC)), 3) self.assertEqual( map.add("Math::max(int, int)", "namespaceMath.html#abcdef1234", suffix_length=8, flags=ResultFlag.from_type( ResultFlag.DEPRECATED | ResultFlag.DELETED, EntryType.FUNC)), 4) self.assertEqual(map.add("Rectangle", "", alias=2), 5) self.assertEqual( map.add("Rectangle::Rect()", "", suffix_length=2, alias=2), 6) serialized = map.serialize() self.compare( serialized, """ 0: Math [type=NAMESPACE] -> namespaceMath.html 1: ::Vector [prefix=0[:0], type=CLASS] -> classMath_1_1Vector.html 2: ::Range [prefix=0[:0], type=CLASS] -> classMath_1_1Range.html 3: ::min() [prefix=0[:18], type=FUNC] -> #abcdef2875 4: ::max(int, int) [prefix=0[:18], suffix_length=8, deprecated, deleted, type=FUNC] -> #abcdef1234 5: Rectangle [alias=2] -> 6: ::Rect() [alias=2, prefix=5[:0], suffix_length=2] -> """) self.assertEqual(len(serialized), 203)
def test_empty(self): map = ResultMap() serialized = map.serialize() self.compare(serialized, "") self.assertEqual(len(serialized), 4)