Skip to content

Commit

Permalink
Correct version comparison in Client.supports_fakeroot.
Browse files Browse the repository at this point in the history
Fixing type error that was caused be comparing wrong types.
  • Loading branch information
sergey-serebryakov committed Aug 2, 2023
1 parent c400698 commit 12f8948
Show file tree
Hide file tree
Showing 2 changed files with 45 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -104,16 +104,18 @@ def from_env(cls) -> "Client":
def supports_fakeroot(self) -> bool:
singularity_35 = (
self.version.runtime == Runtime.SINGULARITY
and self.version >= semver.VersionInfo(major=3, minor=5)
and self.version.version >= semver.VersionInfo(major=3, minor=5)
)
apptainer = self.version.runtime == Runtime.APPTAINER
return singularity_35 or apptainer

def __init__(self, singularity: t.Union[str, t.List]) -> None:
def __init__(
self, singularity: t.Union[str, t.List], version: t.Optional[Version] = None
) -> None:
if isinstance(singularity, str):
singularity = singularity.split(" ")
self.singularity: t.List[str] = [c.strip() for c in singularity if c.strip()]
self.version: t.Optional[Version] = None
self.version: t.Optional[Version] = version
self.init()
logger.debug(
"Client.__init__ executable=%s, version=%s", self.singularity, self.version
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
from unittest import TestCase

import semver
from mlcube_singularity.singularity_client import Client, Runtime, Version


class TestSingularityRunner(TestCase):
def test___init__(self) -> None:
client = Client(
"sudo singularity", Version(Runtime.APPTAINER, semver.VersionInfo(3, 7, 5))
)
self.assertListEqual(["sudo", "singularity"], client.singularity)
self.assertEqual(Runtime.APPTAINER, client.version.runtime)
self.assertEqual(3, client.version.version.major)
self.assertEqual(7, client.version.version.minor)
self.assertEqual(5, client.version.version.patch)

def test_supports_fakeroot(self) -> None:
client = Client(
"sudo singularity", Version(Runtime.APPTAINER, semver.VersionInfo(3, 7, 5))
)
self.assertTrue(client.supports_fakeroot())

client = Client(
"sudo singularity",
Version(Runtime.SINGULARITY, semver.VersionInfo(3, 7, 5)),
)
self.assertTrue(client.supports_fakeroot())

client = Client(
"sudo singularity",
Version(Runtime.SINGULARITY, semver.VersionInfo(3, 5, 0)),
)
self.assertTrue(client.supports_fakeroot())

client = Client(
"sudo singularity",
Version(Runtime.SINGULARITY, semver.VersionInfo(3, 4, 9)),
)
self.assertFalse(client.supports_fakeroot())

0 comments on commit 12f8948

Please sign in to comment.