def test_shot_addition(self, shot_name='ABC123'):
prod_manager = production_manager.ProductionManager('mongodb')
new_project = production.Project(
'test_this_new_project',
'/mnt/VFX/work',
max_users=225
)
new_shot = production.Shot(
new_project.name,
'F11024',
start_frame=1,
end_frame=216,
users=['jacob.kaniuk']
)
new_shot_b = production.Shot(
new_project.name,
'FZ3025',
start_frame=1,
end_frame=92,
users=['jacob.kaniuk']
)
new_shot_c = production.Shot(
new_project.name,
'FZJ025',
start_frame=1,
end_frame=92,
users=['jacob.kaniuk']
)
prod_manager.create_shot(new_project.name, new_shot)
prod_manager.create_shot(new_project.name, new_shot_b)
prod_manager.create_shot(new_project.name, new_shot_c, permission=4)
prod_manager.update_shot(new_project.name, new_shot_c, permission=None)
def test_project_deletion(self):
prod_manager = production_manager.ProductionManager('mongodb')
new_project = production.Project(
'delete_me_test_project',
'/mnt/VFX/delete_me',
max_users=15
)
prod_manager.create_project(new_project)
prod_manager.db_client.drop_database(new_project.name)
def test_project_update(self):
prod_manager = production_manager.ProductionManager('mongodb')
new_project = production.Project(
'test_this_new_project',
'/mnt/VFX/works/doesthiswork',
max_users=150
)
prod_manager.update_project(new_project)
prod_manager.set_project_path(new_project.name, '/mnt/VFX/works/new_project_ath')
def test_project_addition(self):
# test if we can add a project to the databsase
prod_manager = production_manager.ProductionManager('mongodb')
new_project = production.Project('test_this_new_project', '/mnt/VFX/work', max_users=15)
newer_project = production.Project('new_project', '/mnt/s/stuff/', max_users=50)
print prod_manager.create_project(new_project)
print prod_manager.create_project(newer_project)
self.assertRaises(TypeError, prod_manager.create_project, 'testingin')
self.assertRaises(TypeError, prod_manager.create_project, 155)
self.assertRaises(TypeError, prod_manager.create_project, None)
def test_asset_addition(self):
prod_manager = production_manager.ProductionManager('mongodb')
new_project = production.Project(
'test_this_new_project',
'/mnt/VFX/work',
max_users=225
)
prod_manager.create_asset(new_project, 'F11024', 'VHCL_GO_KART')
prod_manager.create_asset(new_project, 'F11024', 'PROP_TELESCOPE')
prod_manager.create_asset(new_project, 'F11024', 'PROP_BICYCLE')
prod_manager.create_asset(new_project, 'F11024', 'PROP_SLED')
prod_manager.create_asset(new_project, 'F11024', 'PROP_SOFA')
prod_manager.create_asset(new_project, 'F11024', 'PROP_TELEVISION')
prod_manager.create_asset(new_project, 'F11024', 'PROP_HAIR_DRYER')
prod_manager.create_asset(new_project, 'F11024', 'PROP_HAMMER', max_weight='2Kg')
prod_manager.update_asset('test_this_new_project', 'F11024', 'PROP_TELESCOPE', max_size=8001)
prod_manager.update_asset('test_this_new_project', 'F11024', 'VHCL_GO_KART', validators=['validate.py', 'validateme.py'])
prod_manager.update_asset(project_name='test_this_new_project', asset_id='5e4b7fbbe82acf15b012bf5d', max_size='22GB')
prod_manager.update_asset(asset_id='5e4b88ecc7a6ce3288e184cc', food='GOOOD')
def test_project_info(self, project_name=None):
prod_manager = production_manager.ProductionManager('mongodb')
info = prod_manager.get_project_info('test_this_new_project')
print "\nPROJECT INFO", info
def test_shot_deletion(self):
prod_manager = production_manager.ProductionManager('mongodb')
prod_manager.remove_shot('test_this_new_project', 'F11024')
prod_manager.remove_shot('test_this_new_project', shot_id='5e4b8869c7a6ce3288e1847b')
prod_manager.remove_shot(shot_id='5e4c92b7c7a6ce3288e18809')
def test_db_client(self):
# test if we can establish a connection with the database
prod_manager = production_manager.ProductionManager('mongodb')
return prod_manager.connect_to_database('localhost')
def test_get_user(self):
from bson import ObjectId
prod_manager = production_manager.ProductionManager('mongodb')
id = ObjectId('5e4e93bbc7a6ce4178e18abb')
result = prod_manager.db_client['test_this_new_project'][ASSET_BIN].find_one({'_id': id})
print result
def test_get_assets_by_general(self):
prod_manager = production_manager.ProductionManager('mongodb')
print prod_manager.get_assets_by(project='test_this_new_project', shot='F11024')
print prod_manager.get_assets_by(project='test_this_new_project', name='PROP_SOFA')
print prod_manager.get_assets_by(project='test_this_new_project', size='40GB')
def test_get_assets_by_status(self):
prod_manager = production_manager.ProductionManager('mongodb')
stuff = prod_manager.get_assets_by_status(1, project_name='test_this_new_project')
for item in stuff:
print item
def test_asset_deletion(self):
prod_manager = production_manager.ProductionManager('mongodb')
prod_manager.remove_asset('test_this_new_project', asset_id='5e4c8905c7a6ce3288e186f1')
# production_manager.remove_asset('test_this_new_project', 'F11024', 'PROP_TELESCOPE')
prod_manager.remove_asset(asset_id='5e4e93bbc7a6ce3288e18abb')