def create(self, project, entity_type, task_type, task_name): """Creates data according to the template :param stalker.models.project.Project project: A stalker Project instance :param str entity_type: Type type of entity to be created. Should be one of 'Asset', 'Shot' or 'Sequence' :param str task_type: The type of task. :param str task_name: The name of the task. :return: """ data = self.task_data[entity_type][task_type] child_task_data = None if project.type and project.type.name in data: child_task_data = data[project.type.name] else: child_Task_data = data['Default'] from stalker import Task, Asset, Shot, Sequence entity_class = Task if entity_type == 'Asset': entity = Asset( name=task_name, code=task_name, parent= )
def test_ReferenceMixin_initialization(self): """testing if the ReferenceMixin part is initialized correctly """ from stalker import Link, Type link_type_1 = Type( name="Image", code='image', target_entity_type="Link" ) link1 = Link( name="Artwork 1", full_path="/mnt/M/JOBs/TEST_PROJECT", filename="a.jpg", type=link_type_1 ) link2 = Link( name="Artwork 2", full_path="/mnt/M/JOBs/TEST_PROJECT", filename="b.jbg", type=link_type_1 ) references = [link1, link2] self.kwargs["code"] = "SH12314" self.kwargs["references"] = references from stalker import Asset new_asset = Asset(**self.kwargs) assert new_asset.references == references
def test_inequality(self): """testing inequality of two Asset objects """ new_asset1 = Asset(**self.kwargs) new_asset2 = Asset(**self.kwargs) new_entity1 = Entity(**self.kwargs) self.kwargs["type"] = self.asset_type2 new_asset3 = Asset(**self.kwargs) self.kwargs["name"] = "another name" new_asset4 = Asset(**self.kwargs) self.assertFalse(new_asset1 != new_asset2) self.assertTrue(new_asset1 != new_asset3) self.assertTrue(new_asset1 != new_asset4) self.assertTrue(new_asset3 != new_asset4) self.assertTrue(new_asset1 != new_entity1)
def test_inequality(self): """testing inequality of two Asset objects """ from stalker import Asset, Entity new_asset1 = Asset(**self.kwargs) new_asset2 = Asset(**self.kwargs) new_entity1 = Entity(**self.kwargs) self.kwargs["type"] = self.asset_type2 new_asset3 = Asset(**self.kwargs) self.kwargs["name"] = "another name" new_asset4 = Asset(**self.kwargs) assert not new_asset1 != new_asset2 assert new_asset1 != new_asset3 assert new_asset1 != new_asset4 assert new_asset3 != new_asset4 assert new_asset1 != new_entity1
def test_equality(self): """testing equality of two Asset objects """ from stalker import Asset, Entity new_asset1 = Asset(**self.kwargs) new_asset2 = Asset(**self.kwargs) new_entity1 = Entity(**self.kwargs) self.kwargs["type"] = self.asset_type2 new_asset3 = Asset(**self.kwargs) self.kwargs["name"] = "another name" new_asset4 = Asset(**self.kwargs) self.assertTrue(new_asset1 == new_asset2) self.assertFalse(new_asset1 == new_asset3) self.assertFalse(new_asset1 == new_asset4) self.assertFalse(new_asset3 == new_asset4) self.assertFalse(new_asset1 == new_entity1)
def test_StatusMixin_initialization(self): """testing if the StatusMixin part is initialized correctly """ status_list = \ StatusList.query.filter_by(target_entity_type='Asset').first() self.kwargs["code"] = "SH12314" self.kwargs["status"] = 0 self.kwargs["status_list"] = status_list new_asset = Asset(**self.kwargs) self.assertEqual(new_asset.status_list, status_list)
def test_TaskMixin_initialization(self): """testing if the TaskMixin part is initialized correctly """ commercial_project_type = Type( name="Commercial", code='comm', target_entity_type=Project, ) new_project = Project( name="Commercial", code='COM', type=commercial_project_type, status_list=self.project_status_list, repository=self.repository, ) character_asset_type = Type( name="Character", code='char', target_entity_type=Asset ) new_asset = Asset( name="test asset", type=character_asset_type, code="tstasset", project=new_project, responsible=[self.test_user1] ) task1 = Task( name="Modeling", parent=new_asset ) task2 = Task( name="Lighting", parent=new_asset ) tasks = [task1, task2] self.assertEqual( sorted(new_asset.tasks, key=lambda x: x.name), sorted(tasks, key=lambda x: x.name) )
def test_TaskMixin_initialization(self): """testing if the TaskMixin part is initialized correctly """ from stalker import Type, Project, Asset, Task commercial_project_type = Type( name="Commercial", code='comm', target_entity_type='Project', ) new_project = Project( name="Commercial", code='COM', type=commercial_project_type, repository=self.repository, ) character_asset_type = Type( name="Character", code='char', target_entity_type='Asset' ) new_asset = Asset( name="test asset", type=character_asset_type, code="tstasset", project=new_project, responsible=[self.test_user1] ) task1 = Task( name="Modeling", parent=new_asset ) task2 = Task( name="Lighting", parent=new_asset ) tasks = [task1, task2] assert \ sorted(new_asset.tasks, key=lambda x: x.name) == \ sorted(tasks, key=lambda x: x.name)
def setUp(self): """set the test up """ super(EntityGroupTestCase, self).setUp() from stalker import (db, defaults, Status, User, StatusList, Repository, Project, Type, Asset, Task) # create a couple of task self.status_new = Status.query.filter(Status.code == 'NEW').first() self.status_wip = Status.query.filter(Status.code == 'WIP').first() self.status_cmpl = \ Status.query.filter(Status.code == 'CMPL').first() self.test_user1 = User( name="User1", login="******", email="*****@*****.**", password="******", ) self.test_user2 = User( name="User2", login="******", email="*****@*****.**", password="******", ) self.test_user3 = User( name="User3", login="******", email="*****@*****.**", password="******", ) self.project_status_list = StatusList( name="Project Status List", statuses=[self.status_new, self.status_wip, self.status_cmpl], target_entity_type="Project", ) self.repo = Repository( name='Test Repo', linux_path='/mnt/M/JOBs', windows_path='M:/JOBs', osx_path='/Users/Shared/Servers/M', ) self.project1 = Project( name='Tests Project', code='tp', status_list=self.project_status_list, repository=self.repo, ) self.char_asset_type = Type( name='Character Asset', code='char', target_entity_type="Asset" ) self.asset1 = Asset( name='Char1', code='char1', type=self.char_asset_type, project=self.project1, responsible=[self.test_user1] ) self.task1 = Task( name="Test Task", watchers=[self.test_user3], parent=self.asset1, schedule_timing=5, schedule_unit='h', bid_timing=52, bid_unit='h' ) self.child_task1 = Task( name='Child Task 1', resources=[self.test_user1, self.test_user2], parent=self.task1, ) self.child_task2 = Task( name='Child Task 2', resources=[self.test_user1, self.test_user2], parent=self.task1, ) self.task2 = Task( name='Another Task', project=self.project1, resources=[self.test_user1], responsible=[self.test_user2] ) self.entity_group1 = EntityGroup( name='My Tasks', entities=[ self.task1, self.child_task2, self.task2 ] ) db.DBSession.add_all([ self.task1, self.child_task1, self.child_task2, self.task2, self.test_user1, self.test_user2, self.project1, self.status_cmpl, self.status_new, self.status_wip, self.asset1 ]) db.DBSession.commit()
def test_readme_tutorial_code(setup_sqlite3): """Tests the tutorial code in README.rst """ from stalker import db db.setup() db.init() from stalker.db.session import DBSession assert str(DBSession.connection().engine.url) == 'sqlite://' from stalker import User me = User(name='Erkan Ozgur Yilmaz', login='******', email='*****@*****.**', password='******') # Save the user to database DBSession.save(me) from stalker import Repository repo = Repository(name='Commercial Projects Repository', code='CPR', windows_path='Z:/Projects', linux_path='/mnt/Z/Projects', osx_path='/Volumes/Z/Projects') from stalker import FilenameTemplate task_template = FilenameTemplate( name='Standard Task Filename Template', target_entity_type='Task', # This is for files saved for Tasks path='{{project.repository.path}}/{{project.code}}/' '{%- for parent_task in parent_tasks -%}' '{{parent_task.nice_name}}/' '{%- endfor -%}', # This is Jinja2 template code filename= '{{version.nice_name}}_v{{"%03d"|format(version.version_number)}}') from stalker import Structure standard_folder_structure = Structure( name='Standard Project Folder Structure', templates=[task_template], custom_template= '{{project.code}}/References' # If you need extra folders ) from stalker import Project new_project = Project( name='Test Project', code='TP', structure=standard_folder_structure, repositories=[repo ], # if you have more than one repository you can do it ) from stalker import ImageFormat hd1080 = ImageFormat(name='1080p', width=1920, height=1080) new_project.image_format = hd1080 # Save the project and all the other data it is connected to it DBSession.save(new_project) from stalker import Task, Asset, Shot, Type # define Character asset type char_type = Type(name='Character', code='CHAR', target_entity_type='Asset') character1 = Asset(name='Character 1', code='CHAR1', type=char_type, project=new_project) # Save the Asset DBSession.save(character1) model = Task(name='Model', parent=character1) rigging = Task( name='Rig', parent=character1, depends=[model ], # For project management, define that Rig can not start # before Model ends. ) # Save the new tasks DBSession.save([model, rigging]) # A shot and some tasks for it shot = Shot(name='SH001', code='SH001', project=new_project) # Save the Shot DBSession.save(shot) animation = Task( name='Animation', parent=shot, ) lighting = Task( name='Lighting', parent=shot, depends=[animation], # Lighting can not start before Animation ends, schedule_timing=1, schedule_unit='d', # The task expected to take 1 day to complete resources=[me]) DBSession.save([animation, lighting]) from stalker import Version new_version = Version(task=animation) new_version.update_paths() # to render the naming convention template new_version.extension = '.ma' # let's say that we have created under Maya assert new_version.absolute_full_path == \ "%sTP/SH001/Animation/SH001_Animation_Main_v001.ma" % \ repo.path assert new_version.version_number == 1 new_version2 = Version(task=animation) new_version2.update_paths() # to render the naming convention template new_version2.extension = '.ma' # let's say that we have created under Maya assert new_version2.version_number == 2
def setUp(self): """setup the tests """ # ----------------------------------------------------------------- # start of the setUp # create the environment variable and point it to a temp directory db.setup() db.init() self.temp_repo_path = tempfile.mkdtemp() self.user1 = User(name='User 1', login='******', email='*****@*****.**', password='******') db.DBSession.add(self.user1) db.DBSession.commit() # login as self.user1 from stalker import LocalSession local_session = LocalSession() local_session.store_user(self.user1) local_session.save() self.repo1 = Repository(name='Test Project Repository', linux_path=self.temp_repo_path, windows_path=self.temp_repo_path, osx_path=self.temp_repo_path) self.status_new = Status.query.filter_by(code='NEW').first() self.status_wip = Status.query.filter_by(code='WIP').first() self.status_comp = Status.query.filter_by(code='CMPL').first() self.task_template = FilenameTemplate( name='Task Template', target_entity_type='Task', path='{{project.code}}/' '{%- for parent_task in parent_tasks -%}' '{{parent_task.nice_name}}/' '{%- endfor -%}', filename='{{version.nice_name}}' '_v{{"%03d"|format(version.version_number)}}', ) self.asset_template = FilenameTemplate( name='Asset Template', target_entity_type='Asset', path='{{project.code}}/' '{%- for parent_task in parent_tasks -%}' '{{parent_task.nice_name}}/' '{%- endfor -%}', filename='{{version.nice_name}}' '_v{{"%03d"|format(version.version_number)}}', ) self.shot_template = FilenameTemplate( name='Shot Template', target_entity_type='Shot', path='{{project.code}}/' '{%- for parent_task in parent_tasks -%}' '{{parent_task.nice_name}}/' '{%- endfor -%}', filename='{{version.nice_name}}' '_v{{"%03d"|format(version.version_number)}}', ) self.sequence_template = FilenameTemplate( name='Sequence Template', target_entity_type='Sequence', path='{{project.code}}/' '{%- for parent_task in parent_tasks -%}' '{{parent_task.nice_name}}/' '{%- endfor -%}', filename='{{version.nice_name}}' '_v{{"%03d"|format(version.version_number)}}', ) self.structure = Structure(name='Project Struture', templates=[ self.task_template, self.asset_template, self.shot_template, self.sequence_template ]) self.project_status_list = StatusList( name='Project Statuses', target_entity_type='Project', statuses=[self.status_new, self.status_wip, self.status_comp]) self.image_format = ImageFormat(name='HD 1080', width=1920, height=1080, pixel_aspect=1.0) # create a test project self.project = Project(name='Test Project', code='TP', repository=self.repo1, status_list=self.project_status_list, structure=self.structure, image_format=self.image_format) self.task_status_list =\ StatusList.query.filter_by(target_entity_type='Task').first() self.asset_status_list =\ StatusList.query.filter_by(target_entity_type='Asset').first() self.shot_status_list =\ StatusList.query.filter_by(target_entity_type='Shot').first() self.sequence_status_list =\ StatusList.query.filter_by(target_entity_type='Sequence').first() self.character_type = Type(name='Character', code='CHAR', target_entity_type='Asset') # create a test series of root task self.task1 = Task(name='Test Task 1', project=self.project) self.task2 = Task(name='Test Task 2', project=self.project) self.task3 = Task(name='Test Task 3', project=self.project) # then a couple of child tasks self.task4 = Task(name='Test Task 4', parent=self.task1) self.task5 = Task(name='Test Task 5', parent=self.task1) self.task6 = Task(name='Test Task 6', parent=self.task1) # create a root asset self.asset1 = Asset(name='Asset 1', code='asset1', type=self.character_type, project=self.project) # create a child asset self.asset2 = Asset(name='Asset 2', code='asset2', type=self.character_type, parent=self.task4) # create a root Sequence self.sequence1 = Sequence(name='Sequence1', code='SEQ1', project=self.project) # create a child Sequence self.sequence2 = Sequence(name='Sequence2', code='SEQ2', parent=self.task2) # create a root Shot self.shot1 = Shot(name='SH001', code='SH001', project=self.project) # create a child Shot (child of a Sequence) self.shot2 = Shot(name='SH002', code='SH002', parent=self.sequence1) # create a child Shot (child of a child Sequence) self.shot3 = Shot(name='SH003', code='SH003', parent=self.sequence2) # commit everything db.DBSession.add_all([ self.repo1, self.status_new, self.status_wip, self.status_comp, self.project_status_list, self.project, self.task_status_list, self.asset_status_list, self.shot_status_list, self.sequence_status_list, self.task1, self.task2, self.task3, self.task4, self.task5, self.task6, self.asset1, self.asset2, self.shot1, self.shot2, self.shot3, self.sequence1, self.sequence2, self.task_template, self.asset_template, self.shot_template, self.sequence_template ]) db.DBSession.commit() # now create versions def create_version(task, take_name): """Creates a new version :param task: the task :param take_name: the take_name name :return: the version """ # just renew the scene #pymel.core.newFile(force=True) v = Version(task=task, take_name=take_name) v.update_paths() db.DBSession.add(v) db.DBSession.commit() #self.maya_env.save_as(v) return v # asset2 self.version1 = create_version(self.asset2, 'Main') self.version2 = create_version(self.asset2, 'Main') self.version3 = create_version(self.asset2, 'Main') self.version3.description = 'Test Description' self.version4 = create_version(self.asset2, 'Take1') self.version5 = create_version(self.asset2, 'Take1') self.version6 = create_version(self.asset2, 'Take1') # task5 self.version7 = create_version(self.task5, 'Main') self.version8 = create_version(self.task5, 'Main') self.version9 = create_version(self.task5, 'Main') self.version10 = create_version(self.task5, 'Take1') self.version11 = create_version(self.task5, 'Take1') self.version12 = create_version(self.task5, 'Take1') # task6 self.version13 = create_version(self.task6, 'Main') self.version14 = create_version(self.task6, 'Main') self.version15 = create_version(self.task6, 'Main') self.version16 = create_version(self.task6, 'Take1') self.version17 = create_version(self.task6, 'Take1') self.version18 = create_version(self.task6, 'Take1') # shot3 self.version19 = create_version(self.shot3, 'Main') self.version20 = create_version(self.shot3, 'Main') self.version21 = create_version(self.shot3, 'Main') self.version22 = create_version(self.shot3, 'Take1') self.version23 = create_version(self.shot3, 'Take1') self.version24 = create_version(self.shot3, 'Take1') # task3 self.version25 = create_version(self.task3, 'Main') self.version26 = create_version(self.task3, 'Main') self.version27 = create_version(self.task3, 'Main') self.version28 = create_version(self.task3, 'Take1') self.version29 = create_version(self.task3, 'Take1') self.version30 = create_version(self.task3, 'Take1') # asset1 self.version31 = create_version(self.asset1, 'Main') self.version32 = create_version(self.asset1, 'Main') self.version33 = create_version(self.asset1, 'Main') self.version34 = create_version(self.asset1, 'Take1') self.version35 = create_version(self.asset1, 'Take1') self.version36 = create_version(self.asset1, 'Take1') # shot2 self.version37 = create_version(self.shot2, 'Main') self.version38 = create_version(self.shot2, 'Main') self.version39 = create_version(self.shot2, 'Main') self.version40 = create_version(self.shot2, 'Take1') self.version41 = create_version(self.shot2, 'Take1') self.version42 = create_version(self.shot2, 'Take1') # shot1 self.version43 = create_version(self.shot1, 'Main') self.version44 = create_version(self.shot1, 'Main') self.version45 = create_version(self.shot1, 'Main') self.version46 = create_version(self.shot1, 'Take1') self.version47 = create_version(self.shot1, 'Take1') self.version48 = create_version(self.shot1, 'Take1') # +- task1 # | | # | +- task4 # | | | # | | +- asset2 # | | +- Main # | | | +- version1 # | | | +- version2 (P) # | | | +- version3 (P) # | | +- Take1 # | | +- version4 (P) # | | +- version5 # | | +- version6 (P) # | | # | +- task5 # | | +- Main # | | | +- version7 # | | | +- version8 # | | | +- version9 # | | +- Take1 # | | +- version10 # | | +- version11 # | | +- version12 (P) # | | # | +- task6 # | +- Main # | | +- version13 # | | +- version14 # | | +- version15 # | +- Take1 # | +- version16 (P) # | +- version17 # | +- version18 (P) # | # +- task2 # | | # | +- sequence2 # | | # | +- shot3 # | +- Main # | | +- version19 # | | +- version20 # | | +- version21 # | +- Take1 # | +- version22 # | +- version23 # | +- version24 # | # +- task3 # | +- Main # | | +- version25 # | | +- version26 # | | +- version27 # | +- Take1 # | +- version28 # | +- version29 # | +- version30 # | # +- asset1 # | +- Main # | | +- version31 # | | +- version32 # | | +- version33 # | +- Take1 # | +- version34 # | +- version35 # | +- version36 # | # +- sequence1 # | | # | +- shot2 # | +- Main # | | +- version37 # | | +- version38 # | | +- version39 # | +- Take1 # | +- version40 # | +- version41 # | +- version42 # | # +- shot1 # +- Main # | +- version43 # | +- version44 # | +- version45 # +- Take1 # +- version46 # +- version47 # +- version48 # Start Condition: # # version15 # version12 # version5 # version2 -> has new published version (version3) # version5 -> Referenced a second time # version2 -> has new published version (version3) # version12 -> Referenced a second time # version5 # version2 -> has new published version (version3) # version5 # version2 -> has new published version (version3) # version45 -> no change # version48 -> no change # # Expected Final Result # version15A -> Derived from version15 # version12A -> Derived from version12 # version5A -> Derived from version5 # version3 -> has new published version (version3) # version5A -> Derived from version5 # version3 -> has new published version (version3) # version12A -> Derived from version12 - The second reference # version5A -> Derived from version5 # version3 -> has new published version (version3) # version5A -> Derived from version5 # version3 -> has new published version (version3) # version45 -> no change # version48 -> no change # create a deep relation self.version2.is_published = True self.version3.is_published = True # new scene # version5 references version2 self.version5.inputs.append(self.version2) self.version5.is_published = True # version12 references version5 self.version12.inputs.append(self.version5) self.version12.is_published = True # version45 references version48 self.version45.is_published = True self.version48.is_published = True self.version45.inputs.append(self.version48) # version15 references version12 and version48 self.version15.inputs.append(self.version12) self.version15.inputs.append(self.version45) # reference_resolution self.reference_resolution = { 'root': [self.version12, self.version45], 'leave': [self.version48, self.version45], 'update': [self.version2], 'create': [self.version5, self.version12] } # create a buffer for extra created files, which are to be removed self.remove_these_files_buffer = [] self.test_environment = TestEnvironment(name='Test Environment') self.test_environment._version = self.version15 if not QtGui.QApplication.instance(): logger.debug('creating a new QApplication') self.app = QtGui.QApplication(sys.argv) else: logger.debug('using the present QApplication: %s' % QtGui.qApp) # self.app = QtGui.qApp self.app = QtGui.QApplication.instance() self.dialog = version_updater.MainDialog( environment=self.test_environment, reference_resolution=self.reference_resolution)
def setUp(self): """setup the test """ super(ShotTester, self).setUp() # statuses # types from stalker.db.session import DBSession from stalker import Type self.test_commercial_project_type = Type( name="Commercial Project", code='comm', target_entity_type='Project', ) DBSession.add(self.test_commercial_project_type) self.test_character_asset_type = Type( name="Character", code='char', target_entity_type='Asset', ) DBSession.add(self.test_character_asset_type) self.test_repository_type = Type( name="Test Repository Type", code='test', target_entity_type='Repository' ) DBSession.add(self.test_repository_type) # repository from stalker import Repository self.test_repository = Repository( name="Test Repository", type=self.test_repository_type, ) DBSession.add(self.test_repository) # image format from stalker import ImageFormat self.test_image_format1 = ImageFormat( name='Test Image Format 1', width=1920, height=1080, pixel_aspect=1.0 ) DBSession.add(self.test_image_format1) self.test_image_format2 = ImageFormat( name='Test Image Format 2', width=1280, height=720, pixel_aspect=1.0 ) DBSession.add(self.test_image_format2) # project and sequences from stalker import Project self.test_project1 = Project( name='Test Project1', code='tp1', type=self.test_commercial_project_type, repository=self.test_repository, image_format=self.test_image_format1 ) DBSession.add(self.test_project1) DBSession.commit() self.test_project2 = Project( name='Test Project2', code='tp2', type=self.test_commercial_project_type, repository=self.test_repository, image_format=self.test_image_format1 ) DBSession.add(self.test_project2) DBSession.commit() from stalker import Sequence self.test_sequence1 = Sequence( name="Test Seq1", code='ts1', project=self.test_project1, ) DBSession.add(self.test_sequence1) DBSession.commit() self.test_sequence2 = Sequence( name="Test Seq2", code='ts2', project=self.test_project1, ) DBSession.add(self.test_sequence2) DBSession.commit() self.test_sequence3 = Sequence( name="Test Seq3", code='ts3', project=self.test_project1, ) DBSession.add(self.test_sequence3) DBSession.commit() from stalker import Scene self.test_scene1 = Scene( name='Test Sce1', code='tsc1', project=self.test_project1, ) DBSession.add(self.test_scene1) DBSession.commit() self.test_scene2 = Scene( name='Test Sce2', code='tsc2', project=self.test_project1, ) DBSession.add(self.test_scene2) DBSession.commit() self.test_scene3 = Scene( name='Test Sce3', code='tsc3', project=self.test_project1 ) DBSession.add(self.test_scene3) DBSession.commit() from stalker import Asset self.test_asset1 = Asset( name="Test Asset1", code='ta1', project=self.test_project1, type=self.test_character_asset_type, ) DBSession.add(self.test_asset1) DBSession.commit() self.test_asset2 = Asset( name="Test Asset2", code='ta2', project=self.test_project1, type=self.test_character_asset_type, ) DBSession.add(self.test_asset2) DBSession.commit() self.test_asset3 = Asset( name="Test Asset3", code='ta3', project=self.test_project1, type=self.test_character_asset_type, ) DBSession.add(self.test_asset3) DBSession.commit() self.kwargs = dict( name='SH123', code='SH123', description='This is a test Shot', project=self.test_project1, sequences=[self.test_sequence1, self.test_sequence2], scenes=[self.test_scene1, self.test_scene2], cut_in=112, cut_out=149, source_in=120, source_out=140, record_in=85485, status=0, image_format=self.test_image_format2 ) # create a mock shot object self.test_shot = Shot(**self.kwargs) DBSession.add(self.test_project1) DBSession.commit()
def setUp(self): """set the test up """ super(EntityGroupTestCase, self).setUp() from stalker import (Status, User, StatusList, Repository, Project, Type, Asset, Task) # create a couple of task self.status_new = Status(name='Mew', code='NEW') self.status_wfd = Status(name='Waiting For Dependency', code='WFD') self.status_rts = Status(name='Ready To Start', code='RTS') self.status_wip = Status(name='Work In Progress', code='WIP') self.status_prev = Status(name='Pending Review', code='PREV') self.status_hrev = Status(name='Has Revision', code='HREV') self.status_drev = Status(name='Dependency Has Revision', code='DREV') self.status_cmpl = Status(name='Completed', code='CMPL') self.test_user1 = User( name="User1", login="******", email="*****@*****.**", password="******", ) self.test_user2 = User( name="User2", login="******", email="*****@*****.**", password="******", ) self.test_user3 = User( name="User3", login="******", email="*****@*****.**", password="******", ) self.project_status_list = StatusList( name="Project Status List", statuses=[self.status_new, self.status_wip, self.status_cmpl], target_entity_type="Project", ) self.repo = Repository( name='Test Repo', linux_path='/mnt/M/JOBs', windows_path='M:/JOBs', osx_path='/Users/Shared/Servers/M', ) self.project1 = Project( name='Tests Project', code='tp', status_list=self.project_status_list, repository=self.repo, ) self.char_asset_type = Type(name='Character Asset', code='char', target_entity_type="Asset") self.task_status_list = StatusList( name='Task Statuses', statuses=[ self.status_wfd, self.status_rts, self.status_wip, self.status_prev, self.status_hrev, self.status_drev, self.status_cmpl ], target_entity_type='Task') self.asset_status_list = StatusList( name='Asset Statuses', statuses=[ self.status_wfd, self.status_rts, self.status_wip, self.status_prev, self.status_hrev, self.status_drev, self.status_cmpl ], target_entity_type='Asset') self.asset1 = Asset(name='Char1', code='char1', type=self.char_asset_type, project=self.project1, responsible=[self.test_user1], status_list=self.asset_status_list) self.task1 = Task(name="Test Task", watchers=[self.test_user3], parent=self.asset1, schedule_timing=5, schedule_unit='h', bid_timing=52, bid_unit='h', status_list=self.task_status_list) self.child_task1 = Task(name='Child Task 1', resources=[self.test_user1, self.test_user2], parent=self.task1, status_list=self.task_status_list) self.child_task2 = Task(name='Child Task 2', resources=[self.test_user1, self.test_user2], parent=self.task1, status_list=self.task_status_list) self.task2 = Task(name='Another Task', project=self.project1, resources=[self.test_user1], responsible=[self.test_user2], status_list=self.task_status_list) self.entity_group1 = EntityGroup( name='My Tasks', entities=[self.task1, self.child_task2, self.task2])
def create_project(): """creates test data """ from stalker.db.session import DBSession from stalker import (Task, Asset, Type, Sequence, Shot, Version, FilenameTemplate, Repository, Project, Structure) repo = Repository(name='Test Repository', code='TR', windows_path='T:/', linux_path='/mnt/T/', osx_path='/Volumes/T/') task_filename_template = FilenameTemplate( name='Task Filename Template', path='$REPO{{project.repository.code}}/{{project.code}}/' '{%- for parent_task in parent_tasks -%}{{parent_task.nice_name}}' '/{%- endfor -%}', filename= '{{version.nice_name}}_v{{"%03d"|format(version.version_number)}}', target_entity_type='Task') asset_filename_template = FilenameTemplate( name='Asset Filename Template', path='$REPO{{project.repository.code}}/{{project.code}}/' '{%- for parent_task in parent_tasks -%}{{parent_task.nice_name}}' '/{%- endfor -%}', filename= '{{version.nice_name}}_v{{"%03d"|format(version.version_number)}}', target_entity_type='Asset') shot_filename_template = FilenameTemplate( name='Shot Filename Template', path='$REPO{{project.repository.code}}/{{project.code}}/' '{%- for parent_task in parent_tasks -%}{{parent_task.nice_name}}' '/{%- endfor -%}', filename= '{{version.nice_name}}_v{{"%03d"|format(version.version_number)}}', target_entity_type='Shot') sequence_filename_template = FilenameTemplate( name='Sequence Filename Template', path='$REPO{{project.repository.code}}/{{project.code}}/' '{%- for parent_task in parent_tasks -%}{{parent_task.nice_name}}' '/{%- endfor -%}', filename= '{{version.nice_name}}_v{{"%03d"|format(version.version_number)}}', target_entity_type='Sequence') structure = Structure(name='Default Project Structure', templates=[ task_filename_template, asset_filename_template, shot_filename_template, sequence_filename_template ]) DBSession.add_all([ structure, task_filename_template, asset_filename_template, shot_filename_template, sequence_filename_template ]) DBSession.commit() project = Project( name='Test Project', code='TP', repository=repo, structure=structure, ) DBSession.add(project) DBSession.commit() assets_task = Task(name='Assets', project=project) characters_task = Task(name='Characters', parent=assets_task) props_task = Task(name='Props', parent=assets_task) environments_task = Task(name='Environments', parent=assets_task) sequences_task = Task(name='Sequences', project=project) # Asset Types char_type = Type(name='Character', code='Char', target_entity_type='Asset') prop_type = Type(name='Prop', code='Prop', target_entity_type='Asset') exterior_type = Type(name='Exterior', code='Exterior', target_entity_type='Asset') interior_type = Type(name='Interior', code='Interior', target_entity_type='Asset') # Task Types model_type = Type(name='Model', code='Model', target_entity_type='Task') look_dev_type = Type(name='Look Development', code='LookDev', target_entity_type='Task') rig_type = Type(name='Rig', code='Rig', target_entity_type='Task') layout_type = Type(name='Layout', code='Layout', target_entity_type='Task') anim_type = \ Type(name='Animation', code='Animation', target_entity_type='Task') camera_type = Type(name='Camera', code='Camera', target_entity_type='Task') plate_type = Type(name='Plate', code='Plate', target_entity_type='Task') fx_type = Type(name='FX', code='FX', target_entity_type='Task') lighting_type = \ Type(name='Lighting', code='Lighting', target_entity_type='Task') comp_type = Type(name='Comp', code='Comp', target_entity_type='Task') DBSession.add_all([ char_type, prop_type, exterior_type, interior_type, model_type, look_dev_type, rig_type, layout_type, anim_type, camera_type, plate_type, fx_type, lighting_type, comp_type ]) DBSession.commit() # char1 char1 = Asset(name='Char1', code='Char1', parent=characters_task, type=char_type) model = Task(name='Model', type=model_type, parent=char1) look_dev_task = Task(name='LookDev', type=look_dev_type, parent=char1) rig = Task(name='Rig', type=rig_type, parent=char1) DBSession.add_all([char1, model, look_dev_task, rig]) DBSession.commit() # char2 char2 = Asset(name='Char2', code='Char2', parent=characters_task, type=char_type) model = Task(name='Model', type=model_type, parent=char2) look_dev_task = Task(name='LookDev', type=look_dev_type, parent=char2) rig = Task(name='Rig', type=rig_type, parent=char2) DBSession.add_all([char2, model, look_dev_task, rig]) DBSession.commit() # prop1 prop1 = \ Asset(name='Prop2', code='Prop2', parent=props_task, type=prop_type) model = Task(name='Model', type=model_type, parent=prop1) look_dev_task = Task(name='LookDev', type=look_dev_type, parent=prop1) DBSession.add_all([prop1, model, look_dev_task]) DBSession.commit() # prop2 prop2 = \ Asset(name='Prop2', code='Prop2', parent=props_task, type=prop_type) model = Task(name='Model', type=model_type, parent=prop2) look_dev_task = Task(name='LookDev', type=look_dev_type, parent=prop2) DBSession.add_all([prop2, model, look_dev_task]) DBSession.commit() # environments # env1 env1 = Asset(name='Env1', code='Env1', type=exterior_type, parent=environments_task) layout_task = Task(name='Layout', type=layout_type, parent=env1) props_task = Task(name='Props', parent=env1) yapi1_asset = \ Asset(name='Yapi1', code='Yapi1', type=prop_type, parent=props_task) model_task = Task(name='Model', type=model_type, parent=yapi1_asset) look_dev_task = \ Task(name='LookDev', type=look_dev_type, parent=yapi1_asset) DBSession.add_all([ env1, layout_task, props_task, yapi1_asset, model_task, look_dev_task ]) DBSession.commit() # env2 env2 = Asset(name='Env2', code='Env2', type=exterior_type, parent=environments_task) layout_task = Task(name='Layout', type=layout_type, parent=env2) props_task = Task(name='Props', parent=env2) yapi1_asset = \ Asset(name='Yapi2', code='Yapi2', type=prop_type, parent=props_task) model_task = Task(name='Model', type=model_type, parent=yapi1_asset) look_dev_task = \ Task(name='LookDev', type=look_dev_type, parent=yapi1_asset) DBSession.add_all([ env2, layout_task, props_task, yapi1_asset, model_task, look_dev_task ]) DBSession.commit() # sequences and shots seq1 = Sequence(name='Seq1', code='Seq1', parent=sequences_task) edit_task = Task(name='Edit', parent=seq1) shots_task = Task(name='Shots', parent=seq1) DBSession.add_all([seq1, edit_task, shots_task]) DBSession.commit() # shot1 shot1 = Shot(name='Seq001_001_0010', code='Seq001_001_0010', sequences=[seq1], parent=shots_task) anim_task = Task(name='Anim', type=anim_type, parent=shot1) camera_task = Task(name='Camera', type=camera_type, parent=shot1) plate_task = Task(name='Plate', type=plate_type, parent=shot1) fx_task = Task(name='FX', type=fx_type, parent=shot1) lighting_task = Task(name='Lighting', type=lighting_type, parent=shot1) comp_task = Task(name='Comp', type=comp_type, parent=shot1) DBSession.add_all([ shot1, anim_task, camera_task, plate_task, fx_task, lighting_task, comp_task ]) DBSession.commit() # shot2 shot2 = Shot(name='Seq001_001_0020', code='Seq001_001_0020', sequences=[seq1], parent=shots_task) anim_task = Task(name='Anim', type=anim_type, parent=shot2) camera_task = Task(name='Camera', type=camera_type, parent=shot2) plate_task = Task(name='Plate', type=plate_type, parent=shot2) fx_task = Task(name='FX', type=fx_type, parent=shot2) lighting_task = Task(name='Lighting', type=lighting_type, parent=shot2) comp_task = Task(name='Comp', type=comp_type, parent=shot2) DBSession.add_all([ shot2, anim_task, camera_task, plate_task, fx_task, lighting_task, comp_task ]) DBSession.commit() # shot3 shot3 = Shot(name='Seq001_001_0030', code='Seq001_001_0030', sequences=[seq1], parent=shots_task) anim_task = Task(name='Anim', type=anim_type, parent=shot3) camera_task = Task(name='Camera', type=camera_type, parent=shot3) plate_task = Task(name='Plate', type=plate_type, parent=shot3) fx_task = Task(name='FX', type=fx_type, parent=shot3) lighting_task = Task(name='Lighting', type=lighting_type, parent=shot3) comp_task = Task(name='Comp', type=comp_type, parent=shot3) DBSession.add_all([ shot3, anim_task, camera_task, plate_task, fx_task, lighting_task, comp_task ]) DBSession.commit() # Seq2 # sequences and shots seq2 = Sequence(name='Seq2', code='Seq2', parent=sequences_task) edit_task = Task(name='Edit', parent=seq2) shots_task = Task(name='Shots', parent=seq2) DBSession.add_all([seq2, edit_task, shots_task]) DBSession.commit() # shot1 shot1 = Shot(name='Seq002_001_0010', code='Seq002_001_0010', sequences=[seq2], parent=shots_task) anim_task = Task(name='Anim', type=anim_type, parent=shot1) camera_task = Task(name='Camera', type=camera_type, parent=shot1) plate_task = Task(name='Plate', type=plate_type, parent=shot1) fx_task = Task(name='FX', type=fx_type, parent=shot1) lighting_task = Task(name='Lighting', type=lighting_type, parent=shot1) comp_task = Task(name='Comp', type=comp_type, parent=shot1) DBSession.add_all([ shot1, anim_task, camera_task, plate_task, fx_task, lighting_task, comp_task ]) DBSession.commit() # shot2 shot2 = Shot(name='Seq002_001_0020', code='Seq002_001_0020', sequences=[seq2], parent=shots_task) anim_task = Task(name='Anim', type=anim_type, parent=shot2) camera_task = Task(name='Camera', type=camera_type, parent=shot2) plate_task = Task(name='Plate', type=plate_type, parent=shot2) fx_task = Task(name='FX', type=fx_type, parent=shot2) lighting_task = Task(name='Lighting', type=lighting_type, parent=shot2) comp_task = Task(name='Comp', type=comp_type, parent=shot2) DBSession.add_all([ shot2, anim_task, camera_task, plate_task, fx_task, lighting_task, comp_task ]) DBSession.commit() # shot3 shot3 = Shot(name='Seq002_001_0030', code='Seq002_001_0030', sequences=[seq2], parent=shots_task) anim_task = Task(name='Anim', type=anim_type, parent=shot3) camera_task = Task(name='Camera', type=camera_type, parent=shot3) plate_task = Task(name='Plate', type=plate_type, parent=shot3) fx_task = Task(name='FX', type=fx_type, parent=shot3) lighting_task = Task(name='Lighting', type=lighting_type, parent=shot3) comp_task = Task(name='Comp', type=comp_type, parent=shot3) DBSession.add_all([ shot3, anim_task, camera_task, plate_task, fx_task, lighting_task, comp_task ]) DBSession.commit() yield project
def setUp(self): """set up the test """ super(TicketTester, self).setUp() # create statuses from stalker import Status self.test_status1 = Status(name='N', code='N') self.test_status2 = Status(name='R', code='R') # get the ticket types from stalker import Type ticket_types = Type.query \ .filter(Type.target_entity_type == 'Ticket').all() self.ticket_type_1 = ticket_types[0] self.ticket_type_2 = ticket_types[1] # create a User from stalker import User self.test_user = User(name='Test User', login='******', email='*****@*****.**', password='******') # create a Repository from stalker import Repository self.test_repo = Repository(name="Test Repo") # create a Project Type self.test_project_type = Type( name='Commercial Project', code='comm', target_entity_type='Project', ) # create a Project StatusList self.test_project_status1 = Status(name='PrjStat1', code='PrjStat1') self.test_project_status2 = Status(name='PrjStat2', code='PrjStat2') from stalker import StatusList self.test_project_status_list = StatusList( name="Project Status List", target_entity_type='Project', statuses=[ self.test_project_status1, self.test_project_status2, ]) self.test_task_status_list = \ StatusList.query.filter_by(target_entity_type='Task').first() # create a Project from stalker import Project self.test_project = Project(name="Test Project 1", code="TEST_PROJECT_1", type=self.test_project_type, repository=self.test_repo, status_list=self.test_project_status_list) # create an Asset self.test_asset_status_list = StatusList.query\ .filter_by(target_entity_type='Asset').first() self.test_asset_type = Type(name='Character Asset', code='char', target_entity_type='Asset') from stalker import Asset self.test_asset = Asset(name="Test Asset", code='ta', project=self.test_project, status_list=self.test_asset_status_list, type=self.test_asset_type) # create a Task from stalker import Task self.test_task = Task(name="Modeling of Asset 1", resources=[self.test_user], status_list=self.test_task_status_list, parent=self.test_asset) # create a Version self.test_version_status_list = StatusList( name='Version Statuses', target_entity_type='Version', statuses=[self.test_status1, self.test_status2]) from stalker import Version self.test_version = Version(name='Test Version', task=self.test_task, status_list=self.test_version_status_list, version=1, full_path='some/path') # create the Ticket self.kwargs = { 'project': self.test_project, 'links': [self.test_version], 'summary': 'This is a test ticket', 'description': 'This is the long description', 'priority': 'TRIVIAL', 'reported_by': self.test_user, } from stalker import db, Ticket self.test_ticket = Ticket(**self.kwargs) db.DBSession.add(self.test_ticket) db.DBSession.commit() # get the Ticket Statuses self.status_new = Status.query.filter_by(name='New').first() self.status_accepted = Status.query.filter_by(name='Accepted').first() self.status_assigned = Status.query.filter_by(name='Assigned').first() self.status_reopened = Status.query.filter_by(name='Reopened').first() self.status_closed = Status.query.filter_by(name='Closed').first()
def setUp(self): """setup the test """ db.setup() db.init() # users self.test_user1 = User( name='User1', login='******', password='******', email='*****@*****.**' ) self.test_user2 = User( name='User2', login='******', password='******', email='*****@*****.**' ) # statuses self.status_wip = Status.query.filter_by(code='WIP').first() self.status_cmpl = Status.query.filter_by(code='CMPL').first() # status lists self.project_status_list = StatusList( name="Project Status List", target_entity_type=Project, statuses=[ self.status_cmpl, self.status_wip ] ) self.task_status_list = \ StatusList.query.filter_by(target_entity_type='Task').first() self.asset_status_list = \ StatusList.query.filter_by(target_entity_type='Asset').first() self.shot_status_list = \ StatusList.query.filter_by(target_entity_type='Shot').first() self.sequence_status_list = \ StatusList.query.filter_by(target_entity_type='Sequence').first() # types self.commercial_project_type = Type( name="Commercial Project", code='commproj', target_entity_type=Project, ) self.asset_type1 = Type( name="Character", code='char', target_entity_type=Asset ) self.asset_type2 = Type( name="Environment", code='env', target_entity_type=Asset ) self.repository_type = Type( name="Test Repository Type", code='testrepo', target_entity_type=Repository, ) # repository self.repository = Repository( name="Test Repository", type=self.repository_type, ) # project self.project1 = Project( name="Test Project1", code='tp1', type=self.commercial_project_type, status_list=self.project_status_list, repositories=[self.repository], ) # sequence self.seq1 = Sequence( name="Test Sequence", code='tseq', project=self.project1, status_list=self.sequence_status_list, responsible=[self.test_user1] ) # shots self.shot1 = Shot( code="TestSH001", status_list=self.shot_status_list, project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) self.shot2 = Shot( code="TestSH002", status_list=self.shot_status_list, project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) self.shot3 = Shot( code="TestSH003", status_list=self.shot_status_list, project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) self.shot4 = Shot( code="TestSH004", status_list=self.shot_status_list, project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) self.kwargs = { "name": "Test Asset", 'code': 'ta', "description": "This is a test Asset object", "project": self.project1, "type": self.asset_type1, "status": 0, "status_list": self.asset_status_list, 'responsible': [self.test_user1] } self.asset1 = Asset(**self.kwargs) # tasks self.task1 = Task( name="Task1", parent=self.asset1, status_list=self.task_status_list ) self.task2 = Task( name="Task2", parent=self.asset1, status_list=self.task_status_list ) self.task3 = Task( name="Task3", parent=self.asset1, status_list=self.task_status_list )
def setUp(self): """setup the test """ super(AssetTester, self).setUp() # users from stalker import User from stalker.db.session import DBSession self.test_user1 = User( name='User1', login='******', password='******', email='*****@*****.**' ) DBSession.add(self.test_user1) self.test_user2 = User( name='User2', login='******', password='******', email='*****@*****.**' ) DBSession.add(self.test_user2) DBSession.commit() # statuses from stalker import Status, Project self.status_wip = Status.query.filter_by(code='WIP').first() self.status_cmpl = Status.query.filter_by(code='CMPL').first() # types from stalker import Type self.commercial_project_type = Type( name="Commercial Project", code='commproj', target_entity_type='Project', ) DBSession.add(self.commercial_project_type) self.asset_type1 = Type( name="Character", code='char', target_entity_type='Asset' ) DBSession.add(self.asset_type1) self.asset_type2 = Type( name="Environment", code='env', target_entity_type='Asset' ) DBSession.add(self.asset_type2) self.repository_type = Type( name="Test Repository Type", code='testrepo', target_entity_type='Repository', ) DBSession.add(self.repository_type) # repository from stalker import Repository self.repository = Repository( name="Test Repository", type=self.repository_type, ) DBSession.add(self.repository) # project self.project1 = Project( name="Test Project1", code='tp1', type=self.commercial_project_type, repositories=[self.repository], ) DBSession.add(self.project1) DBSession.commit() # sequence from stalker import Sequence self.seq1 = Sequence( name="Test Sequence", code='tseq', project=self.project1, responsible=[self.test_user1] ) DBSession.add(self.seq1) # shots from stalker import Shot self.shot1 = Shot( code="TestSH001", project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) DBSession.add(self.shot1) self.shot2 = Shot( code="TestSH002", project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) DBSession.add(self.shot2) self.shot3 = Shot( code="TestSH003", project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) DBSession.add(self.shot3) self.shot4 = Shot( code="TestSH004", project=self.project1, sequences=[self.seq1], responsible=[self.test_user1] ) DBSession.add(self.shot4) self.kwargs = { "name": "Test Asset", 'code': 'ta', "description": "This is a test Asset object", "project": self.project1, "type": self.asset_type1, "status": 0, 'responsible': [self.test_user1] } from stalker import Asset, Task self.asset1 = Asset(**self.kwargs) DBSession.add(self.asset1) # tasks self.task1 = Task( name="Task1", parent=self.asset1, ) DBSession.add(self.task1) self.task2 = Task( name="Task2", parent=self.asset1, ) DBSession.add(self.task2) self.task3 = Task( name="Task3", parent=self.asset1, ) DBSession.add(self.task3) DBSession.commit()
def setUp(self): """setup the test """ db.setup() db.init() # statuses self.status_new = Status.query.filter_by(code='NEW').first() self.status_wfd = Status.query.filter_by(code='WFD').first() self.status_rts = Status.query.filter_by(code='RTS').first() self.status_wip = Status.query.filter_by(code='WIP').first() self.status_prev = Status.query.filter_by(code='PREV').first() self.status_hrev = Status.query.filter_by(code='HREV').first() self.status_drev = Status.query.filter_by(code='DREV').first() self.status_oh = Status.query.filter_by(code='OH').first() self.status_stop = Status.query.filter_by(code='STOP').first() self.status_cmpl = Status.query.filter_by(code='CMPL').first() # status lists self.test_project_status_list = StatusList( name="Project Status List", statuses=[self.status_new, self.status_wip, self.status_cmpl], target_entity_type=Project, ) self.test_sequence_status_list = \ StatusList.query.filter_by(target_entity_type='Sequence').first() self.test_shot_status_list = \ StatusList.query.filter_by(target_entity_type='Shot').first() self.test_asset_status_list = \ StatusList.query.filter_by(target_entity_type='Asset').first() # types self.test_commercial_project_type = Type( name="Commercial Project", code='comm', target_entity_type=Project, ) self.test_character_asset_type = Type( name="Character", code='char', target_entity_type=Asset, ) self.test_repository_type = Type(name="Test Repository Type", code='test', target_entity_type=Repository) # repository self.test_repository = Repository( name="Test Repository", type=self.test_repository_type, ) # image format self.test_image_format1 = ImageFormat(name='Test Image Format 1', width=1920, height=1080, pixel_aspect=1.0) self.test_image_format2 = ImageFormat(name='Test Image Format 2', width=1280, height=720, pixel_aspect=1.0) # project and sequences self.test_project1 = Project(name='Test Project1', code='tp1', type=self.test_commercial_project_type, status_list=self.test_project_status_list, repository=self.test_repository, image_format=self.test_image_format1) self.test_project2 = Project(name='Test Project2', code='tp2', type=self.test_commercial_project_type, status_list=self.test_project_status_list, repository=self.test_repository, image_format=self.test_image_format1) self.test_sequence1 = Sequence( name="Test Seq1", code='ts1', project=self.test_project1, status_list=self.test_sequence_status_list, ) self.test_sequence2 = Sequence( name="Test Seq2", code='ts2', project=self.test_project1, status_list=self.test_sequence_status_list, ) self.test_sequence3 = Sequence( name="Test Seq3", code='ts3', project=self.test_project1, status_list=self.test_sequence_status_list, ) self.test_scene1 = Scene( name='Test Sce1', code='tsc1', project=self.test_project1, ) self.test_scene2 = Scene( name='Test Sce2', code='tsc2', project=self.test_project1, ) self.test_scene3 = Scene(name='Test Sce3', code='tsc3', project=self.test_project1) self.test_asset1 = Asset( name="Test Asset1", code='ta1', project=self.test_project1, status_list=self.test_asset_status_list, type=self.test_character_asset_type, ) self.test_asset2 = Asset( name="Test Asset2", code='ta2', project=self.test_project1, status_list=self.test_asset_status_list, type=self.test_character_asset_type, ) self.test_asset3 = Asset( name="Test Asset3", code='ta3', project=self.test_project1, status_list=self.test_asset_status_list, type=self.test_character_asset_type, ) self.kwargs = dict( name='SH123', code='SH123', description='This is a test Shot', project=self.test_project1, sequences=[self.test_sequence1, self.test_sequence2], scenes=[self.test_scene1, self.test_scene2], cut_in=112, cut_out=149, source_in=120, source_out=140, record_in=85485, status=0, status_list=self.test_shot_status_list, image_format=self.test_image_format2) # create a mock shot object self.test_shot = Shot(**self.kwargs) db.DBSession.add(self.test_project1) db.DBSession.commit()
def setUp(self): """set up the test """ # create the db DBSession.remove() db.setup() # db.init() # we just need statuses so create them instead of initializing the db db.create_ticket_statuses() db.create_entity_statuses(entity_type='Task', status_names=defaults.task_status_names, status_codes=defaults.task_status_codes) db.create_entity_statuses(entity_type='Asset', status_names=defaults.task_status_names, status_codes=defaults.task_status_codes) db.create_entity_statuses(entity_type='Shot', status_names=defaults.task_status_names, status_codes=defaults.task_status_codes) db.create_entity_statuses(entity_type='Sequence', status_names=defaults.task_status_names, status_codes=defaults.task_status_codes) db.create_entity_statuses(entity_type='Review', status_names=defaults.review_status_names, status_codes=defaults.review_status_codes) # create statuses self.test_status1 = Status(name='N', code='N') self.test_status2 = Status(name='R', code='R') # get the ticket types ticket_types = Type.query \ .filter(Type.target_entity_type == 'Ticket').all() self.ticket_type_1 = ticket_types[0] self.ticket_type_2 = ticket_types[1] # create a User self.test_user = User(name='Test User', login='******', email='*****@*****.**', password='******') # create a Repository self.test_repo = Repository(name="Test Repo") # create a Project Type self.test_project_type = Type( name='Commercial Project', code='comm', target_entity_type=Project, ) # create a Project StatusList self.test_project_status1 = Status(name='PrjStat1', code='PrjStat1') self.test_project_status2 = Status(name='PrjStat2', code='PrjStat2') self.test_project_status_list = StatusList( name="Project Status List", target_entity_type=Project, statuses=[ self.test_project_status1, self.test_project_status2, ]) self.test_task_status_list = StatusList.query\ .filter_by(target_entity_type='Task').first() # create a Project self.test_project = Project(name="Test Project 1", code="TEST_PROJECT_1", type=self.test_project_type, repository=self.test_repo, status_list=self.test_project_status_list) # create an Asset self.test_asset_status_list = StatusList.query\ .filter_by(target_entity_type='Asset').first() self.test_asset_type = Type(name='Character Asset', code='char', target_entity_type=Asset) self.test_asset = Asset(name="Test Asset", code='ta', project=self.test_project, status_list=self.test_asset_status_list, type=self.test_asset_type) # create a Task self.test_task = Task(name="Modeling of Asset 1", resources=[self.test_user], status_list=self.test_task_status_list, parent=self.test_asset) # create a Version self.test_version_status_list = StatusList( name='Version Statuses', target_entity_type=Version, statuses=[self.test_status1, self.test_status2]) self.test_version = Version(name='Test Version', task=self.test_task, status_list=self.test_version_status_list, version=1, full_path='some/path') # create the Ticket self.kwargs = { 'project': self.test_project, 'links': [self.test_version], 'summary': 'This is a test ticket', 'description': 'This is the long description', 'priority': 'TRIVIAL', 'reported_by': self.test_user, } self.test_ticket = Ticket(**self.kwargs) DBSession.add(self.test_ticket) DBSession.commit() # get the Ticket Statuses self.status_new = Status.query.filter_by(name='New').first() self.status_accepted = Status.query.filter_by(name='Accepted').first() self.status_assigned = Status.query.filter_by(name='Assigned').first() self.status_reopened = Status.query.filter_by(name='Reopened').first() self.status_closed = Status.query.filter_by(name='Closed').first()