Test get_omero_userdir

test/unit/test_util.py

@@ -39,7 +39,7 @@
 from omero.util.text import CSVStyle, JSONStyle, PlainStyle, TableBuilder
 from omero.util.upgrade_check import UpgradeCheck
 from omero.util.temp_files import manager
-from omero.util import get_user_dir
+from omero.util import get_omero_userdir, get_user_dir
 from omero_version import omero_version
 import omero.util.image_utils as image_utils
 try:
@@ -314,3 +314,20 @@ def test_paste_image(self):
         data_canvas[256, 256] = [255, 255, 0]
         canvas = Image.fromarray(data_canvas, 'RGB')
         image_utils.paste_image(img, canvas, 0, 0)
+
+
+class TestUserdirs(object):
+
+    def testUserdirEnvironmentDefault(self, monkeypatch):
+        monkeypatch.delenv('OMERO_USERDIR', raising=False)
+        assert get_omero_userdir().basename() == 'omero'
+
+    def testUserdirEnvironmentAppdir(self, monkeypatch):
+        monkeypatch.setenv('OMERO_USERDIR', '')
+        d = get_omero_userdir()
+        assert d.basename() == omero_version.split('.')[0]
+        assert 'OMERO.cli' in str(d)
+
+    def testUserdirEnvironmentSet(self, monkeypatch, tmpdir):
+        monkeypatch.setenv('OMERO_USERDIR', str(tmpdir))
+        assert get_omero_userdir() == tmpdir