class TestDatabaseHandler(unittest.TestCase):
def setUp(self):
self.db = DatabaseHandler()
def test_get(self):
results = self.db.get_entries()
self.assertTrue(results)
def test_new_entry(self):
new_entry = EntryDataObject('foo', 'bar', 100.35)
result = self.db.upsert_entry(new_entry)
self.assertTrue(result)
# cleanup
self.db.delete_entry(result)
def test_invalid_entry(self):
new_entry = EntryDataObject('foo', 'bar', 'amount')
result = self.db.upsert_entry(new_entry)
self.assertFalse(result)
def test_delete_entry(self):
# setup
new_entry = EntryDataObject('foo', 'bar', 100)
result = self.db.upsert_entry(new_entry)
success = self.db.delete_entry(result)
self.assertTrue(success)
def test_update_entry(self):
existing_entry = EntryDataObject('foo', 'bar', 6, False, 8)
result = self.db.upsert_entry(existing_entry)
self.assertTrue(result)
def delete(self, entry_id):
db = DatabaseHandler()
success = db.delete_entry(entry_id)
if success:
return '', 204, {'Access-Control-Allow-Origin': '*'}
else:
api.abort(500, 'Server failed to upsert entry.')