91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
# Copyright 2022 Canonical, Ltd.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
# from unittest.mock import Mock
|
|
|
|
from subiquitycore.tests import SubiTestCase
|
|
from subiquitycore.utils import _zsys_uuid_charset, gen_zsys_uuid, orig_environ
|
|
|
|
|
|
class TestOrigEnviron(SubiTestCase):
|
|
def test_empty(self):
|
|
env = {}
|
|
expected = env
|
|
self.assertEqual(expected, orig_environ(env))
|
|
|
|
def test_orig_path(self):
|
|
env = {"PATH": "a", "PATH_ORIG": "b"}
|
|
expected = {"PATH": "b"}
|
|
self.assertEqual(expected, orig_environ(env))
|
|
|
|
def test_not_this_key(self):
|
|
env = {"PATH": "a", "PATH_ORIG_AAAAA": "b"}
|
|
expected = env
|
|
self.assertEqual(expected, orig_environ(env))
|
|
|
|
def test_remove_empty_key(self):
|
|
env = {"STUFF": "a", "STUFF_ORIG": ""}
|
|
expected = {}
|
|
self.assertEqual(expected, orig_environ(env))
|
|
|
|
def test_no_ld_library_path(self):
|
|
env = {"LD_LIBRARY_PATH": "a"}
|
|
expected = {}
|
|
self.assertEqual(expected, orig_environ(env))
|
|
|
|
def test_practical(self):
|
|
snap = "/snap/subiquity/1234"
|
|
env = {
|
|
"TERM": "linux",
|
|
"LD_LIBRARY_PATH": "/var/lib/snapd/lib/gl",
|
|
"PYTHONIOENCODING_ORIG": "",
|
|
"PYTHONIOENCODING": "utf-8",
|
|
"SUBIQUITY_ROOT_ORIG": "",
|
|
"SUBIQUITY_ROOT": snap,
|
|
"PYTHON_ORIG": "",
|
|
"PYTHON": f"{snap}/usr/bin/python3.10",
|
|
"PYTHONPATH_ORIG": "",
|
|
"PYTHONPATH": f"{snap}/stuff/things",
|
|
"PY3OR2_PYTHON_ORIG": "",
|
|
"PY3OR2_PYTHON": f"{snap}/usr/bin/python3.10",
|
|
"PATH_ORIG": "/usr/bin:/bin",
|
|
"PATH": "/usr/bin:/bin:/snap/bin",
|
|
}
|
|
expected = {
|
|
"TERM": "linux",
|
|
"PATH": "/usr/bin:/bin",
|
|
}
|
|
self.assertEqual(expected, orig_environ(env))
|
|
|
|
|
|
class TestZsysUUID(SubiTestCase):
|
|
def test_charset(self):
|
|
charset = _zsys_uuid_charset()
|
|
for c in "0", "9", "a", "z":
|
|
self.assertIn(c, charset)
|
|
bads = [
|
|
chr(ord("0") - 1),
|
|
chr(ord("9") + 1),
|
|
chr(ord("a") - 1),
|
|
chr(ord("z") + 1),
|
|
]
|
|
for c in bads:
|
|
self.assertNotIn(c, charset)
|
|
|
|
def test_zsys_uuid(self):
|
|
for i in range(10):
|
|
uuid = gen_zsys_uuid()
|
|
self.assertEqual(6, len(uuid), uuid)
|