metaprogramming examples

metaprog/oscon-schedule/test_schedule2.py

@@ -0,0 +1,61 @@
+import shelve
+import pytest
+
+import schedule2 as schedule
+
+
+@pytest.yield_fixture
+def db():
+    with shelve.open(schedule.DB_NAME) as the_db:
+        if schedule.CONFERENCE not in the_db:
+            schedule.load_db(the_db)
+        yield the_db
+
+
+def test_record_attr_access():
+    rec = schedule.Record({'spam': 99, 'eggs': 12})
+    assert rec.spam == 99
+    assert rec.eggs == 12
+
+
+def test_record_repr():
+    rec = schedule.Record({'spam': 99, 'eggs': 12})
+    assert repr(rec).startswith('<Record object at 0x')
+    rec2 = schedule.Record({'name': 'Fido'})
+    assert repr(rec2) == "<Record 'Fido'>"
+
+
+def test_conference_record(db):
+    assert schedule.CONFERENCE in db
+
+
+def test_speaker_record(db):
+    speaker = db['speaker.3471']
+    assert speaker.name == 'Anna Martelli Ravenscroft'
+
+
+def test_event_record(db):
+    event = db['event.33950']
+    assert event.name == 'There *Will* Be Bugs'
+
+
+def test_event_venue(db):
+    schedule.Event.set_db(db)
+    event = db['event.33950']
+    assert event.venue_serial == 1449
+    assert event.venue == db['venue.1449']
+    assert event.venue.name == 'Portland 251'
+
+
+def test_event_speakers(db):
+    schedule.Event.set_db(db)
+    event = db['event.33950']
+    assert len(event.speakers) == 2
+    anna_and_alex = [db['speaker.3471'], db['speaker.5199']]
+    assert event.speakers == anna_and_alex
+
+def test_event_no_speakers(db):
+    schedule.Event.set_db(db)
+    event = db['event.36848']
+    assert len(event.speakers) == 0
+