Skip to content

Commit

Permalink
Fix tests, improve validation
Browse files Browse the repository at this point in the history
  • Loading branch information
giacomognosis committed Apr 10, 2024
1 parent 659f43b commit 3014ca5
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 11 deletions.
20 changes: 10 additions & 10 deletions api/api/services/validator.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def __init__(self, request_data, validate_captcha, access_key=None, *args, **kwa
self.errors = []

def validate(self):
self.user_validation()
self.blocked_user_validation()
if len(self.errors) > 0:
return False

Expand Down Expand Up @@ -64,16 +64,16 @@ def validate(self):
return False
return True

def user_validation(self):
def blocked_user_validation(self):
recipient = self.request_data.get('recipient', None)
if not Web3.is_address(recipient):
self.errors.append(self.messages['INVALID_RECIPIENT'])

# check if recipient in blocked_users, return 403
user = BlockedUsers.get_by_address(recipient)
if user:
self.errors.append(self.messages['BLOCKED_RECIPIENT'])
self.http_return_code = 403
# Run validation on blocked users only if `recipient` is available.
# Let next validation steps do the rest.
if recipient:
# check if recipient in blocked_users, return 403
user = BlockedUsers.get_by_address(recipient)
if user:
self.errors.append(self.messages['BLOCKED_RECIPIENT'])
self.http_return_code = 403

def data_validation(self):
if self.request_data.get('chainId') != current_app.config['FAUCET_CHAIN_ID']:
Expand Down
2 changes: 1 addition & 1 deletion api/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def _mock(self, env_variables=None):
]
if env_variables:
self.patchers.append(mock.patch.dict(os.environ, env_variables))

for p in self.patchers:
p.start()

Expand Down

0 comments on commit 3014ca5

Please sign in to comment.