def test_track_playtime_counts_time_between_track_start_and_track_end(
self):
session = Session()
session.add_event(Event('track_start', 50, 'A', '101'))
session.add_event(Event('track_end', 100, 'A', '101'))
assert session.track_playtime == 50
def test_duration_is_time_between_last_and_first_event(self):
session = Session()
session.add_event(Event('track_end', 50, 'A', '101'))
session.add_event(Event('track_end', 150, 'A', '101'))
session.add_event(Event('track_end', 100, 'A', '101'))
assert session.duration == 50
def test_last_time_comes_from_last_event(self):
session = Session()
session.add_event(Event('track_end', 50, 'A', '101'))
session.add_event(Event('track_end', 150, 'A', '101'))
session.add_event(Event('track_end', 100, 'A', '101'))
assert session.last_time == 100
def test_ad_count_counts_number_of_ad_starts(self):
session = Session()
session.add_event(Event('ad_start', 50, 'A', '101'))
session.add_event(Event('ad_start', 50, 'A', '101'))
session.add_event(Event('ad_end', 50, 'A', '101'))
session.add_event(Event('ad_start', 50, 'A', '101'))
assert session.ad_count == 3
def test_track_playtime_supports_paused_heartbeated_unfinished_sessions(
self):
session = Session()
session.add_event(Event('track_start', 50, 'A', '101'))
session.add_event(Event('pause', 60, 'A', '101'))
session.add_event(Event('track_heartbeat', 70, 'A', '101'))
assert session.track_playtime == 10
def test_track_playtime_supports_pauses(self):
session = Session()
session.add_event(Event('track_start', 50, 'A', '101'))
session.add_event(Event('pause', 60, 'A', '101'))
session.add_event(Event('play', 70, 'A', '101'))
session.add_event(Event('track_end', 100, 'A', '101'))
assert session.track_playtime == 40
def test_end(self):
event = Event(start=datetime.datetime(year=2000, month=6, day=15, hour=12, minute=30, second=30),
duration=datetime.timedelta(seconds=3700))
self.assertEqual(event.end().year, 2000)
self.assertEqual(event.end().month, 6)
self.assertEqual(event.end().day, 15)
self.assertEqual(event.end().hour, 13)
self.assertEqual(event.end().minute, 32)
self.assertEqual(event.end().second, 10)
def test_get_formatted_output_returns_in_correct_format(self):
session = Session()
session.add_event(Event('stream_start', 50, 'A', '101'))
session.add_event(Event('ad_start', 60, 'A', '101'))
session.add_event(Event('ad_end', 70, 'A', '101'))
session.add_event(Event('track_start', 80, 'A', '101'))
session.add_event(Event('track_heartbeat', 90, 'A', '101'))
assert session.get_formatted_output() == {
'user_id': 'A',
'content_id': '101',
'session_start': 50,
'session_end': 90,
'total_time': 40,
'track_playtime': 10,
'event_count': 5,
'ad_count': 1
}
description='something',
vendor='somewhere',
amount_cents=1234,
creator=u1)
t1.participants.append(u1)
t1.participants.append(u2)
t2 = Transaction(name='Burgers',
description='whoah',
vendor='somewhere2',
amount_cents=5555,
creator=u2)
t2.participants.append(u3)
t2.participants.append(u2)
db.session.add(t1)
db.session.add(t2)
db.session.commit()
e1 = Event(name='Pennapps',
description='sleeping? bah.',
creator=u1,
participants=[u1, u2, u3],
settled=False)
e1.transactions.append(t1)
e1.transactions.append(t2)
db.session.add(e1)
db.session.commit()
else:
print 'dropping all tables...'
db.drop_all()
print 'creating all tables...'
db.create_all()
def test_is_expired_is_true_60_seconds_after_last_event(self):
session = Session()
session.add_event(Event('ad_start', 50, 'A', '101'))
assert session.is_expired(50 + 60)
def test_is_expired_is_false_59_seconds_after_last_event(self):
session = Session()
session.add_event(Event('ad_start', 50, 'A', '101'))
assert not session.is_expired(50 + 59)
def test_track_playtime_works_without_track_end(self):
session = Session()
session.add_event(Event('track_start', 50, 'A', '101'))
session.add_event(Event('track_heartbeat', 60, 'A', '101'))
assert session.track_playtime == 10
def test_session_id_comes_from_first_event(self):
session = Session()
session.add_event(Event('track_end', 100, 'A', '101'))
session.add_event(Event('track_end', 100, 'B', '101'))
assert session.session_id == ('A', '101')
def test_repr(self):
event = Event('track_end', 100, 'A', '101')
assert (repr(event) == "Event(event_type='track_end', "
"timestamp=100, user_id='A', "
"content_id='101')")
def test_session_id_property(self):
event = Event('track_end', 100, 'A', '101')
assert event.session_id == ('A', '101')
{
'name': "Test event 2",
'lat': "40.123435",
'lng': "40.23432"
},
{
'name': "Test event 3",
'lat': "40.123435",
'lng': "40.23432"
},
{
'name': "Test event 4",
'lat': "40.123435",
'lng': "40.23432"
},
]
try:
db.drop_all()
except:
pass
db.create_all()
for event in events:
db.session.add(Event(**event))
db.session.commit()
loaded_events = Event.query.all()
print(json.dumps([x.as_dict() for x in loaded_events]))