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 get(self):
db = DatabaseHandler()
return db.get_entries(), 200, {'Access-Control-Allow-Origin': '*'}
