def __init_db__():
"""initializes the just setup database
It adds:
- Users
- VersionTypes
to the database.
"""
logger.debug("db is newly created, initializing the db")
global query
global session
# get the users from the config
from oyProjectManager import conf
# ------------------------------------------------------
# create the users
from oyProjectManager.models.auth import User
# get all users from db
users_from_db = query(User).all()
for user_data in conf.users_data:
name = user_data.get("name")
initials = user_data.get("initials")
email = user_data.get("email")
user_from_config = User(name, initials, email)
if user_from_config not in users_from_db:
session.add(user_from_config)
# ------------------------------------------------------
# add the VersionTypes
from oyProjectManager.models.version import VersionType
version_types_from_db = query(VersionType).all()
for version_type in conf.version_types:
version_type_from_conf = VersionType(**version_type)
if version_type_from_conf not in version_types_from_db:
session.add(version_type_from_conf)
session.commit()
logger.debug("finished initialization of the db")
def setUp(self):
"""setup the tests
"""
# -----------------------------------------------------------------
# start of the setUp
# create the environment variable and point it to a temp directory
conf.database_url = "sqlite://"
self.temp_config_folder = tempfile.mkdtemp()
self.temp_projects_folder = tempfile.mkdtemp()
os.environ["OYPROJECTMANAGER_PATH"] = self.temp_config_folder
os.environ[conf.repository_env_key] = self.temp_projects_folder
# create a test project
self.project = Project("Test Project")
self.project.create()
# create a test asset
self.asset1 = Asset(self.project, "Test Asset 1")
# version type
self.asset_vtypes = VersionType.query()\
.filter(VersionType.type_for == "Asset").all()
self.shot_vtypes = VersionType.query()\
.filter(VersionType.type_for == "Shot").all()
self.user1 = User(name="Test User 1", email="*****@*****.**")
# create a test version
self.kwargs = {
"version_of": self.asset1,
"base_name": self.asset1.code,
"type": self.asset_vtypes[0],
"created_by": self.user1,
"extension": "ma"
}
self.version1 = Version(**self.kwargs)
self.version1.save()
# create the environment instance
self.mEnv = mayaEnv.Maya()
# just renew the scene
pm.newFile(force=True)
def test_shots_tableWidget_is_filled_with_Shot_data(self):
"""testing if the shots_tableWidget is filled with shot data properly
"""
db.setup()
# create a project
proj1 = Project('Test Project 1')
proj1.save()
proj2 = Project('Test Project 2')
proj2.save()
shot_vtypes = VersionType.query()\
.filter(VersionType.type_for=='Shot')\
.order_by(VersionType.name)\
.all()
admin = User.query().first()
# seqs for proj1
seq1 = Sequence(proj1, 'Test Seq 1')
seq1.save()
seq2 = Sequence(proj1, 'Test Seq 2')
seq2.save()
# seqs for proj2
seq3 = Sequence(proj2, 'Test Seq 3')
seq3.save()
seq4 = Sequence(proj2, 'Test Seq 4')
seq4.save()
# shots for seq1
shot1 = Shot(seq1, 1)
shot1.save()
shot2 = Shot(seq1, 2)
shot2.save()
shot3 = Shot(seq1, 3)
shot3.save()
# shots for seq2
shot4 = Shot(seq2, 4)
shot4.save()
shot5 = Shot(seq2, 5)
shot5.save()
shot6 = Shot(seq2, 6)
shot6.save()
# shots for seq3
shot7 = Shot(seq3, 7)
shot7.save()
shot8 = Shot(seq3, 8)
shot8.save()
# shots for seq4
shot9 = Shot(seq4, 9)
shot9.save()
shot10 = Shot(seq4, 10)
shot10.save()
# versions for shot1
version1 = Version(version_of=shot1,
base_name=shot1.code,
type=shot_vtypes[0],
created_by=admin,
status=conf.status_list[0])
version1.save()
version2 = Version(version_of=shot1,
base_name=shot1.code,
type=shot_vtypes[1],
created_by=admin,
status=conf.status_list[1])
version2.save()
# versions for shot2
version3 = Version(version_of=shot2,
base_name=shot2.code,
type=shot_vtypes[2],
created_by=admin,
status=conf.status_list[2])
version3.save()
version4 = Version(
version_of=shot2,
base_name=shot2.code,
type=shot_vtypes[3],
created_by=admin,
status=conf.status_list[3],
)
version4.save()
# versions for shot3
version5 = Version(
version_of=shot3,
base_name=shot3.code,
type=shot_vtypes[4],
created_by=admin,
status=conf.status_list[4],
)
version5.save()
version6 = Version(
version_of=shot3,
base_name=shot3.code,
type=shot_vtypes[5],
created_by=admin,
status=conf.status_list[4],
)
version6.save()
# versions for shot4
version7 = Version(version_of=shot4,
base_name=shot4.code,
type=shot_vtypes[5],
created_by=admin,
status=conf.status_list[4])
version7.save()
version8 = Version(version_of=shot4,
base_name=shot4.code,
type=shot_vtypes[5],
created_by=admin,
status=conf.status_list[0])
version8.save()
dialog = status_manager.MainDialog()
# self.show_dialog(dialog)
# start tests
# set the project to project1
dialog.projects_comboBox.setCurrentIndex(0)
#self.show_dialog(dialog)
# asset1's vtypes[0] vtypes[1] vtypes[2] vtypes[3]
self.fail('test is not finished yet!')
def test_assets_tableWidget_is_filled_with_Asset_data(self):
"""testing if the assets_tableWidget is filled with asset data properly
"""
db.setup()
# create a project
proj1 = Project('Test Project 1')
proj1.save()
proj2 = Project('Test Project 2')
proj2.save()
asset_vtypes = VersionType.query()\
.filter(VersionType.type_for=='Asset')\
.order_by(VersionType.name)\
.all()
admin = User.query().first()
asset1 = Asset(proj1, 'Test Asset 1', type='Char')
asset1.save()
asset2 = Asset(proj1, 'Test Asset 2', type='Prop')
asset2.save()
asset3 = Asset(proj1, 'Test Asset 3', type='Environment')
asset3.save()
asset4 = Asset(proj1, 'Test Asset 4', type='Prop')
asset4.save()
# versions for asset1
version1 = Version(version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[0],
created_by=admin,
status=conf.status_list[0])
version1.save()
version2 = Version(version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[1],
created_by=admin,
status=conf.status_list[1])
version2.save()
version3 = Version(version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[2],
created_by=admin,
status=conf.status_list[2])
version3.save()
# version for asset1 with different takes
version4 = Version(version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[3],
created_by=admin,
status=conf.status_list[3],
take_name='Test_A')
version4.save()
version5 = Version(version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[4],
created_by=admin,
status=conf.status_list[4],
take_name='Test_A')
version5.save()
version6 = Version(version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[5],
created_by=admin,
status=conf.status_list[4],
take_name='Test_B')
version6.save()
version7 = Version(version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[5],
created_by=admin,
status=conf.status_list[4],
take_name='Test_B')
version7.save()
dialog = status_manager.MainDialog()
# self.show_dialog(dialog)
# start tests
# What we should see shoul be:
#
# +-----------+-------------+--------------+--------+---------------------+
# | Thumbnail | Type | Name | Take | Asset Version Types |
# +===========+=============+==============+========+=====================+
# | (IMAGE) | Char | Test Asset 1 | MAIN | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Char | Test Asset 1 | Test_A | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Char | Test Asset 1 | Test_B | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Environment | Test Asset 3 | ---- | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Prop | Test Asset 2 | ---- | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Prop | Test Asset 4 | ---- | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
#
# set the project to project1
dialog.projects_comboBox.setCurrentIndex(0)
# asset1's vtypes[0] vtypes[1] vtypes[2] vtypes[3]
# set to assets
dialog.tabWidget.setCurrentIndex(0)
#self.show_dialog(dialog)
# expect to see 6 rows
self.assertEqual(6, dialog.assets_tableWidget.rowCount())
# expect the assets types listed in the first column
# first three should be Char
dialog.assets_tableWidget.setCurrentCell(0, 1)
self.assertEqual('Char',
dialog.assets_tableWidget.currentItem().text())
dialog.assets_tableWidget.setCurrentCell(1, 1)
self.assertEqual('Char',
dialog.assets_tableWidget.currentItem().text())
dialog.assets_tableWidget.setCurrentCell(2, 1)
self.assertEqual('Char',
dialog.assets_tableWidget.currentItem().text())
# next should be Environment
dialog.assets_tableWidget.setCurrentCell(3, 1)
self.assertEqual('Environment',
dialog.assets_tableWidget.currentItem().text())
# the next two should be Prop
dialog.assets_tableWidget.setCurrentCell(4, 1)
self.assertEqual('Prop',
dialog.assets_tableWidget.currentItem().text())
dialog.assets_tableWidget.setCurrentCell(5, 1)
self.assertEqual('Prop',
dialog.assets_tableWidget.currentItem().text())
def test_shots_tableWidget_is_filled_with_Shot_data(self):
"""testing if the shots_tableWidget is filled with shot data properly
"""
db.setup()
# create a project
proj1 = Project('Test Project 1')
proj1.save()
proj2 = Project('Test Project 2')
proj2.save()
shot_vtypes = VersionType.query()\
.filter(VersionType.type_for=='Shot')\
.order_by(VersionType.name)\
.all()
admin = User.query().first()
# seqs for proj1
seq1 = Sequence(proj1, 'Test Seq 1')
seq1.save()
seq2 = Sequence(proj1, 'Test Seq 2')
seq2.save()
# seqs for proj2
seq3 = Sequence(proj2, 'Test Seq 3')
seq3.save()
seq4 = Sequence(proj2, 'Test Seq 4')
seq4.save()
# shots for seq1
shot1 = Shot(seq1, 1)
shot1.save()
shot2 = Shot(seq1, 2)
shot2.save()
shot3 = Shot(seq1, 3)
shot3.save()
# shots for seq2
shot4 = Shot(seq2, 4)
shot4.save()
shot5 = Shot(seq2, 5)
shot5.save()
shot6 = Shot(seq2, 6)
shot6.save()
# shots for seq3
shot7 = Shot(seq3, 7)
shot7.save()
shot8 = Shot(seq3, 8)
shot8.save()
# shots for seq4
shot9 = Shot(seq4, 9)
shot9.save()
shot10 = Shot(seq4, 10)
shot10.save()
# versions for shot1
version1 = Version(
version_of=shot1,
base_name=shot1.code,
type=shot_vtypes[0],
created_by=admin,
status=conf.status_list[0]
)
version1.save()
version2 = Version(
version_of=shot1,
base_name=shot1.code,
type=shot_vtypes[1],
created_by=admin,
status=conf.status_list[1]
)
version2.save()
# versions for shot2
version3 = Version(
version_of=shot2,
base_name=shot2.code,
type=shot_vtypes[2],
created_by=admin,
status=conf.status_list[2]
)
version3.save()
version4 = Version(
version_of=shot2,
base_name=shot2.code,
type=shot_vtypes[3],
created_by=admin,
status=conf.status_list[3],
)
version4.save()
# versions for shot3
version5 = Version(
version_of=shot3,
base_name=shot3.code,
type=shot_vtypes[4],
created_by=admin,
status=conf.status_list[4],
)
version5.save()
version6 = Version(
version_of=shot3,
base_name=shot3.code,
type=shot_vtypes[5],
created_by=admin,
status=conf.status_list[4],
)
version6.save()
# versions for shot4
version7 = Version(
version_of=shot4,
base_name=shot4.code,
type=shot_vtypes[5],
created_by=admin,
status=conf.status_list[4]
)
version7.save()
version8 = Version(
version_of=shot4,
base_name=shot4.code,
type=shot_vtypes[5],
created_by=admin,
status=conf.status_list[0]
)
version8.save()
dialog = status_manager.MainDialog()
# self.show_dialog(dialog)
# start tests
# set the project to project1
dialog.projects_comboBox.setCurrentIndex(0)
#self.show_dialog(dialog)
# asset1's vtypes[0] vtypes[1] vtypes[2] vtypes[3]
self.fail('test is not finished yet!')
def test_assets_tableWidget_is_filled_with_Asset_data(self):
"""testing if the assets_tableWidget is filled with asset data properly
"""
db.setup()
# create a project
proj1 = Project('Test Project 1')
proj1.save()
proj2 = Project('Test Project 2')
proj2.save()
asset_vtypes = VersionType.query()\
.filter(VersionType.type_for=='Asset')\
.order_by(VersionType.name)\
.all()
admin = User.query().first()
asset1 = Asset(proj1, 'Test Asset 1', type='Char')
asset1.save()
asset2 = Asset(proj1, 'Test Asset 2', type='Prop')
asset2.save()
asset3 = Asset(proj1, 'Test Asset 3', type='Environment')
asset3.save()
asset4 = Asset(proj1, 'Test Asset 4', type='Prop')
asset4.save()
# versions for asset1
version1 = Version(
version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[0],
created_by=admin,
status=conf.status_list[0]
)
version1.save()
version2 = Version(
version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[1],
created_by=admin,
status=conf.status_list[1]
)
version2.save()
version3 = Version(
version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[2],
created_by=admin,
status=conf.status_list[2]
)
version3.save()
# version for asset1 with different takes
version4 = Version(
version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[3],
created_by=admin,
status=conf.status_list[3],
take_name='Test_A'
)
version4.save()
version5 = Version(
version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[4],
created_by=admin,
status=conf.status_list[4],
take_name='Test_A'
)
version5.save()
version6 = Version(
version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[5],
created_by=admin,
status=conf.status_list[4],
take_name='Test_B'
)
version6.save()
version7 = Version(
version_of=asset1,
base_name=asset1.code,
type=asset_vtypes[5],
created_by=admin,
status=conf.status_list[4],
take_name='Test_B'
)
version7.save()
dialog = status_manager.MainDialog()
# self.show_dialog(dialog)
# start tests
# What we should see shoul be:
#
# +-----------+-------------+--------------+--------+---------------------+
# | Thumbnail | Type | Name | Take | Asset Version Types |
# +===========+=============+==============+========+=====================+
# | (IMAGE) | Char | Test Asset 1 | MAIN | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Char | Test Asset 1 | Test_A | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Char | Test Asset 1 | Test_B | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Environment | Test Asset 3 | ---- | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Prop | Test Asset 2 | ---- | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
# | (IMAGE) | Prop | Test Asset 4 | ---- | ******************* |
# +-----------+-------------+--------------+--------+---------------------+
#
# set the project to project1
dialog.projects_comboBox.setCurrentIndex(0)
# asset1's vtypes[0] vtypes[1] vtypes[2] vtypes[3]
# set to assets
dialog.tabWidget.setCurrentIndex(0)
#self.show_dialog(dialog)
# expect to see 6 rows
self.assertEqual(6, dialog.assets_tableWidget.rowCount())
# expect the assets types listed in the first column
# first three should be Char
dialog.assets_tableWidget.setCurrentCell(0, 1)
self.assertEqual('Char', dialog.assets_tableWidget.currentItem().text())
dialog.assets_tableWidget.setCurrentCell(1, 1)
self.assertEqual('Char', dialog.assets_tableWidget.currentItem().text())
dialog.assets_tableWidget.setCurrentCell(2, 1)
self.assertEqual('Char', dialog.assets_tableWidget.currentItem().text())
# next should be Environment
dialog.assets_tableWidget.setCurrentCell(3, 1)
self.assertEqual(
'Environment',
dialog.assets_tableWidget.currentItem().text()
)
# the next two should be Prop
dialog.assets_tableWidget.setCurrentCell(4, 1)
self.assertEqual(
'Prop',
dialog.assets_tableWidget.currentItem().text()
)
dialog.assets_tableWidget.setCurrentCell(5, 1)
self.assertEqual(
'Prop',
dialog.assets_tableWidget.currentItem().text()
)
