Mocked out db.py