From 919540038ff969c966acb7fe098cc64da77363d7 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakubjezek001@gmail.com>
Date: Wed, 4 Oct 2023 13:24:41 +0200
Subject: [PATCH] unit testing of labeling

---
 .../pipeline/test_colorspace_labels.py        | 81 +++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 tests/unit/openpype/pipeline/test_colorspace_labels.py

diff --git a/tests/unit/openpype/pipeline/test_colorspace_labels.py b/tests/unit/openpype/pipeline/test_colorspace_labels.py
new file mode 100644
index 0000000000..a135c3258b
--- /dev/null
+++ b/tests/unit/openpype/pipeline/test_colorspace_labels.py
@@ -0,0 +1,81 @@
+import unittest
+from unittest.mock import patch
+from openpype.pipeline.colorspace import get_labeled_colorspaces
+
+
+class TestGetLabeledColorspaces(unittest.TestCase):
+    @patch('openpype.pipeline.colorspace.get_ocio_config_colorspaces')
+    def test_returns_list_of_tuples(self, mock_get_ocio_config_colorspaces):
+        mock_get_ocio_config_colorspaces.return_value = {
+            'colorspace': {
+                'sRGB': {},
+                'Rec.709': {},
+            },
+            'look': {
+                'sRGB to Rec.709': {
+                    'process_space': 'Rec.709',
+                },
+            },
+            'role': {
+                'reference': {
+                    'colorspace': 'sRGB',
+                },
+            },
+        }
+        result = get_labeled_colorspaces('config.ocio')
+        self.assertIsInstance(result, list)
+        self.assertTrue(all(isinstance(item, tuple) for item in result))
+
+    @patch('openpype.pipeline.colorspace.get_ocio_config_colorspaces')
+    def test_includes_colorspaces(self, mock_get_ocio_config_colorspaces):
+        mock_get_ocio_config_colorspaces.return_value = {
+            'colorspace': {
+                'sRGB': {}
+            },
+            'look': {},
+            'role': {},
+        }
+        result = get_labeled_colorspaces('config.ocio', include_aliases=False, include_looks=False, include_roles=False)
+        self.assertEqual(result, [('sRGB', '[colorspace] sRGB')])
+
+    @patch('openpype.pipeline.colorspace.get_ocio_config_colorspaces')
+    def test_includes_aliases(self, mock_get_ocio_config_colorspaces):
+        mock_get_ocio_config_colorspaces.return_value = {
+            'colorspace': {
+                'sRGB': {
+                    'aliases': ['sRGB (D65)'],
+                },
+            },
+            'look': {},
+            'role': {},
+        }
+        result = get_labeled_colorspaces('config.ocio', include_aliases=True, include_looks=False, include_roles=False)
+        self.assertEqual(result, [('sRGB', '[colorspace] sRGB'), ('sRGB (D65)', '[alias] sRGB (D65) (sRGB)')])
+
+    @patch('openpype.pipeline.colorspace.get_ocio_config_colorspaces')
+    def test_includes_looks(self, mock_get_ocio_config_colorspaces):
+        mock_get_ocio_config_colorspaces.return_value = {
+            'colorspace': {},
+            'look': {
+                'sRGB to Rec.709': {
+                    'process_space': 'Rec.709',
+                },
+            },
+            'role': {},
+        }
+        result = get_labeled_colorspaces('config.ocio', include_aliases=False, include_looks=True, include_roles=False)
+        self.assertEqual(result, [('sRGB to Rec.709', '[look] sRGB to Rec.709 (Rec.709)')])
+
+    @patch('openpype.pipeline.colorspace.get_ocio_config_colorspaces')
+    def test_includes_roles(self, mock_get_ocio_config_colorspaces):
+        mock_get_ocio_config_colorspaces.return_value = {
+            'colorspace': {},
+            'look': {},
+            'role': {
+                'reference': {
+                    'colorspace': 'sRGB',
+                },
+            },
+        }
+        result = get_labeled_colorspaces('config.ocio', include_aliases=False, include_looks=False, include_roles=True)
+        self.assertEqual(result, [('reference', '[role] reference (sRGB)')])