def test_database(self):
db = Database(":memory:")
db.init_tables()
db.store(FileInfo.from_file("tests/test.txt"))
db.store(FileInfo.from_file("tests/symlink.lnk"))
fileinfo = db.get_by_path(os.path.abspath("tests/test.txt"))
self.assertEqual(fileinfo.path, os.path.abspath("tests/test.txt"))
def generate_fileinfos(path,
relative=False,
prefix=None,
onerror=None,
checksums=False):
for p in generate_files(path=path, onerror=onerror):
try:
fileinfo = FileInfo.from_file(p,
checksums=checksums,
relative=relative)
if prefix is not None:
fileinfo.path = os.path.join(prefix, fileinfo.path)
yield fileinfo
except:
if onerror is not None:
onerror(sys.exc_info()[1])
def test_from_file(self):
fileinfo = FileInfo.from_file("tests/test.txt")
self.assertEqual(11, fileinfo.size)
self.assertEqual("6df4d50a41a5d20bc4faad8a6f09aa8f", fileinfo.blob.md5)
self.assertEqual("bc9faaae1e35d52f3dea9651da12cd36627b8403", fileinfo.blob.sha1)
