def test_references_application_test(self): """testing an example of ReferenceMixin usage """ class GreatEntity(SimpleEntity, ReferenceMixin): __tablename__ = 'GreatEntities' __mapper_args__ = {"polymorphic_identity": "GreatEntity"} ge_id = Column('id', Integer, ForeignKey('SimpleEntities.id'), primary_key=True) my_ge = GreatEntity(name="Test") # we should have a references attribute right now var = my_ge.references image_link_type = Type(name='Image', code='image', target_entity_type="Link") new_link = Link(name="NewTestLink", full_path="nopath", filename="nofilename", type=image_link_type) test_value = [new_link] my_ge.references = test_value assert my_ge.references == test_value
def setUp(self): """setup the test """ # link type self.test_link_type = Type( name='Test Link Type', code='testlink', target_entity_type=Link, ) # create a couple of Link objects self.test_link1 = Link( name="Test Link 1", type=self.test_link_type, full_path="test_path", filename="test_filename", ) self.test_link2 = Link( name="Test Link 2", type=self.test_link_type, full_path="test_path", filename="test_filename", ) self.test_link3 = Link( name="Test Link 3", type=self.test_link_type, full_path="test_path", filename="test_filename", ) self.test_link4 = Link( name="Test Link 4", type=self.test_link_type, full_path="test_path", filename="test_filename", ) self.test_entity1 = Entity( name="Test Entity 1", ) self.test_entity2 = Entity( name="Test Entity 2", ) self.test_links = [ self.test_link1, self.test_link2, self.test_link3, self.test_link4, ] self.test_foo_obj = RefMixFooClass(name="Ref Mixin Test")
def setUp(self): """setup the test """ super(ProjectMixinTester, self).setUp() self.test_stat1 = Status(name="On Hold", code="OH") self.test_stat2 = Status(name="Work In Progress", code="WIP") self.test_stat3 = Status(name="Approved", code="APP") self.test_status_list_1 = StatusList( name="A Statuses", statuses=[self.test_stat1, self.test_stat3], target_entity_type=DeclProjMixA, ) self.test_status_list_2 = StatusList( name="B Statuses", statuses=[self.test_stat2, self.test_stat3], target_entity_type=DeclProjMixB ) self.test_project_statuses = StatusList( name="Project Statuses", statuses=[self.test_stat2, self.test_stat3], target_entity_type='Project' ) self.test_project_type = Type( name='Test Project Type', code='testproj', target_entity_type='Project', ) self.test_repository = Repository( name="Test Repo", code='TR', ) self.test_project = Project( name="Test Project", code='tp', type=self.test_project_type, status_list=self.test_project_statuses, repository=self.test_repository, ) self.kwargs = { "name": "ozgur", "status_list": self.test_status_list_1, "project": self.test_project, } self.test_a_obj = DeclProjMixA(**self.kwargs)
from stalker.models.type import Type DBSession.remove() DBSession.configure(extension=None) db.setup() status1 = Status(name="Complete", code="CMPLT") status2 = Status(name="Pending Review", code="PRev") repo1 = Repository(name="TestRepo") project_status_list = StatusList(name="test", target_entity_type=Project, statuses=[status1]) project_type = Type(name="test", code='test', target_entity_type=Project) new_project1 = Project(name="test project1", code='tp1', type=project_type, status_list=project_status_list, repository=repo1) new_project2 = Project(name="test project2", code='tp2', type=project_type, status_list=project_status_list, repository=repo1) character_asset_type = Type(name="Character", code='char',