def test_update_rql_driver_error(self):
with self.app.test_request_context():
with patch.object(Interactions, 'update',
side_effect=RqlDriverError(None)):
update(DEFAULT_REGISTRATIONS_TABLE,
record_id='123',
username=None,
updates={})
self.assertRaises(RqlDriverError)
def test_update_no_record_id(self):
with self.app.test_request_context():
with patch.object(Interactions, 'update', return_value={}):
response = update(DEFAULT_REGISTRATIONS_TABLE,
record_id=None,
username='******',
updates={})
self.assertEqual(response.status_code, client.OK)
def test_update_rql_runtime_error(self):
with self.app.test_request_context():
with patch.object(Interactions, 'update',
side_effect=RqlRuntimeError(None, None, None)):
response = update(DEFAULT_REGISTRATIONS_TABLE,
record_id='123', username=None, updates={})
self.assertRaises(RqlRuntimeError)
self.assertEqual(response.status_code,
client.INTERNAL_SERVER_ERROR)
def test_update_bad_request(self, connection_method):
connection_method.return_value = Mock(__enter__=Mock, __exit__=Mock())
with self.app.test_request_context():
with patch.object(rethink, 'table', return_value=Mock()) as \
table_method:
response = update(DEFAULT_REGISTRATIONS_TABLE,
record_id='123',
username=None,
updates={})
self.assertEqual(response.status_code, client.BAD_REQUEST)
table_method.assert_called_once_with(
DEFAULT_REGISTRATIONS_TABLE
)
def patch(self):
"""
Updates account. Only one field can be updated: endpoint
Updating the endpoint also resets the failed_count
"""
json_data = request.get_json()
username = request.headers['username']
if 'endpoint' in json_data:
update_json = {
'endpoint': json_data['endpoint'],
'failed_count': 0
}
else:
return make_response(jsonify(
{'Error': 'Missing endpoint field'}), client.BAD_REQUEST)
return update(DEFAULT_ACCOUNTS_TABLE, username=username,
updates=update_json)
def patch(self, registration_id):
"""
Updates registration. Only one field can be updated: description
"""
return_val = validate_access(request.headers['username'],
registration_id=registration_id)
if return_val:
return return_val
json_data = request.get_json()
update_json = {}
if 'description' in json_data:
update_json['description'] = json_data['description']
else:
return make_response(
jsonify({'Error': 'Description field missing'}),
client.BAD_REQUEST)
return update(DEFAULT_REGISTRATIONS_TABLE,
record_id=registration_id,
updates=update_json)
def patch(self, registration_id):
"""
Updates registration. Only one field can be updated: description
"""
return_val = validate_access(
request.headers['username'],
registration_id=registration_id)
if return_val:
return return_val
json_data = request.get_json()
update_json = {}
if 'description' in json_data:
update_json['description'] = json_data['description']
else:
return make_response(
jsonify({'Error': 'Description field missing'}),
client.BAD_REQUEST)
return update(DEFAULT_REGISTRATIONS_TABLE,
record_id=registration_id,
updates=update_json)
def test_update_bad_request(self):
with self.app.test_request_context():
with patch.object(Interactions, "update", return_value=None):
response = update(DEFAULT_REGISTRATIONS_TABLE, record_id="123", username=None, updates={})
self.assertEqual(response.status_code, client.BAD_REQUEST)
