def test_get_tag_object_list(stc): proj = CStcSystem.Instance().GetObject("Project") ctor = CScriptableCreator() tags = proj.GetObject('Tags') if tags is None: tags = ctor.Create('Tags', project) assert tags is not None exist_map = {obj.Get('Name'): obj for obj in tags.GetObjects('Tag')} # Create new tag_list = tag_utils.get_tag_object_list(['Tag1']) assert isinstance(tag_list, list) assert 1 == len(tag_list) assert tag_list[0].IsTypeOf('Tag') assert 'Tag1' == tag_list[0].Get('Name') assert 'Tag1' not in exist_map # New map exist_map = {obj.Get('Name'): obj for obj in tags.GetObjects('Tag')} # Retrieve existing, and a new one tag_list = tag_utils.get_tag_object_list(['Tag1', 'Tag2']) assert isinstance(tag_list, list) assert 2 == len(tag_list) assert tag_list[0].IsTypeOf('Tag') assert tag_list[1].IsTypeOf('Tag') assert tag_list[0].Get('Name') in exist_map assert tag_list[1].Get('Name') not in exist_map assert 'Tag2' == tag_list[1].Get('Name')
def test_get_tag_objects_blank(stc): fail_msg = '' try: tag_utils.get_tag_object('') except: exc_info = sys.exc_info() fail_list = traceback.format_exception_only(exc_info[0], exc_info[1]) fail_msg = fail_list[0] if len(fail_list) == 1 else '\n'.join(fail_list) if fail_msg == '': raise AssertionError('function did not fail as expected') if 'must not be blank' not in fail_msg: raise AssertionError('function failed with unexpected exception: "' + fail_msg + '"') try: tag_utils.get_tag_object_list(['non-blank', '']) except: exc_info = sys.exc_info() fail_list = traceback.format_exception_only(exc_info[0], exc_info[1]) fail_msg = fail_list[0] if len(fail_list) == 1 else '\n'.join(fail_list) if fail_msg == '': raise AssertionError('function did not fail as expected') if 'must not be blank' not in fail_msg: raise AssertionError('function failed with unexpected exception: "' + fail_msg + '"')