def test_should_throw_exception_when_requesting_github_event_api_fails(
self, request_mock):
request_mock.get(
url=
'https://api.github.com/users/mirrorhanyu/received_events?page=1',
reason='request event fails',
status_code=500)
with self.assertRaises(HTTPError) as context:
event.get_event(self.event, None)
self.assertEqual(context.exception.response.reason,
'request event fails')
def test_add_mod_del(self):
event.add_event(self.event1)
elist = event.get_event(event.get_email(self.token))
id_event = elist[0]["id"]
self.modevent1["id"] = id_event
event.mod_event(self.modevent1)
ev = r.table("event").get(self.modevent1["id"]).run()
self.assertSequenceEqual(ev["name"], self.modevent1["name"])
event.del_event(self.token, id_event)
elist = event.get_event(event.get_email(self.token))
self.assertListEqual(elist, [])
def test_should_return_correct_response_when_requesting_github_event_api_success(
self, request_mock):
request_mock.get(
url=
'https://api.github.com/users/mirrorhanyu/received_events?page=1',
text=json.dumps(self.received_event),
status_code=200,
headers=self.received_event_headers)
event_response = event.get_event(self.event, None)
event_response_status_code = event_response.get('statusCode')
event_response_body = json.loads(event_response.get('body'))
event_response_headers = event_response.get('headers')
self.assertEqual(event_response_status_code, 200)
self.assertEqual(event_response_headers.get('max-page'), 9)
self.assertEqual(len(event_response_body), 1)
self.assertEqual(len(event_response_body[0].keys()), 6)
self.assertEqual(event_response_body[0].get('type'), 'starred')
self.assertEqual(event_response_body[0].get('createdAt'),
'2019-10-01T13:22:49Z')
self.assertEqual(event_response_body[0].get('actorName'), 'fi3ework')
self.assertEqual(event_response_body[0].get('actorNickname'),
'fi3ework')
self.assertEqual(event_response_body[0].get('actorAvatar'),
'https://avatars.githubusercontent.com/u/12322740?')
self.assertEqual(event_response_body[0].get('repo'), 'unadlib/usm')
def test_flexible(self):
flexible_lunch.set_lunch(fevent)
id = event.get_event(event.get_email(self.token))[0]["id"]
event.add_event(event1)
ev = r.table("event").get(id).run()
self.assertEqual(ev["start"], "2017-10-10 13:00+00:00")
event.add_event(event2)
ev = r.table("event").get(id).run()
self.assertEqual(ev["start"], "2017-10-10 13:00+00:00")
def end_test_api():
r.connect(rt_server.ip, rt_server.port, rt_server.db_name).repl()
user = "******"
ev_list = event.get_event(user)
for e in ev_list:
r.table("event").get(e["id"]).delete().run()
r.table("event_submit").get(e["id"]).delete().run()
r.table("user").get(user).delete().run()
return "OK"
def export_events(uri_list):
for i in range(len(uri_list)):
if i % 10 == 0:
print('Exporting URI #' + str(i))
uri = uri_list[i]
event_dict = event.get_event(uri)
with open(DATA_CACHE_PATH + "events/" + uri + '.pickle', 'wb') as file:
pickle.dump(event_dict, file)
print("Exported " + DATA_CACHE_PATH + "events/" + uri + '.pickle')
def test_overlay(self):
event.add_event(self.event1)
event.add_event(self.modevent1)
elist = event.get_event(self.token)
for e in elist:
self.assertTrue(e["alarm"])
def get_event():
token = flask.request.args.get('token', '')
if not token_check(token):
return "Session Expired"
json_list = event.get_event(event.get_email(token))
return flask.jsonify(json_list)
def get_event(event_uri):
try:
d = event.get_event(event_uri)
except:
d = {'error': "Something's gone wrong"}
return jsonify(d)
#!/usr/bin/env python
# encoding: utf-8
import functools
from .event import register_event
def register(name=None):
def wrapper(func):
@functools.wraps(func)
def _wrapper(*args, **kwargs):
result_val = func(*args, **kwargs)
key = func.__name__ if not name else name
register_event(key, result_val)
return _wrapper
return wrapper
if __name__ == '__main__':
@register()
def a():
return 1
a()
from event import get_event
print get_event('a')
print('time:')
for c in t:
print(c)
print('characters: ', nr)
print('locations: ', ns)
print('time: ', t)
dialogs = dialog.get_dialog(text)
print('dialogues:')
for c in dialogs:
print(c)
print('dialogues:', dialogs)
events = event.get_event(text)
print('events:')
for c in events:
print(c)
print(events)
story_dict = {
"title": "白雪公主",
"contents": {
"summarize":text,
"characters": nr,
"locations": ns,
"time": t,
"dialogue": dialogs,
"events": events
