def test_storage_roots_get_local_storages(self):
"""Test the get_local_storages method."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
(single_root_lookup,
unmapped_roots) = single_root.get_local_storages(self.mockgun)
self.assertTrue("primary" in single_root_lookup)
single_root_default = single_root_lookup["primary"]
self.assertEqual(single_root_default["code"], "primary")
self.assertEqual(single_root_default["type"], "LocalStorage")
self.assertEqual(single_root_default["id"], 1)
single_root_storage_paths = single_root.as_shotgun_paths["primary"]
self.assertEqual(single_root_storage_paths,
ShotgunPath.from_shotgun_dict(single_root_default))
self.assertEqual(unmapped_roots, [])
multiple_roots = StorageRoots.from_metadata(
self._multiple_roots_metadata)
(multiple_root_lookup,
unmapped_roots) = multiple_roots.get_local_storages(self.mockgun)
self.assertTrue("work" in multiple_root_lookup)
multiple_roots_default = multiple_root_lookup["work"]
self.assertEqual(multiple_roots_default["code"], "work")
self.assertEqual(multiple_roots_default["type"], "LocalStorage")
self.assertEqual(multiple_roots_default["id"], 2)
multiple_roots_storage_paths = multiple_roots.as_shotgun_paths["work"]
self.assertEqual(multiple_roots_storage_paths,
ShotgunPath.from_shotgun_dict(multiple_roots_default))
self.assertEqual(unmapped_roots, ["foobar"])
def test_storage_roots_get_local_storages(self):
"""Test the get_local_storages method."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
(single_root_lookup, unmapped_roots) = single_root.get_local_storages(self.mockgun)
self.assertTrue("primary" in single_root_lookup)
single_root_default = single_root_lookup["primary"]
self.assertEqual(single_root_default["code"], "primary")
self.assertEqual(single_root_default["type"], "LocalStorage")
self.assertEqual(single_root_default["id"], 1)
single_root_storage_paths = single_root.as_shotgun_paths["primary"]
self.assertEqual(single_root_storage_paths, ShotgunPath.from_shotgun_dict(single_root_default))
self.assertEqual(unmapped_roots, [])
multiple_roots = StorageRoots.from_metadata(self._multiple_roots_metadata)
(multiple_root_lookup, unmapped_roots) = multiple_roots.get_local_storages(self.mockgun)
self.assertTrue("work" in multiple_root_lookup)
multiple_roots_default = multiple_root_lookup["work"]
self.assertEqual(multiple_roots_default["code"], "work")
self.assertEqual(multiple_roots_default["type"], "LocalStorage")
self.assertEqual(multiple_roots_default["id"], 2)
multiple_roots_storage_paths = multiple_roots.as_shotgun_paths["work"]
self.assertEqual(multiple_roots_storage_paths, ShotgunPath.from_shotgun_dict(multiple_roots_default))
self.assertEqual(unmapped_roots, ["foobar"])
def test_storage_roots_from_metadata(self):
"""Test the from_metadata factory class method."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
self.assertIsInstance(single_root, StorageRoots)
multiple_roots = StorageRoots.from_metadata(
self._multiple_roots_metadata)
self.assertIsInstance(multiple_roots, StorageRoots)
no_roots = StorageRoots.from_metadata(self._no_roots_metadata)
self.assertIsInstance(no_roots, StorageRoots)
def test_storage_roots_metadata(self):
"""Test the metadata property."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
self.assertEqual(single_root.metadata, self._single_root_metadata)
multiple_roots = StorageRoots.from_metadata(
self._multiple_roots_metadata)
self.assertEqual(multiple_roots.metadata,
self._multiple_roots_metadata)
no_roots = StorageRoots.from_metadata(self._no_roots_metadata)
self.assertEqual(no_roots.metadata, self._no_roots_metadata)
def test_storage_roots_metadata(self):
"""Test the metadata property."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
self.assertEqual(single_root.metadata, self._single_root_metadata)
multiple_roots = StorageRoots.from_metadata(self._multiple_roots_metadata)
self.assertEqual(multiple_roots.metadata, self._multiple_roots_metadata)
empty_roots = StorageRoots.from_metadata(self._empty_roots_metadata)
self.assertEqual(empty_roots.metadata, self._empty_roots_metadata)
no_roots = StorageRoots.from_metadata(self._no_roots_metadata)
self.assertEqual(no_roots.metadata, self._no_roots_metadata)
def test_storage_roots_from_metadata(self):
"""Test the from_metadata factory class method."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
self.assertIsInstance(single_root, StorageRoots)
multiple_roots = StorageRoots.from_metadata(self._multiple_roots_metadata)
self.assertIsInstance(multiple_roots, StorageRoots)
empty_roots = StorageRoots.from_metadata(self._empty_roots_metadata)
self.assertIsInstance(empty_roots, StorageRoots)
no_roots = StorageRoots.from_metadata(self._no_roots_metadata)
self.assertIsInstance(no_roots, StorageRoots)
def test_update_root(self):
"""Tests the update_root method."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
single_root.update_root(
"primary",
{
"linux_path": "/tmp/foobar",
"mac_path": "/tmp/foobar",
"windows_path": "X:\\tmp\\foobar",
"shotgun_storage_id": 1,
"default": True,
},
)
self.assertEqual(
single_root.metadata,
{
"primary": {
"linux_path": "/tmp/foobar",
"mac_path": "/tmp/foobar",
"windows_path": "X:\\tmp\\foobar",
"shotgun_storage_id": 1,
"default": True,
}
},
)
def test_update_root(self):
"""Tests the update_root method."""
single_root = StorageRoots.from_metadata(self._single_root_metadata)
single_root.update_root(
"primary",
{
"linux_path": "/tmp/foobar",
"mac_path": "/tmp/foobar",
"windows_path": "X:\\tmp\\foobar",
"shotgun_storage_id": 1,
"default": True
}
)
self.assertEqual(
single_root.metadata,
{
"primary": {
"linux_path": "/tmp/foobar",
"mac_path": "/tmp/foobar",
"windows_path": "X:\\tmp\\foobar",
"shotgun_storage_id": 1,
"default": True
}
}
)
def test_storage_roots_populate_defaults(self):
"""Test the populate_defaults method."""
empty_roots_metadata = {}
empty_roots = StorageRoots.from_metadata(empty_roots_metadata)
empty_roots.populate_defaults()
self.assertTrue("primary" in empty_roots.metadata)
self.assertEqual(
empty_roots.metadata["primary"],
{
"description":
"Default location where project data is stored.",
"mac_path": "/studio/projects",
"linux_path": "/studio/projects",
"windows_path": "\\\\network\\projects",
"default": True,
},
)
partial_roots_metadata = {
"foobar": {
"description": "Foobar",
"mac_path": "/studio/projects",
"default": True,
}
}
partial_roots = StorageRoots.from_metadata(partial_roots_metadata)
partial_roots.populate_defaults()
self.assertEqual(
partial_roots.metadata,
{
"foobar": {
"description": "Foobar",
"mac_path": "/studio/projects",
"linux_path": None,
"windows_path": None,
"default": True,
}
},
)
def test_storage_roots_populate_defaults(self):
"""Test the populate_defaults method."""
empty_roots_metadata = {}
empty_roots = StorageRoots.from_metadata(empty_roots_metadata)
empty_roots.populate_defaults()
self.assertTrue("primary" in empty_roots.metadata)
self.assertEqual(
empty_roots.metadata["primary"],
{
"description": "Default location where project data is stored.",
"mac_path": "/studio/projects",
"linux_path": "/studio/projects",
"windows_path": "\\\\network\\projects",
"default": True,
}
)
partial_roots_metadata = {
"foobar": {
"description": "Foobar",
"mac_path": "/studio/projects",
"default": True,
}
}
partial_roots = StorageRoots.from_metadata(partial_roots_metadata)
partial_roots.populate_defaults()
self.assertEqual(
partial_roots.metadata,
{
"foobar": {
"description": "Foobar",
"mac_path": "/studio/projects",
"linux_path": None,
"windows_path": None,
"default": True,
}
}
)
