def find_asset(self, dbo):
"""Test find_asset, CRUD method
:param dbObject dbo: dbo
"""
self.reset(dbo)
# 1. find find_asset from db
with orm.db_session:
temp_asset, _ = Asset.find_asset_by_id(self.asset.id)
# 2. test value
self.assert_value(temp_asset)
temp_asset, err = Asset.find_asset_by_id(-1)
# 3. assert on error
self.assertEqual(err, "Asset Not Found !")
self.assertEqual(temp_asset, None)
# 4. find_all asset from db
temp_assets = Asset.find_all_assets()
# 5. test value
self.assertEqual(len(temp_assets), 1)
self.assert_value(temp_assets[0])
def remove_asset(self, dbo):
"""Test remove_asset, CRUD method
:param dbObject dbo: dbo
"""
self.reset(dbo)
with orm.db_session:
# 1. find asset from db
temp_asset, _ = Asset.find_asset_by_id(self.asset.id)
# 2. remove
Asset.remove_asset_by_id(temp_asset.id)
# 3. re-get
temp_asset, err = Asset.find_asset_by_id(self.asset.id)
# 4. assert
self.assertEqual(temp_asset, None)
self.assertEqual("Asset Not Found !", err)
def update_asset(self, dbo):
"""Test update_asset, CRUD method
:param dbObject dbo: dbo
"""
self.reset(dbo)
with orm.db_session:
# 1. find update_asset from db
temp_asset, _ = Asset.find_asset_by_id(self.asset.id)
temp_asset.name = "updated_name"
temp_asset.lod = 100
temp_asset, _ = Asset.update_asset_by_id(temp_asset.id, temp_asset)
# 2. assert
self.assertEqual("updated_name", temp_asset.name)
self.assertEqual(100, temp_asset.lod)