def test_create_event():
from server.models.event import Event
from server.models.user import User
with client() as c:
user1 = User.find(key='test')
user2 = User.find(key='test2')
create_event(c, user1, 'test event')
resp = api_get(c, user1, '/api/events/')
assert resp.status_code == 200, 'oops {}'.format(resp.data)
data = ujson.loads(resp.data)
assert data['objects'][0]['name'] == 'test event'
assert data['objects'][0]['group'] == {'$ref': 'Group:1'}
event_id = data['objects'][0]['id']
event = Event.find(event_id)
assert user1.is_attending(event), 'user is attending the new event'
assert 1 == Event.select().count()
event_id = create_event(c, user1, 'test event 2')
user1 = User.find(key='test')
assert user1.get_attending_id() == event_id
assert 2 == Event.select().count(), '2 events total now'
count, page, results = Event.select().group(user1.group).execute()
assert 1 == count, 'only 1 in the group though, since the other has no attendees now'
assert results[0].name == 'test event 2'
def test_private_event():
from server.models.event import Event, EventAttendee
from server.models.user import User
with client() as c:
user1 = User.find(key='test')
user2 = User.find(key='test2')
make_friends(c, user1, user2)
event_id = create_event(c, user1, 'test event 1', privacy='private')
assert 1 == Event.select().count()
assert 0 == Event.select().group(user1.group).count()
assert 1 == Event.select().user(user1).count()
assert 0 == Event.select().user(user2).count()
event = Event.find(event_id)
resp = api_post(c, user1, '/api/events/{}/invites'.format(event_id),
{'invited_id': user2.id})
assert resp.status_code == 200, 'oops {}'.format(resp.data)
assert 1 == Event.select().user(user2).count()
count, page, results = EventAttendee.select().event(event).user(
user2).execute()
assert count == 1
assert results[0] == user1
def delete_history(u, f):
with wigo_db.transaction(commit_on_select=False):
for message in EventMessage.select().key(skey(u,
'event_messages')):
if message.user and message.event:
message.remove_for_user(f)
for event in Event.select().user(u):
if wigo_db.sorted_set_is_member(user_attendees_key(f, event),
u.id):
event.remove_from_user_attending(f, u)
def capture_history(u, f):
# capture each of the users posted photos
with wigo_db.transaction(commit_on_select=False):
for message in EventMessage.select().key(skey(
u, 'event_messages')).min(min):
if message.user and message.event:
message.record_for_user(f)
# capture the events being attended
for event in Event.select().user(u).min(min):
if u.is_attending(event) and f.can_see_event(event):
event.add_to_user_attending(f, u)
def test_user_events():
from server.models.event import Event
from server.models.user import User
with client() as c:
user1 = User.find(key='test')
user2 = User.find(key='test2')
create_event(c, user1, 'test event 1')
create_event(c, user2, 'test event 2')
assert 2 == Event.select().count(), '2 events total now'
assert 2 == Event.select().group(user1.group).count()
count, page, results = Event.select().user(user1).execute()
assert 1 == count
assert results[0].name == 'test event 1'
count, page, results = Event.select().user(user2).execute()
assert 1 == count
assert results[0].name == 'test event 2'
def new_group(group_id):
group = Group.find(group_id)
logger.info('new group {} created, importing events'.format(
group.name.encode('utf-8')))
num_imported = 0
imported = set()
min = epoch(group.get_day_end() - timedelta(days=7))
with wigo_db.transaction(commit_on_select=False):
for close_group in get_close_groups(group.latitude, group.longitude,
100):
if close_group.id == group.id:
continue
for event in Event.select().group(close_group).min(min):
# only import the events the group actually owns
if event.group_id != close_group.id:
continue
# no double imports
if event.id not in imported:
event.update_global_events(group=group)
imported.add(event.id)
num_imported += 1
for event in Event.select().key(skey('global', 'events')).min(min):
if event.id not in imported:
event.update_global_events(group=group)
imported.add(event.id)
num_imported += 1
logger.info('imported {} events into group {}'.format(
num_imported, group.name.encode('utf-8')))
group.track_meta('last_event_change', expire=None)
group.status = 'active'
group.save()
def test_events_with_friends():
from server.models.event import Event
from server.models.user import User
with client() as c:
user1 = User.find(key='test')
user2 = User.find(key='test2')
create_event(c, user1, 'test event 1')
assert 1 == Event.select().user(user1).count()
assert 0 == Event.select().user(user2).count()
make_friends(c, user1, user2)
assert 1 == Event.select().user(user1).count()
assert 1 == Event.select().user(user2).count()
# remove friends, event should disappear from user2
resp = api_delete(c, user2,
'/api/users/me/friends/{}'.format(user1.id))
assert resp.status_code == 200, 'oops {}'.format(resp.data)
assert 1 == Event.select().user(user1).count()
assert 0 == Event.select().user(user2).count()
def __get_city_events(group, min, max):
query = Event.select().group(group).limit(1000)
query = query.min(min).max(max)
return query.execute()