✨ Setup initial unit tests for conductor

pros/conductor/conductor.py

@@ -5,7 +5,7 @@
 from typing import *
 
 import click
-from semantic_version import Spec, Version
+from semantic_version import Spec, Version, SimpleSpec
 
 from pros.common import *
 from pros.conductor.project import TemplateAction
@@ -191,7 +191,7 @@ def resolve_template(self, identifier: Union[str, BaseTemplate], **kwargs) -> Op
         logger(__name__).info(f'Candidates: {", ".join([str(t) for t in templates])}')
         if not any(templates):
             return None
-        query.version = str(Spec(query.version or '>0').select([Version(t.version) for t in templates]))
+        query.version = str(SimpleSpec(query.version or '>0').select([Version(t.version) for t in templates]))
         v = Version(query.version)
         v.prerelease = v.prerelease if len(v.prerelease) else ('',)
         v.build = v.build if len(v.build) else ('',)

pros/conductor/templates/base_template.py

@@ -1,6 +1,6 @@
 from typing import *
 
-from semantic_version import Spec, Version
+from semantic_version import Spec, Version, SimpleSpec
 
 from pros.common import ui
 
@@ -30,7 +30,7 @@ def satisfies(self, query: 'BaseTemplate', kernel_version: Union[str, Version] =
             return False
         if query.target and self.target != query.target:
             return False
-        if query.version and Version(self.version) not in Spec(query.version):
+        if query.version and Version(self.version) not in SimpleSpec(query.version):
             return False
         if kernel_version and isinstance(kernel_version, str):
             kernel_version = Version(kernel_version)
@@ -62,6 +62,12 @@ def __eq__(self, other):
         else:
             return super().__eq__(other)
 
+    def __ge__(self, other):
+        if isinstance(other, BaseTemplate):
+            return self.identifier >= other.identifier
+        else:
+            return super().__ge__(other)
+
     def __hash__(self):
         return self.identifier.__hash__()
 

test/test_conductor.py

@@ -0,0 +1,19 @@
+from pros.conductor import Conductor
+from pros.conductor.templates import LocalTemplate 
+import unittest
+
+
+class TestConductor(unittest.TestCase):
+    ### using __init__() with the unittest module causes errors, setUp is recognized by the module and should be used instead.
+    #   setUp will run before every unit test
+    def setUp(self):
+        self._conductor = Conductor()
+
+    def test_resolve_template(self):
+        okapilib_template = LocalTemplate(name="okapilib", version="5.0.0")
+        resolved_template = self._conductor.resolve_template(identifier="okapilib")
+        self.assertTrue(resolved_template >= okapilib_template)        
+
+
+if __name__ == "__main__":
+    unittest.main()