def _test_groupcheckin(self): # FIXME: can't find maya_render.####.iff return file_path = "./maya_render.####.iff" range = FileRange(1, 12) # create new test files expanded_paths = FileGroup.expand_paths(file_path, range) for expanded_path in expanded_paths: file = open(expanded_path, 'w') file.write("file: %s" % expanded_path) file.close() # checkin the frames file_types = ["main"] file_paths = [file_path] checkin = FileGroupCheckin(self.person, file_paths, file_types, range, \ column="frames") checkin.execute() # delete the test files for expanded_path in expanded_paths: os.unlink(expanded_path)
def _test_preallocation_checkin(self): snapshot_type = "file" context = "preallocation" file_name = 'whatever.jpg' file_type = 'jpg' # create an empty snapshot snapshot = Snapshot.create(self.person, snapshot_type=snapshot_type, context=context) # preallocate with no name or type path = snapshot.get_preallocated_path() server = Config.get_value("install", "server") if server: expected = "%s_preallocation_%s_v001" % (self.person.get_code(), server) else: expected = "%s_preallocation_v001" % (self.person.get_code()) self.assertEquals(True, path.endswith(expected)) # preallocate with a file name and file type # since it's meant for FileAppendCheckin, the checkin_type should be 'strict' path = snapshot.get_preallocated_path(file_type, file_name, checkin_type='strict') if server: self.assertEquals( True, None != re.search( 'unittest/person/\w+/preallocation/whatever_preallocation_\w+_v001.jpg$', path)) else: self.assertEquals( True, None != re.search( 'unittest/person/\w+/preallocation/whatever_preallocation_v001.jpg$', path)) # create a file directly in the path location and register in # transaction f = open(path, 'wb') f.write("wowow") f.close() # add this file to the snapshot and force the name snapshot_code = snapshot.get_code() checkin = FileAppendCheckin(snapshot_code, [path], [file_type], keep_file_name=True, mode='preallocate') checkin.execute() # check that it worked snapshot = checkin.get_snapshot() lib_dir = snapshot.get_lib_dir() file_name = snapshot.get_file_name_by_type(file_type) self.assertEquals(True, os.path.exists("%s/%s" % (lib_dir, file_name))) # test preallocation on a sequence file_name = "images_%0.4d.png" file_type = 'sequence' file_range = FileRange(1, 5) # should specify strict checkin_type for Append checkin after path = snapshot.get_preallocated_path(file_type=file_type, file_name=file_name, checkin_type='strict') # imitate a render by building files directly to the path for i in range(1, 6): cur_path = path % i f = open(cur_path, 'wb') f.write("wowow") f.close() # register these files snapshot_code = snapshot.get_code() # should specify strict checkin_type for Append checkin checkin = FileGroupAppendCheckin(snapshot_code, [path], [file_type], file_range, \ keep_file_name=True, mode='preallocate', checkin_type='strict') checkin.execute() snapshot = checkin.get_snapshot() # get the file paths file_names = snapshot.get_expanded_file_names(file_type) lib_dir = snapshot.get_lib_dir() for file_name in file_names: path = "%s/%s" % (lib_dir, file_name) self.assertEquals(True, os.path.exists(path))