Add chunk relate impl and test case

This commit is contained in:
liyaxing
2023-09-07 20:19:18 +08:00
parent 188595beb6
commit 1e5b9afd0f
5 changed files with 161 additions and 2 deletions
+36
View File
@@ -0,0 +1,36 @@
import sys
import os
dir_path = os.path.dirname(os.path.realpath(__file__))
print(dir_path)
sys.path.append("{}/../src/".format(dir_path))
print(sys.path)
from knowledge import ChunkTracker, ChunkID, HashValue, PositionType
import asyncio
import unittest
class TestChunk(unittest.TestCase):
def test_chunk_tracker(self):
tracker = ChunkTracker()
hash = HashValue.hash_data("1234567890".encode("utf-8"));
cid = ChunkID.new_chunk_id(hash)
print(cid)
tracker.add_position(cid, "/tmp/1", PositionType.File)
ret = tracker.get_position(cid)
print(ret[0])
tracker.remove_position(cid)
ret = tracker.get_position(cid)
self.assertEqual(ret, None)
if __name__ == "__main__":
unittest.main()