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)