class TestMongo(unittest.TestCase):
def setUp(self):
Configuration.FILEPATH = 'test/resources/conf/mongofs.json'
self.obj = Mongo()
GenericFile.mongo = self.obj
GenericFile.configuration = Configuration()
self.utils = Utils(mongo=self.obj)
self.utils.load_files()
def tearDown(self):
self.obj.clean_database()
def test_load_generic_file_file(self):
self.assertIsInstance(self.obj.load_generic_file(self.utils.file_raw), File)
def test_load_generic_file_directory(self):
self.assertIsInstance(self.obj.load_generic_file(self.utils.directory_raw), Directory)
def test_load_generic_file_symbolic_link(self):
self.assertIsInstance(self.obj.load_generic_file(self.utils.symbolic_link_raw), SymbolicLink)
def test_current_user(self):
user = self.obj.current_user()
self.assertIsInstance(user['uid'], int)
self.assertIsInstance(user['gid'], int)
self.assertIsInstance(user['pid'], int)
def test_lock_id(self):
filepath = 'test-file'
hostname = Mongo.configuration.hostname()
pid = self.obj.current_user()['pid']
expected_lock = filepath+';'+str(pid)+';'+hostname
self.assertEqual(self.obj.lock_id(filepath=filepath), expected_lock)
def test_master_lock_id(self):
filepath = 'test-file'
hostname = Mongo.configuration.hostname()
pid = 0
expected_lock = filepath + ';' + str(pid) + ';' + hostname
self.assertEqual(self.obj.master_lock_id(filepath=filepath), expected_lock)
def test_create_generic_file(self):
self.utils.insert_file()
gf = self.utils.files_coll.find_one({'directory_id':self.utils.file.directory_id,'filename':self.utils.file.filename},{'uploadDate':False})
self.assertEqual(json_util.dumps(gf), json_util.dumps(self.utils.file_raw))
def test_remove_generic_file(self):
self.utils.insert_file()
self.obj.remove_generic_file(generic_file=self.utils.file)
gf = self.utils.files_coll.find_one({'filename': self.utils.file.filename})
self.assertEqual(gf, None)
def test_remove_generic_file_directory_not_empty(self):
# Try to delete the parent directory while a file still exist in it
self.utils.insert_directory()
self.obj.create_generic_file(generic_file=self.utils.directory_file)
try:
self.obj.remove_generic_file(generic_file=self.utils.directory)
self.assertTrue(False, msg="It was possible to remove a directory while it was still containing files.")
except FuseOSError as e:
self.assertTrue(True)
def test_remove_generic_file_directory_empty(self):
# Try to delete the parent directory after deleting the file in it
self.utils.insert_directory()
self.utils.insert_directory_file()
self.obj.remove_generic_file(generic_file=self.utils.directory_file)
self.obj.remove_generic_file(generic_file=self.utils.directory)
self.assertTrue(True)
def test_list_generic_files_in_directory(self):
self.utils.insert_directory()
self.utils.insert_file()
self.utils.insert_symbolic_link()
files = self.obj.list_generic_files_in_directory(filepath='/')
self.assertEqual(len(files), 3)
def test_generic_file_exists(self):
self.assertFalse(self.obj.generic_file_exists(self.utils.file.filepath))
self.utils.insert_file()
self.assertTrue(self.obj.generic_file_exists(self.utils.file.filepath))
def test_get_generic_file(self):
self.utils.insert_file()
gf = self.obj.get_generic_file(filepath=self.utils.file.filepath)
self.assertIsInstance(gf, File)
def test_get_generic_file_take_lock(self):
self.utils.insert_file()
gf = self.obj.get_generic_file(filepath=self.utils.file.filepath, take_lock=True)
self.assertIsInstance(gf, File)
# We are the same owner, so normally, we should still be able to take the file if there is a lock on it.
gf = self.obj.get_generic_file(filepath=self.utils.file.filepath, take_lock=True)
self.assertIsInstance(gf, File)
def test_get_generic_file_missing(self):
gf = self.obj.get_generic_file(filepath=self.utils.file.filepath)
self.assertEqual(gf, None)
def test_add_nlink_directory(self):
# By default, a directory has 2 st_nlink. And by default, the "/" directory always exists.
self.obj.add_nlink_directory(directory_id=self.utils.root_id, value=4)
gf = self.utils.files_coll.find_one({'_id': self.utils.root_id})
self.assertEqual(gf['metadata']['st_nlink'], 6)
def test_read_data(self):
self.utils.insert_file()
self.utils.insert_file_chunks()
message = self.utils.read_file_chunks()
data = self.obj.read_data(file=self.utils.file, offset=0, size=4096)
self.assertEqual(data, message)
data = self.obj.read_data(file=self.utils.file, offset=3, size=4096)
self.assertEqual(data, message[3:])
data = self.obj.read_data(file=self.utils.file, offset=0, size=8)
self.assertEqual(data, message[:8])
data = self.obj.read_data(file=self.utils.file, offset=3, size=8)
self.assertEqual(data, message[3:3+8])
def test_add_data_append(self):
self.utils.insert_file()
self.utils.insert_file_chunks()
message = self.utils.read_file_chunks()
self.obj.add_data(file=self.utils.file, data=b'test', offset=len(message))
modified_message = self.utils.read_file_chunks()
self.assertEqual(modified_message, message+b'test')
def test_add_data_replace(self):
self.utils.insert_file()
self.utils.insert_file_chunks()
message = self.utils.read_file_chunks()
message = list(message)
message[2] = ord('t')
message[3] = ord('h')
message[4] = ord('i')
message[5] = ord('n')
message[6] = ord('g')
message[7] = ord('s')
expected_message = ''.join(map(chr, message))
self.obj.add_data(file=self.utils.file, data=b'things', offset=2)
modified_message = self.utils.read_file_chunks()
formatted_modified_message = ''.join(map(chr, list(modified_message)))
self.assertEqual(formatted_modified_message, expected_message)
def test_truncate(self):
self.utils.insert_file()
self.utils.insert_file_chunks()
message = self.utils.read_file_chunks()
self.obj.truncate(file=self.utils.file, length=6)
modified_message = self.utils.read_file_chunks()
self.assertEqual(modified_message, message[0:6])
def test_truncate_zero(self):
self.utils.insert_file()
self.utils.insert_file_chunks()
message = self.utils.read_file_chunks()
self.obj.truncate(file=self.utils.file, length=0)
modified_message = self.utils.read_file_chunks()
self.assertEqual(modified_message, message[0:0])
def test_rename_generic_file_to(self):
self.utils.insert_file()
self.utils.insert_file_chunks()
message = self.utils.read_file_chunks()
initial_filepath = self.utils.file.filepath
self.obj.rename_generic_file_to(generic_file=self.utils.file, initial_filepath=self.utils.file.filepath, destination_filepath='/rename-test')
destination_message = self.utils.read_file_chunks() # Read chunks is based on the _id, so we can call it
# Normally, the chunks should not change at all, but we never know.
self.assertEqual(destination_message, message)
old_file = self.utils.files_coll.find_one({'directory_id':self.utils.file.directory_id,'filename':initial_filepath.split('/')[-1]})
self.assertEqual(old_file, None)
new_file = self.utils.files_coll.find_one({'directory_id':self.utils.root_id,'filename':'rename-test'})
self.assertNotEqual(new_file, None)
def test_unlock_generic_file(self):
self.utils.insert_file()
gf = self.obj.get_generic_file(filepath=self.utils.file.filepath, take_lock=True)
result = self.obj.unlock_generic_file(filepath=self.utils.file.filepath, generic_file=gf)
self.assertTrue(result)
gf = self.obj.get_generic_file(filepath=self.utils.file.filepath)
self.assertTrue('lock' not in gf.json)
def test_unlock_generic_file_no_lock(self):
# Verify if it does not crash if there is no lock
self.utils.insert_file()
gf = self.obj.get_generic_file(filepath=self.utils.file.filepath, take_lock=False)
result = self.obj.unlock_generic_file(filepath=self.utils.file.filepath, generic_file=gf)
self.assertTrue(result)
def test_basic_save(self):
self.utils.insert_file()
self.obj.basic_save(generic_file=self.utils.file, metadata={'st_nlink':1}, attrs={'thing':1})
result = self.utils.files_coll.find_one({'_id':self.utils.file._id})
self.assertTrue('st_nlink' in result['metadata'] and len(result['metadata']) == 1)
self.assertTrue('thing' in result['attrs'] and len(result['attrs']) == 1)
class TestGenericFile(unittest.TestCase):
def setUp(self):
Configuration.FILEPATH = 'test/resources/conf/mongofs.json'
self.mongo = Mongo(do_clean_up=True)
GenericFile.mongo = self.mongo
GenericFile.configuration = Configuration()
self.utils = Utils(mongo=self.mongo)
self.utils.load_files()
def tearDown(self):
self.mongo.clean_database()
def test_basic_save(self):
self.utils.insert_file()
initial_gf = self.utils.files_coll.find_one(
{
'directory_id': self.utils.root_id,
'filename': self.utils.file.filename
}, {'uploadDate': False})
self.utils.file.metadata['st_nlink'] = 37
self.utils.file.basic_save()
modified_gf = self.utils.files_coll.find_one(
{
'directory_id': self.utils.root_id,
'filename': self.utils.file.filename
}, {'uploadDate': False})
self.assertEqual(modified_gf['metadata']['st_nlink'], 37)
# We do not test is_dir / is_file / is_link here as we cannot load a GenericFile only, so the test would have no purpose.
# We need to test those methods in test_File, test_Directory and test_SymbolicLink.
def test_rename_to(self):
self.utils.insert_file()
initial_filename = self.utils.file.filename
self.utils.file.rename_to(initial_filepath=self.utils.file.filepath,
destination_filepath='/rename-file')
old_file = self.utils.files_coll.find_one({
'directory_id':
self.utils.file.directory_id,
'filename':
initial_filename
})
self.assertEqual(old_file, None)
new_file = self.utils.files_coll.find_one({
'directory_id': self.utils.root_id,
'filename': 'rename-file'
})
self.assertNotEqual(new_file, None)
def test_release(self):
self.utils.insert_file()
gf = self.mongo.get_generic_file(filepath=self.utils.file.filepath,
lock={'type': GenericFile.LOCK_WRITE})
self.assertTrue('lock' in gf.json)
gf.release(filepath=self.utils.file.filepath)
gf = self.mongo.get_generic_file(filepath=self.utils.file.filepath)
self.assertTrue('lock' not in gf.json or len(gf.json['lock']) == 0)
def test_new_generic_file_file(self):
# Creation of a basic file
GenericFile.new_generic_file(filepath=self.utils.file.filepath,
mode=0o755,
file_type=GenericFile.FILE_TYPE)
inserted_file = self.utils.files_coll.find_one({
'directory_id':
self.utils.file.directory_id,
'filename':
self.utils.file.filename
})
self.assertEqual(inserted_file['filename'], self.utils.file.filename)
self.assertEqual(inserted_file['metadata']['st_mode'],
(S_IFREG | 0o755))
self.assertEqual(inserted_file['generic_file_type'],
GenericFile.FILE_TYPE)
def test_new_generic_file_directory(self):
# Creation of a directory
GenericFile.new_generic_file(filepath=self.utils.directory.filepath,
mode=0o755,
file_type=GenericFile.DIRECTORY_TYPE)
inserted_file = self.utils.files_coll.find_one({
'directory_id':
self.utils.directory.directory_id,
'filename':
self.utils.directory.filename
})
self.assertEqual(inserted_file['filename'],
self.utils.directory.filename)
self.assertEqual(inserted_file['metadata']['st_mode'],
(S_IFDIR | 0o755))
self.assertEqual(inserted_file['generic_file_type'],
GenericFile.DIRECTORY_TYPE)
def test_new_generic_file_symbolic_link(self):
# Creation of a symbolic link to the initial self.utils.file just below
self.utils.insert_file()
GenericFile.new_generic_file(
filepath=self.utils.symbolic_link.filepath,
mode=0o755,
file_type=GenericFile.SYMBOLIC_LINK_TYPE,
target=self.utils.file.filename)
inserted_file = self.utils.files_coll.find_one({
'directory_id':
self.utils.symbolic_link.directory_id,
'filename':
self.utils.symbolic_link.filename
})
self.assertEqual(inserted_file['filename'],
self.utils.symbolic_link.filename)
self.assertEqual(inserted_file['metadata']['st_mode'],
(S_IFLNK | 0o755))
self.assertEqual(inserted_file['generic_file_type'],
GenericFile.SYMBOLIC_LINK_TYPE)
self.assertEqual(inserted_file['target'], self.utils.file.filename)
def test_get_directory_id(self):
self.utils.insert_directory()
self.utils.insert_directory_file()
directory_id = GenericFile.get_directory_id(
filepath=self.utils.directory_file.filepath)
self.assertEqual(directory_id, self.utils.directory_file.directory_id)
def test_is_generic_filename_available(self):
self.utils.insert_file()
is_available = GenericFile.is_generic_filepath_available(
filepath=self.utils.file.filepath)
self.assertFalse(is_available)
is_available = GenericFile.is_generic_filepath_available(
filepath=self.utils.file.filepath + '.something')
self.assertTrue(is_available)