import unittest
import pymel.core as pm
import artv2.components.base_components.rig_asset as rig_asset


class Test_Rig_Assets(unittest.TestCase):

    def test_create_and_rename_asset(self):
        new_asset = rig_asset.RigAsset()
        new_asset.name = "test"
        self.assertEquals(new_asset.name, "test")

    def test_create_multiple_assets(self):
        asset1 = rig_asset.RigAsset()
        self.assertEquals(asset1.name, "asset")

        with self.assertRaises(RuntimeError):
            asset2 = rig_asset.RigAsset()

    def test_delete_asset(self):

        asset = rig_asset.RigAsset()

        with self.assertRaises(RuntimeError):
            pm.delete(asset.network_node)

        exists = pm.objExists(asset.network_node)
        self.assertEquals(exists, True)

    def test_invalid_asset_name(self):
        new_asset = rig_asset.RigAsset()

        with self.assertRaises(RuntimeError):
            new_asset.name = "123test"

        with self.assertRaises(RuntimeError):
            new_asset.name = "x-man"

        new_asset.name = "dummy"
        self.assertEquals(new_asset.name, "dummy")

    def test_access_asset_details(self):

        new_asset = rig_asset.RigAsset()
        new_asset.name = "dummy"
        network_node = new_asset.network_node

        access_asset = rig_asset.RigAsset(network_node)
        self.assertEquals(access_asset.name, "dummy")