def test_list_user(self):
name1 = self.get_user_name()
rc = nimbus_new_user.main([name1])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
name2 = self.get_user_name()
rc = nimbus_new_user.main([name2])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
name3 = self.get_user_name()
rc = nimbus_new_user.main([name3])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
rc = nimbus_list_users.main(["-b", "-r", "display_name", "-O", outFileName, '%'])
rc = self.find_in_file(outFileName, name1)
self.assertTrue(rc)
rc = self.find_in_file(outFileName, name2)
self.assertTrue(rc)
rc = self.find_in_file(outFileName, name3)
self.assertTrue(rc)
os.unlink(outFileName)
def test_list_user(self):
name1 = self.get_user_name()
rc = nimbus_new_user.main([name1])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
name2 = self.get_user_name()
rc = nimbus_new_user.main([name2])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
name3 = self.get_user_name()
rc = nimbus_new_user.main([name3])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
rc = nimbus_list_users.main(["-b", "-r", "display_name", "-O", outFileName, '%'])
rc = self.find_in_file(outFileName, name1)
self.assertTrue(rc)
rc = self.find_in_file(outFileName, name2)
self.assertTrue(rc)
rc = self.find_in_file(outFileName, name3)
self.assertTrue(rc)
os.unlink(outFileName)
def test_new_user(self):
friendly_name = self.get_user_name()
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
rc = nimbus_new_user.main([friendly_name])
self.assertNotEqual(rc, 0, "should not be 0 %d" % (rc))
# make sure the user was not added
os.environ["NIMBUS_HOME"] = self.nh
rc = nimbus_list_users.main(["-b", "-r", "display_name", "-O", outFileName, friendly_name])
rc = self.find_in_file(outFileName, friendly_name)
self.assertFalse(rc)
def test_make_remove_canid_user(self):
uu = str(uuid.uuid1())
friendly_name = self.get_user_name()
rc = nimbus_new_user.main(["--canonical-id", uu, friendly_name])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
rc = nimbus_list_users.main(["-b", "-r", "canonical_id", "-O", outFileName, "%"])
rc = self.find_in_file(outFileName, uu)
self.assertTrue(rc)
rc = nimbus_remove_user.main([friendly_name])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
def main(argv=sys.argv[1:]):
try:
(opts, args, p) = setup_options(argv)
file_users = read_users(args, delimiter=opts.delim)
file,path = tempfile.mkstemp()
pynimbusauthz.print_msg(opts, 2, "Using temp file: " + path)
try:
nimbus_list_users.main(['-b', '-D', opts.delim, '-r', _fields_csv,
'-O', path, '%'])
current_users = read_users(path, delimiter=opts.delim)
finally:
os.remove(path)
if not walk_users(current_users, file_users, opts):
return 1
except CLIError, clie:
print clie
return clie.get_rc()
def main(argv=sys.argv[1:]):
try:
(opts, args, p) = setup_options(argv)
file_users = read_users(args, delimiter=opts.delim)
file, path = tempfile.mkstemp()
pynimbusauthz.print_msg(opts, 2, "Using temp file: " + path)
try:
nimbus_list_users.main(
['-b', '-D', opts.delim, '-r', _fields_csv, '-O', path, '%'])
current_users = read_users(path, delimiter=opts.delim)
finally:
os.remove(path)
if not walk_users(current_users, file_users, opts):
return 1
except CLIError, clie:
print clie
return clie.get_rc()
def _get_users(self):
"""Returns a dict of existing users keyed by display_name.
Each entry is a dict of user properties.
"""
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
try:
os.close(tmpFD)
rc = nimbus_list_users.main(["-b", "-O", outFileName, "%"])
self.assertEqual(rc, 0)
# cheating here
return nimbus_import_users.read_users(outFileName)
finally:
os.unlink(outFileName)
def test_new_user(self):
friendly_name = self.get_user_name()
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
rc = nimbus_new_user.main([friendly_name])
self.assertNotEqual(rc, 0, "should not be 0 %d" % (rc))
# make sure the user was not added
os.environ['NIMBUS_HOME'] = self.nh
rc = nimbus_list_users.main(
["-b", "-r", "display_name", "-O", outFileName, friendly_name])
rc = self.find_in_file(outFileName, friendly_name)
self.assertFalse(rc)
def _get_users(self):
"""Returns a dict of existing users keyed by display_name.
Each entry is a dict of user properties.
"""
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
try:
os.close(tmpFD)
rc = nimbus_list_users.main(["-b", "-O", outFileName, '%'])
self.assertEqual(rc, 0)
#cheating here
return nimbus_import_users.read_users(outFileName)
finally:
os.unlink(outFileName)
def test_make_remove_canid_user(self):
uu = str(uuid.uuid1())
friendly_name = self.get_user_name()
rc = nimbus_new_user.main(["--canonical-id", uu, friendly_name])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
rc = nimbus_list_users.main(["-b", "-r", "canonical_id", "-O", outFileName, '%'])
rc = self.find_in_file(outFileName, uu)
self.assertTrue(rc)
rc = nimbus_remove_user.main([friendly_name])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
def test_remove_user(self):
friendly_name = self.get_user_name()
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
# add a good user
os.environ["NIMBUS_HOME"] = self.nh
rc = nimbus_new_user.main([friendly_name])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
os.environ["NIMBUS_HOME"] = "/nope"
# remove with an error
rc = nimbus_remove_user.main([friendly_name])
self.assertNotEqual(rc, 0, "should not be 0 %d" % (rc))
# relist to see user is still there
os.environ["NIMBUS_HOME"] = self.nh
rc = nimbus_list_users.main(["-b", "-r", "display_name", "-O", outFileName, friendly_name])
self.assertEqual(rc, 0, "should not be 0 %d" % (rc))
rc = self.find_in_file(outFileName, friendly_name)
self.assertTrue(rc)
def test_remove_user(self):
friendly_name = self.get_user_name()
(tmpFD, outFileName) = tempfile.mkstemp("cumulustests")
os.close(tmpFD)
# add a good user
os.environ['NIMBUS_HOME'] = self.nh
rc = nimbus_new_user.main([friendly_name])
self.assertEqual(rc, 0, "should be 0 %d" % (rc))
os.environ['NIMBUS_HOME'] = "/nope"
# remove with an error
rc = nimbus_remove_user.main([friendly_name])
self.assertNotEqual(rc, 0, "should not be 0 %d" % (rc))
# relist to see user is still there
os.environ['NIMBUS_HOME'] = self.nh
rc = nimbus_list_users.main(
["-b", "-r", "display_name", "-O", outFileName, friendly_name])
self.assertEqual(rc, 0, "should not be 0 %d" % (rc))
rc = self.find_in_file(outFileName, friendly_name)
self.assertTrue(rc)
