def generate(vo, issuer): for usage in get_global_account_usage( account=account, rse_expression=rse_expression, issuer=issuer, vo=vo): yield dumps(usage, cls=APIEncoder) + '\n'
def GET(self, account, rse_expression=None): """ Return the account usage of the account. HTTP Success: 200 OK HTTP Error: 401 Unauthorized 406 Not Acceptable 404 Not Found :param account: The account name. :param rse_expression: The RSE expression. """ header('Content-Type', 'application/x-json-stream') try: for usage in get_global_account_usage( account=account, rse_expression=rse_expression, issuer=ctx.env.get('issuer'), vo=ctx.env.get('vo')): yield dumps(usage, cls=APIEncoder) + '\n' except AccountNotFound as error: raise generate_http_error(404, 'AccountNotFound', error.args[0]) except RSENotFound as error: raise generate_http_error(404, 'RSENotFound', error.args[0]) except AccessDenied as error: raise generate_http_error(401, 'AccessDenied', error.args[0]) except Exception as error: print(format_exc()) raise InternalError(error)
def get(self, account, rse_expression=None): """ Return the global account usage of the account. .. :quickref: GlobalUsage; Get global account usage. :param account: The account name. :resheader Content-Type: application/x-json-stream :status 200: OK. :status 401: Invalid auth token. :status 404: Account not found. :status 404: RSE not found. :status 406: Not Acceptable. :status 500: Database exception. :returns: Line separated list of account usages. """ try: data = "" for usage in get_global_account_usage(account=account, rse_expression=rse_expression, issuer=request.environ.get('issuer'), vo=request.environ.get('vo')): data += dumps(usage, cls=APIEncoder) + '\n' return Response(data, content_type="application/x-json-stream") except AccountNotFound as error: return generate_http_error_flask(404, 'AccountNotFound', error.args[0]) except AccessDenied as error: return generate_http_error_flask(401, 'AccessDenied', error.args[0]) except RSENotFound as error: return generate_http_error_flask(404, 'RSENotFound', error.args[0]) except Exception as error: print(format_exc()) return error, 500
def test_api_account_limit(self): """ ACCOUNT_LIMIT (API): Test external representation of account limits """ # Add mock account limits rse_expr = '{}|{}'.format(self.rse_name, self.rse2_name) api_acc_lim.set_local_account_limit(self.account_name, self.rse_name, 10000, issuer='root', **self.vo) api_acc_lim.set_global_account_limit(self.account_name, rse_expr, 20000, issuer='root', **self.vo) out = api_acc_lim.get_local_account_limits(self.account_name, **self.vo) assert_in(self.rse_name, out) assert_not_in(self.rse_id, out) out = api_acc_lim.get_local_account_limit(self.account_name, self.rse_name, **self.vo) assert_in(self.rse_name, out) assert_not_in(self.rse_id, out) out = api_acc_lim.get_global_account_limits(self.account_name, **self.vo) assert_in(rse_expr, out) if self.multi_vo: assert_not_in('vo={}&({})'.format(self.vo['vo'], rse_expr), out) out = api_acc_lim.get_global_account_limit(self.account_name, rse_expr, **self.vo) assert_in(rse_expr, out) if self.multi_vo: assert_not_in('vo={}&({})'.format(self.vo['vo'], rse_expr), out) out = api_acc_lim.get_local_account_usage(self.account_name, self.rse_name, issuer='root', **self.vo) out = list(out) assert_not_equal(0, len(out)) assert_in(self.rse_id, [usage['rse_id'] for usage in out if 'rse_id' in usage]) for usage in out: if 'rse_id' in usage: assert_in('rse', usage) if usage['rse_id'] == self.rse_id: assert_equal(self.rse_name, usage["rse"]) out = api_acc_lim.get_global_account_usage(self.account_name, rse_expr, issuer='root', **self.vo) out = list(out) assert_not_equal(0, len(out)) assert_in(rse_expr, [usage['rse_expression'] for usage in out if 'rse_expression' in usage])
def test_api_account_limit(self): """ ACCOUNT_LIMIT (API): Test external representation of account limits """ # Add mock account limits rse_expr = '{}|{}'.format(self.rse_name, self.rse2_name) api_acc_lim.set_local_account_limit(self.account_name, self.rse_name, 10000, issuer='root', **self.vo) api_acc_lim.set_global_account_limit(self.account_name, rse_expr, 20000, issuer='root', **self.vo) out = api_acc_lim.get_local_account_limits(self.account_name, **self.vo) assert self.rse_name in out assert self.rse_id not in out out = api_acc_lim.get_local_account_limit(self.account_name, self.rse_name, **self.vo) assert self.rse_name in out assert self.rse_id not in out out = api_acc_lim.get_global_account_limits(self.account_name, **self.vo) assert rse_expr in out if self.multi_vo: assert 'vo={}&({})'.format(self.vo['vo'], rse_expr) not in out out = api_acc_lim.get_global_account_limit(self.account_name, rse_expr, **self.vo) assert rse_expr in out if self.multi_vo: assert 'vo={}&({})'.format(self.vo['vo'], rse_expr) not in out out = api_acc_lim.get_local_account_usage(self.account_name, self.rse_name, issuer='root', **self.vo) out = list(out) assert 0 != len(out) assert self.rse_id in [usage['rse_id'] for usage in out if 'rse_id' in usage] for usage in out: if 'rse_id' in usage: assert 'rse' in usage if usage['rse_id'] == self.rse_id: assert self.rse_name == usage["rse"] out = api_acc_lim.get_global_account_usage(self.account_name, rse_expr, issuer='root', **self.vo) out = list(out) assert 0 != len(out) assert rse_expr in [usage['rse_expression'] for usage in out if 'rse_expression' in usage]