def test_create_user_with_valid_username_and_password_parameters(self): code = scripts.create_user({"registry": self.registry}, "username", "password") assert self.registry.storage.update.call_count == 1 self.registry.permission.add_principal_to_ace.assert_called_with( "/accounts/username", "write", "account:username" ) assert code == 0
def test_create_user_in_read_only_displays_an_error(self): with mock.patch('kinto.plugins.accounts.scripts.logger') as mocked: self.registry.settings['readonly'] = 'true' code = scripts.create_user({'registry': self.registry}) assert code == 51 mocked.error.assert_called_once_with('Cannot create a user with ' 'a readonly server.')
def test_create_user_when_not_included_displays_an_error(self): with mock.patch('kinto.plugins.accounts.scripts.logger') as mocked: self.registry.settings['includes'] = '' code = scripts.create_user({'registry': self.registry}) assert code == 52 mocked.error.assert_called_once_with('Cannot create a user when the accounts ' 'plugin is not installed.')
def test_create_user_when_not_included_displays_an_error(self): with mock.patch("kinto.plugins.accounts.scripts.logger") as mocked: self.registry.settings["includes"] = "" code = scripts.create_user({"registry": self.registry}) assert code == 52 mocked.error.assert_called_once_with( "Cannot create a user when the accounts " "plugin is not installed." )
def test_create_user_with_a_valid_username_and_password_confirmation(self): with mock.patch('kinto.plugins.accounts.scripts.input', return_value="username"): with mock.patch('kinto.plugins.accounts.scripts.getpass.getpass', return_value="password"): code = scripts.create_user({'registry': self.registry}) self.registry.storage.update.assert_called_once() self.registry.permission.add_principal_to_ace.assert_called_with( "/accounts/username", "write", "account:username") assert code == 0
def test_create_user_with_an_invalid_username_and_password_confirmation_recovers(self): with mock.patch('kinto.plugins.accounts.scripts.input', side_effect=["&zert", "username"]): with mock.patch('kinto.plugins.accounts.scripts.getpass.getpass', side_effect=[ "password", "p4ssw0rd", "password", "password"]): code = scripts.create_user({'registry': self.registry}, None, None) self.registry.storage.update.assert_called_once() self.registry.permission.add_principal_to_ace.assert_called_with( "/accounts/username", "write", "account:username") assert code == 0
def test_create_user_with_an_invalid_username_and_password_confirmation_recovers(self): with mock.patch("kinto.plugins.accounts.scripts.input", side_effect=["&zert", "username"]): with mock.patch( "kinto.plugins.accounts.scripts.getpass.getpass", side_effect=["password", "p4ssw0rd", "password", "password"], ): code = scripts.create_user({"registry": self.registry}, None, None) assert self.registry.storage.update.call_count == 1 self.registry.permission.add_principal_to_ace.assert_called_with( "/accounts/username", "write", "account:username" ) assert code == 0
def test_create_user_aborted_by_eof(self): with mock.patch('kinto.plugins.accounts.scripts.input', side_effect=EOFError): code = scripts.create_user({'registry': self.registry}) assert code == 53
def test_create_user_with_valid_username_and_password_parameters(self): code = scripts.create_user({'registry': self.registry}, "username", "password") self.registry.storage.create.assert_called_once() self.registry.permission.add_principal_to_ace.assert_called_with( "/accounts/username", "write", "account:username") assert code == 0
def setUp(self): # Create alice user create_user({"registry": self.app.app.registry}, "alice", "123456")
def test_create_user_in_read_only_displays_an_error(self): with mock.patch("kinto.plugins.accounts.scripts.logger") as mocked: self.registry.settings["readonly"] = "true" code = scripts.create_user({"registry": self.registry}) assert code == 51 mocked.error.assert_called_once_with("Cannot create a user with " "a readonly server.")
def main(args=None): """The main routine.""" if args is None: args = sys.argv[1:] parser = argparse.ArgumentParser(description='Kinto Command-Line ' 'Interface') commands = ('init', 'start', 'migrate', 'delete-collection', 'version', 'rebuild-quotas', 'create-user') subparsers = parser.add_subparsers(title='subcommands', description='Main Kinto CLI commands', dest='subcommand', help='Choose and run with --help') subparsers.required = True for command in commands: subparser = subparsers.add_parser(command) subparser.set_defaults(which=command) subparser.add_argument('--ini', help='Application configuration file', dest='ini_file', required=False, default=DEFAULT_CONFIG_FILE) subparser.add_argument('-q', '--quiet', action='store_const', const=logging.CRITICAL, dest='verbosity', help='Show only critical errors.') subparser.add_argument( '-v', '--debug', action='store_const', const=logging.DEBUG, dest='verbosity', help='Show all messages, including debug messages.') if command == 'init': subparser.add_argument('--backend', help='{memory,redis,postgresql}', dest='backend', required=False, default=None) subparser.add_argument('--cache-backend', help='{memory,redis,postgresql,memcached}', dest='cache-backend', required=False, default=None) subparser.add_argument('--host', help='Host to listen() on.', dest='host', required=False, default='127.0.0.1') elif command == 'migrate': subparser.add_argument('--dry-run', action='store_true', help='Simulate the migration operations ' 'and show information', dest='dry_run', required=False, default=False) elif command == 'delete-collection': subparser.add_argument('--bucket', help='The bucket where the collection ' 'belongs to.', required=True) subparser.add_argument('--collection', help='The collection to remove.', required=True) elif command == 'rebuild-quotas': subparser.add_argument('--dry-run', action='store_true', help='Simulate the rebuild operation ' 'and show information', dest='dry_run', required=False, default=False) elif command == 'start': subparser.add_argument('--reload', action='store_true', help='Restart when code or config changes', required=False, default=False) subparser.add_argument('--port', type=int, help='Listening port number', required=False, default=DEFAULT_PORT) elif command == 'create-user': subparser.add_argument('-u', '--username', help='Superuser username', required=False, default=None) subparser.add_argument('-p', '--password', help='Superuser password', required=False, default=None) # Parse command-line arguments parsed_args = vars(parser.parse_args(args)) config_file = parsed_args['ini_file'] which_command = parsed_args['which'] # Initialize logging from level = parsed_args.get('verbosity') or DEFAULT_LOG_LEVEL logging.basicConfig(level=level, format=DEFAULT_LOG_FORMAT) if which_command == 'init': if os.path.exists(config_file): print('{} already exists.'.format(config_file), file=sys.stderr) return 1 backend = parsed_args['backend'] cache_backend = parsed_args['cache-backend'] if not backend: while True: prompt = ('Select the backend you would like to use: ' '(1 - postgresql, 2 - redis, default - memory) ') answer = input(prompt).strip() try: backends = {'1': 'postgresql', '2': 'redis', '': 'memory'} backend = backends[answer] break except KeyError: pass if not cache_backend: while True: prompt = ( 'Select the cache backend you would like to use: ' '(1 - postgresql, 2 - redis, 3 - memcached, default - memory) ' ) answer = input(prompt).strip() try: cache_backends = { '1': 'postgresql', '2': 'redis', '3': 'memcached', '': 'memory' } cache_backend = cache_backends[answer] break except KeyError: pass init(config_file, backend, cache_backend, parsed_args['host']) # Install postgresql libraries if necessary if backend == 'postgresql' or cache_backend == 'postgresql': try: import psycopg2 # NOQA except ImportError: subprocess.check_call([ sys.executable, '-m', 'pip', 'install', 'kinto[postgresql]' ]) elif backend == 'redis' or cache_backend == 'redis': try: import kinto_redis # NOQA except ImportError: subprocess.check_call( [sys.executable, '-m', 'pip', 'install', 'kinto[redis]']) elif cache_backend == 'memcached': try: import memcache # NOQA except ImportError: subprocess.check_call([ sys.executable, '-m', 'pip', 'install', 'kinto[memcached]' ]) elif which_command == 'migrate': dry_run = parsed_args['dry_run'] env = bootstrap(config_file, options={'command': 'migrate'}) scripts.migrate(env, dry_run=dry_run) elif which_command == 'delete-collection': env = bootstrap(config_file, options={'command': 'delete-collection'}) return scripts.delete_collection(env, parsed_args['bucket'], parsed_args['collection']) elif which_command == 'rebuild-quotas': dry_run = parsed_args['dry_run'] env = bootstrap(config_file, options={'command': 'rebuild-quotas'}) return scripts.rebuild_quotas(env, dry_run=dry_run) elif which_command == 'create-user': username = parsed_args['username'] password = parsed_args['password'] env = bootstrap(config_file, options={'command': 'create-user'}) return create_user(env, username=username, password=password) elif which_command == 'start': pserve_argv = ['pserve'] if parsed_args['reload']: pserve_argv.append('--reload') if level == logging.DEBUG: pserve_argv.append('-v') if level == logging.CRITICAL: pserve_argv.append('-q') pserve_argv.append(config_file) pserve_argv.append('http_port={}'.format(parsed_args['port'])) pserve.main(argv=pserve_argv) else: print(__version__) return 0
def main(args=None): """The main routine.""" if args is None: args = sys.argv[1:] parser = argparse.ArgumentParser(description='Kinto Command-Line ' 'Interface') commands = ('init', 'start', 'migrate', 'delete-collection', 'version', 'rebuild-quotas', 'create-user') subparsers = parser.add_subparsers(title='subcommands', description='Main Kinto CLI commands', dest='subcommand', help='Choose and run with --help') subparsers.required = True for command in commands: subparser = subparsers.add_parser(command) subparser.set_defaults(which=command) subparser.add_argument('--ini', help='Application configuration file', dest='ini_file', required=False, default=DEFAULT_CONFIG_FILE) subparser.add_argument('-q', '--quiet', action='store_const', const=logging.CRITICAL, dest='verbosity', help='Show only critical errors.') subparser.add_argument('-v', '--debug', action='store_const', const=logging.DEBUG, dest='verbosity', help='Show all messages, including debug messages.') if command == 'init': subparser.add_argument('--backend', help='{memory,redis,postgresql}', dest='backend', required=False, default=None) subparser.add_argument('--host', help='Host to listen() on.', dest='host', required=False, default='127.0.0.1') elif command == 'migrate': subparser.add_argument('--dry-run', action='store_true', help='Simulate the migration operations ' 'and show information', dest='dry_run', required=False, default=False) elif command == 'delete-collection': subparser.add_argument('--bucket', help='The bucket where the collection ' 'belongs to.', required=True) subparser.add_argument('--collection', help='The collection to remove.', required=True) elif command == 'rebuild-quotas': subparser.add_argument('--dry-run', action='store_true', help='Simulate the rebuild operation ' 'and show information', dest='dry_run', required=False, default=False) elif command == 'start': subparser.add_argument('--reload', action='store_true', help='Restart when code or config changes', required=False, default=False) subparser.add_argument('--port', type=int, help='Listening port number', required=False, default=DEFAULT_PORT) elif command == 'create-user': subparser.add_argument('-u', '--username', help='Superuser username', required=False, default=None) subparser.add_argument('-p', '--password', help='Superuser password', required=False, default=None) # Parse command-line arguments parsed_args = vars(parser.parse_args(args)) config_file = parsed_args['ini_file'] which_command = parsed_args['which'] # Initialize logging from level = parsed_args.get('verbosity') or DEFAULT_LOG_LEVEL logging.basicConfig(level=level, format=DEFAULT_LOG_FORMAT) if which_command == 'init': if os.path.exists(config_file): print('{} already exists.'.format(config_file), file=sys.stderr) return 1 backend = parsed_args['backend'] if not backend: while True: prompt = ('Select the backend you would like to use: ' '(1 - postgresql, 2 - redis, default - memory) ') answer = input(prompt).strip() try: backends = {'1': 'postgresql', '2': 'redis', '': 'memory'} backend = backends[answer] break except KeyError: pass init(config_file, backend, parsed_args['host']) # Install postgresql libraries if necessary if backend == 'postgresql': try: import psycopg2 # NOQA except ImportError: subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'kinto[postgresql]']) elif backend == 'redis': try: import kinto_redis # NOQA except ImportError: subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'kinto[redis]']) elif which_command == 'migrate': dry_run = parsed_args['dry_run'] env = bootstrap(config_file) scripts.migrate(env, dry_run=dry_run) elif which_command == 'delete-collection': env = bootstrap(config_file) return scripts.delete_collection(env, parsed_args['bucket'], parsed_args['collection']) elif which_command == 'rebuild-quotas': dry_run = parsed_args['dry_run'] env = bootstrap(config_file) return scripts.rebuild_quotas(env, dry_run=dry_run) elif which_command == 'create-user': username = parsed_args['username'] password = parsed_args['password'] env = bootstrap(config_file) return create_user(env, username=username, password=password) elif which_command == 'start': pserve_argv = ['pserve'] if parsed_args['reload']: pserve_argv.append('--reload') if level == logging.DEBUG: pserve_argv.append('-v') if level == logging.CRITICAL: pserve_argv.append('-q') pserve_argv.append(config_file) pserve_argv.append('http_port={}'.format(parsed_args['port'])) pserve.main(argv=pserve_argv) else: print(__version__) return 0
def test_create_user_aborted_by_eof(self): with mock.patch("kinto.plugins.accounts.scripts.input", side_effect=EOFError): code = scripts.create_user({"registry": self.registry}) assert code == 53
def test_create_user_in_read_only_displays_an_error(self): with mock.patch("kinto.plugins.accounts.scripts.logger") as mocked: self.registry.settings["readonly"] = "true" code = scripts.create_user({"registry": self.registry}) assert code == 51 mocked.error.assert_called_once_with("Cannot create a user with " "a readonly server.")
def main(args=None): """The main routine.""" if args is None: args = sys.argv[1:] parser = argparse.ArgumentParser( description="Kinto Command-Line Interface") commands = ( "init", "start", "migrate", "flush-cache", "version", "rebuild-quotas", "create-user", ) subparsers = parser.add_subparsers( title="subcommands", description="Main Kinto CLI commands", dest="subcommand", help="Choose and run with --help", ) subparsers.required = True for command in commands: subparser = subparsers.add_parser(command) subparser.set_defaults(which=command) subparser.add_argument( "--ini", help="Application configuration file", dest="ini_file", required=False, default=DEFAULT_CONFIG_FILE, ) subparser.add_argument( "-q", "--quiet", action="store_const", const=logging.CRITICAL, dest="verbosity", help="Show only critical errors.", ) subparser.add_argument( "-v", "--debug", action="store_const", const=logging.DEBUG, dest="verbosity", help="Show all messages, including debug messages.", ) if command == "init": subparser.add_argument( "--backend", help="{memory,redis,postgresql}", dest="backend", required=False, default=None, ) subparser.add_argument( "--cache-backend", help="{memory,redis,postgresql,memcached}", dest="cache-backend", required=False, default=None, ) subparser.add_argument( "--host", help="Host to listen() on.", dest="host", required=False, default="127.0.0.1", ) elif command == "migrate": subparser.add_argument( "--dry-run", action="store_true", help="Simulate the migration operations and show information", dest="dry_run", required=False, default=False, ) elif command == "rebuild-quotas": subparser.add_argument( "--dry-run", action="store_true", help="Simulate the rebuild operation and show information", dest="dry_run", required=False, default=False, ) elif command == "start": subparser.add_argument( "--reload", action="store_true", help="Restart when code or config changes", required=False, default=False, ) subparser.add_argument( "--port", type=int, help="Listening port number", required=False, default=DEFAULT_PORT, ) elif command == "create-user": subparser.add_argument("-u", "--username", help="Superuser username", required=False, default=None) subparser.add_argument("-p", "--password", help="Superuser password", required=False, default=None) # Parse command-line arguments parsed_args = vars(parser.parse_args(args)) config_file = parsed_args["ini_file"] which_command = parsed_args["which"] # Initialize logging from level = parsed_args.get("verbosity") or DEFAULT_LOG_LEVEL logging.basicConfig(level=level, format=DEFAULT_LOG_FORMAT) if which_command == "init": if os.path.exists(config_file): print(f"{config_file} already exists.", file=sys.stderr) return 1 backend = parsed_args["backend"] cache_backend = parsed_args["cache-backend"] if not backend: while True: prompt = ("Select the backend you would like to use: " "(1 - postgresql, 2 - redis, default - memory) ") answer = input(prompt).strip() try: backends = {"1": "postgresql", "2": "redis", "": "memory"} backend = backends[answer] break except KeyError: pass if not cache_backend: while True: prompt = ( "Select the cache backend you would like to use: " "(1 - postgresql, 2 - redis, 3 - memcached, default - memory) " ) answer = input(prompt).strip() try: cache_backends = { "1": "postgresql", "2": "redis", "3": "memcached", "": "memory", } cache_backend = cache_backends[answer] break except KeyError: pass init(config_file, backend, cache_backend, parsed_args["host"]) # Install postgresql libraries if necessary if backend == "postgresql" or cache_backend == "postgresql": try: import psycopg2 # NOQA except ImportError: subprocess.check_call([ sys.executable, "-m", "pip", "install", "kinto[postgresql]" ]) elif backend == "redis" or cache_backend == "redis": try: import kinto_redis # NOQA except ImportError: subprocess.check_call( [sys.executable, "-m", "pip", "install", "kinto[redis]"]) elif cache_backend == "memcached": try: import memcache # NOQA except ImportError: subprocess.check_call([ sys.executable, "-m", "pip", "install", "kinto[memcached]" ]) elif which_command == "migrate": dry_run = parsed_args["dry_run"] env = bootstrap(config_file, options={"command": "migrate"}) core_scripts.migrate(env, dry_run=dry_run) elif which_command == "flush-cache": env = bootstrap(config_file, options={"command": "flush-cache"}) core_scripts.flush_cache(env) elif which_command == "rebuild-quotas": dry_run = parsed_args["dry_run"] env = bootstrap(config_file, options={"command": "rebuild-quotas"}) return kinto_scripts.rebuild_quotas(env, dry_run=dry_run) elif which_command == "create-user": username = parsed_args["username"] password = parsed_args["password"] env = bootstrap(config_file, options={"command": "create-user"}) return accounts_scripts.create_user(env, username=username, password=password) elif which_command == "start": pserve_argv = ["pserve"] if parsed_args["reload"]: pserve_argv.append("--reload") if level == logging.DEBUG: pserve_argv.append("-v") if level == logging.CRITICAL: pserve_argv.append("-q") pserve_argv.append(config_file) pserve_argv.append(f"http_port={parsed_args['port']}") pserve.main(argv=pserve_argv) else: print(__version__) return 0
def main(args=None): """The main routine.""" if args is None: args = sys.argv[1:] parser = argparse.ArgumentParser(description="Kinto Command-Line " "Interface") commands = ( "init", "start", "migrate", "delete-collection", "version", "rebuild-quotas", "create-user", ) subparsers = parser.add_subparsers( title="subcommands", description="Main Kinto CLI commands", dest="subcommand", help="Choose and run with --help", ) subparsers.required = True for command in commands: subparser = subparsers.add_parser(command) subparser.set_defaults(which=command) subparser.add_argument( "--ini", help="Application configuration file", dest="ini_file", required=False, default=DEFAULT_CONFIG_FILE, ) subparser.add_argument( "-q", "--quiet", action="store_const", const=logging.CRITICAL, dest="verbosity", help="Show only critical errors.", ) subparser.add_argument( "-v", "--debug", action="store_const", const=logging.DEBUG, dest="verbosity", help="Show all messages, including debug messages.", ) if command == "init": subparser.add_argument( "--backend", help="{memory,redis,postgresql}", dest="backend", required=False, default=None, ) subparser.add_argument( "--cache-backend", help="{memory,redis,postgresql,memcached}", dest="cache-backend", required=False, default=None, ) subparser.add_argument( "--host", help="Host to listen() on.", dest="host", required=False, default="127.0.0.1", ) elif command == "migrate": subparser.add_argument( "--dry-run", action="store_true", help="Simulate the migration operations " "and show information", dest="dry_run", required=False, default=False, ) elif command == "delete-collection": subparser.add_argument( "--bucket", help="The bucket where the collection " "belongs to.", required=True ) subparser.add_argument("--collection", help="The collection to remove.", required=True) elif command == "rebuild-quotas": subparser.add_argument( "--dry-run", action="store_true", help="Simulate the rebuild operation " "and show information", dest="dry_run", required=False, default=False, ) elif command == "start": subparser.add_argument( "--reload", action="store_true", help="Restart when code or config changes", required=False, default=False, ) subparser.add_argument( "--port", type=int, help="Listening port number", required=False, default=DEFAULT_PORT, ) elif command == "create-user": subparser.add_argument( "-u", "--username", help="Superuser username", required=False, default=None ) subparser.add_argument( "-p", "--password", help="Superuser password", required=False, default=None ) # Parse command-line arguments parsed_args = vars(parser.parse_args(args)) config_file = parsed_args["ini_file"] which_command = parsed_args["which"] # Initialize logging from level = parsed_args.get("verbosity") or DEFAULT_LOG_LEVEL logging.basicConfig(level=level, format=DEFAULT_LOG_FORMAT) if which_command == "init": if os.path.exists(config_file): print("{} already exists.".format(config_file), file=sys.stderr) return 1 backend = parsed_args["backend"] cache_backend = parsed_args["cache-backend"] if not backend: while True: prompt = ( "Select the backend you would like to use: " "(1 - postgresql, 2 - redis, default - memory) " ) answer = input(prompt).strip() try: backends = {"1": "postgresql", "2": "redis", "": "memory"} backend = backends[answer] break except KeyError: pass if not cache_backend: while True: prompt = ( "Select the cache backend you would like to use: " "(1 - postgresql, 2 - redis, 3 - memcached, default - memory) " ) answer = input(prompt).strip() try: cache_backends = { "1": "postgresql", "2": "redis", "3": "memcached", "": "memory", } cache_backend = cache_backends[answer] break except KeyError: pass init(config_file, backend, cache_backend, parsed_args["host"]) # Install postgresql libraries if necessary if backend == "postgresql" or cache_backend == "postgresql": try: import psycopg2 # NOQA except ImportError: subprocess.check_call( [sys.executable, "-m", "pip", "install", "kinto[postgresql]"] ) elif backend == "redis" or cache_backend == "redis": try: import kinto_redis # NOQA except ImportError: subprocess.check_call([sys.executable, "-m", "pip", "install", "kinto[redis]"]) elif cache_backend == "memcached": try: import memcache # NOQA except ImportError: subprocess.check_call([sys.executable, "-m", "pip", "install", "kinto[memcached]"]) elif which_command == "migrate": dry_run = parsed_args["dry_run"] env = bootstrap(config_file, options={"command": "migrate"}) scripts.migrate(env, dry_run=dry_run) elif which_command == "delete-collection": env = bootstrap(config_file, options={"command": "delete-collection"}) return scripts.delete_collection(env, parsed_args["bucket"], parsed_args["collection"]) elif which_command == "rebuild-quotas": dry_run = parsed_args["dry_run"] env = bootstrap(config_file, options={"command": "rebuild-quotas"}) return scripts.rebuild_quotas(env, dry_run=dry_run) elif which_command == "create-user": username = parsed_args["username"] password = parsed_args["password"] env = bootstrap(config_file, options={"command": "create-user"}) return create_user(env, username=username, password=password) elif which_command == "start": pserve_argv = ["pserve"] if parsed_args["reload"]: pserve_argv.append("--reload") if level == logging.DEBUG: pserve_argv.append("-v") if level == logging.CRITICAL: pserve_argv.append("-q") pserve_argv.append(config_file) pserve_argv.append("http_port={}".format(parsed_args["port"])) pserve.main(argv=pserve_argv) else: print(__version__) return 0