def test_get_webhook(client, test_application_token, test_application):
webhook_normal = Webhook.create('http://www.foo.bar')
headers = {'Authorization': test_application_token}
query_string = {'application_name': test_application.application_name}
r = client.get('/api/webhook/%s' % webhook_normal.id, headers=headers,
query_string=query_string)
assert_response_ok(r)
assert r.json['data'] == {
'webhook_id': webhook_normal.id,
'webhook_url': webhook_normal.url,
'webhook_type': webhook_normal.hook_type,
'event_list': []
}
webhook_universal = Webhook.create('http://www.foo.bar', 1)
headers = {'Authorization': test_application_token}
r = client.get('/api/webhook/%s' % webhook_universal.id, headers=headers,
query_string=query_string)
assert_response_ok(r)
assert r.json['data'] == {
'webhook_id': webhook_universal.id,
'webhook_url': webhook_universal.url,
'webhook_type': webhook_universal.hook_type,
'event_list': []
}
def test_create_webhook(db):
instance = Webhook.create('http://www.foo.bar')
assert instance.id > 0
assert instance.url == 'http://www.foo.bar'
assert instance.hook_type == Webhook.TYPE_NORMAL
assert instance.is_normal
instance = Webhook.create('http://foo.foo.bar', Webhook.TYPE_UNIVERSAL)
assert instance.id > 0
assert instance.url == 'http://foo.foo.bar'
assert instance.hook_type == Webhook.TYPE_UNIVERSAL
assert instance.id in Webhook.get_all_ids()
assert instance.id in Webhook.get_ids_by_type(instance.hook_type)
def test_delete_webhook(db):
instance = Webhook.create('http://www.foo.bar')
assert instance.id > 0
instance.delete()
assert Webhook.get(instance.id) is None
assert instance.id not in Webhook.get_all_ids()
assert instance.id not in Webhook.get_ids_by_type(instance.hook_type)
def post(self):
"""Create a new webhook.
The request accepting a JSON body, the schema likes::
{
"webhook_url": "http://www.example.com",
"event_list": [
"CREATE_CONFIG_CLUSTER",
"DELETE_CONFIG_CLUSTER"
]
}
The content of ``event_list`` should be a list of action that
already defined in Huskar.
The ``application_name`` is only required when the ``webhook_type``
is 0, it means the webhook want to subscribe some events of specified
application. If the ``webhook_type`` value specified with 1,
a universal webhook will be registered which will receive all
the events of Huskar site, so the ``event_list`` will be ignored
because that is unnecessary.
:param webhook_type: default 0, set ``site`` level with 1.
:param application_name: The name of application, optional.
:form webhook_url: the webhook url.
:form event_list: event list want subscribed
:status 404: The application not found.
:status 200: successful request.
"""
webhook_type = request.args.get('webhook_type', default=0, type=int)
self._check_authority(webhook_type)
data = request.get_json() or {}
validate_fields(webhook_schema, data, partial=False)
if webhook_type == Webhook.TYPE_UNIVERSAL:
webhook = Webhook.create(data['webhook_url'], webhook_type)
return api_response()
application_name = request.args['application_name']
application = Application.get_by_name(application_name)
webhook = Webhook.create(data['webhook_url'], webhook_type)
for action_name in data.get('event_list', []):
action_type = getattr(action_types, action_name)
webhook.subscribe(application.id, action_type)
return api_response()
def add_test_webhook_sub(prepare_data, application_id):
for data in prepare_data:
webhook = Webhook.create(
data['webhook_url'], data.get('webhook_type', 0))
if webhook.is_normal:
for action_name in data['event_list']:
action_type = getattr(action_types, action_name)
webhook.subscribe(application_id, action_type)
yield webhook, data
def test_update_webhook_url(db):
instance = Webhook.create('http://www.foo.bar')
assert instance.id > 0
assert instance.url == 'http://www.foo.bar'
assert instance.hook_type == Webhook.TYPE_NORMAL
instance.update_url('http://www.foo.me')
assert instance.url == 'http://www.foo.me'
assert Webhook.get(instance.id).url == instance.url
def test_delete_webhook_fialed(
client, test_application, test_application_token):
headers = {'Authorization': test_application_token}
r = client.get('/api/webhook/100', headers=headers)
assert r.status_code == 404
assert r.json['status'] == 'NotFound'
assert r.json['message'] == 'Webhook not registered.'
webhook = Webhook.create('http://www.foo.bar')
r = client.delete('/api/webhook/%s' % webhook.id, headers=headers,
query_string={'application_name': 'foo'})
assert r.status_code == 400
def test_get_webhook_list(client, test_application_token):
webhook_list = [
{
'webhook_url': 'http://www.foo.bar',
'webhook_type': 0,
},
{
'webhook_url': 'http://www.foo.me',
'webhook_type': 1,
},
{
'webhook_url': 'http://www.bar.me',
'webhook_type': 0
}
]
for data in webhook_list:
webhook = Webhook.create(data['webhook_url'], data['webhook_type'])
data.update(webhook_id=webhook.id)
headers = {'Authorization': test_application_token}
r = client.get('/api/webhook', headers=headers)
assert_response_ok(r)
assert r.json['data']['webhook_list'] == webhook_list
def test_update_webhook_then_get(
client, test_application_token, test_application):
webhook_normal = Webhook.create('http://www.foo.bar')
headers = {
'Authorization': test_application_token,
'Content-Type': 'application/json',
}
query_string = {'application_name': test_application.application_name}
r = client.get('/api/webhook/%s' % webhook_normal.id, headers=headers,
query_string=query_string)
assert_response_ok(r)
assert r.json['data'] == {
'webhook_id': webhook_normal.id,
'webhook_url': webhook_normal.url,
'webhook_type': webhook_normal.hook_type,
'event_list': []
}
payload = {
'event_list': ['UPDATE_CONFIG', 'DELETE_CONFIG'],
'webhook_url': 'http://abc.example.com',
'webhook_type': 0,
}
r = client.put('/api/webhook/%s' % webhook_normal.id, headers=headers,
query_string=query_string, data=json.dumps(payload))
assert_response_ok(r)
r = client.get('/api/webhook/%s' % webhook_normal.id, headers=headers,
query_string=query_string)
assert_response_ok(r)
assert r.json['data'] == {
'webhook_id': webhook_normal.id,
'webhook_url': payload['webhook_url'],
'webhook_type': webhook_normal.hook_type,
'event_list': payload['event_list'],
}
def add_webhook_subscriptions(test_application):
webhook = Webhook.create('http://foo.foo.bar')
for action_name in action_types._action_map:
action_type = getattr(action_types, action_name)
webhook.subscribe(test_application.id, action_type)
def universal_webhook():
return Webhook.create('http://universal.foo.bar', Webhook.TYPE_UNIVERSAL)
def test_get_all_universal(db):
Webhook.create('http://uni.foo.bar', hook_type=Webhook.TYPE_UNIVERSAL)
Webhook.create('http://normal.foo.bar')
hooks = Webhook.get_all_universal()
assert [x.url for x in hooks] == ['http://uni.foo.bar']
def test_universal_webhook_subscribe(db, test_application):
webhook = Webhook.create('http://foo.foo.bar', Webhook.TYPE_UNIVERSAL)
action_type = action_types.CREATE_CONFIG_CLUSTER
sub = webhook.subscribe(test_application.id, action_type)
assert sub is None
assert webhook.get_subscription(test_application.id, action_type) is None