Skip to content

Commit

Permalink
Add check_group tests for module_generator unit
Browse files Browse the repository at this point in the history
  • Loading branch information
xdelaruelle committed Dec 27, 2023
1 parent 6cf9474 commit 6e674b2
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions test/framework/module_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -1598,6 +1598,31 @@ def test_generated_module_file_swap(self):
# one/1.0 module was swapped for one/1.1
self.assertEqual(loaded_mods[-2]['mod_name'], 'one/1.1')

def test_check_group(self):
"""Test check_group method."""
if self.MODULE_GENERATOR_CLASS == ModuleGeneratorTcl:
if self.modtool.supports_tcl_check_group:
expected = '\n'.join([
"if { ![ module-info usergroups group_name ] } {",
" error \"mesg\"",
"}",
'',
])
self.assertEqual(expected, self.modgen.check_group("group_name", error_msg="mesg"))
else:
self.assertEqual('', self.modgen.check_group("group_name", error_msg="mesg"))
else:
if self.modtool.supports_lua_check_group:
expected = '\n'.join([
'if not ( userInGroup("group_name") ) then',
' LmodError("mesg")',
'end',
'',
])
self.assertEqual(expected, self.modgen.check_group("group_name", error_msg="mesg"))
else:
self.assertEqual('', self.modgen.check_group("group_name", error_msg="mesg"))


class TclModuleGeneratorTest(ModuleGeneratorTest):
"""Test for module_generator module for Tcl syntax."""
Expand Down

0 comments on commit 6e674b2

Please sign in to comment.