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',