class TestDbPolygons(unittest.TestCase):
def setUp(self):
self.db = RiakDb()
self.db.create_user(U_EMAIL, U_NAME, U_PASS, U_ADDR)
p = self.db.create_polygon("Nevada", "Area 51", U_EMAIL, date.today(),
None, ['type1', 'typeB'])
self.poly_id = p['id']
def tearDown(self):
self.db.delete_polygon(self.poly_id)
self.db.delete_user(U_EMAIL)
def test_get_own_polygon(self):
poly = self.db.get_polygon(self.poly_id, U_EMAIL)
self.assertEqual(poly['name'], "Area 51")
self.assertEqual(poly['type'], ['type1', 'typeB'])
def test_polygon_start_date_as_datetime(self):
dt = self.db.get_polygon(self.poly_id, U_EMAIL)['start_date']
self.assertEqual(dt, date.today())
def test_get_other_polygon(self):
self.assertIsNone(self.db.get_polygon(self.poly_id, "wrong_user"))
def test_delete_polygon(self):
self.db.delete_polygon(self.poly_id)
self.assertIsNone(self.db.get_polygon(self.poly_id, U_EMAIL))
class ServerTest(AsyncHTTPTestCase):
def setUp(self):
self.db = RiakDb()
self.db.create_user(TEST_EMAIL, "test name", "testpass", "New York")
super().setUp()
def tearDown(self):
self.db.delete_user(TEST_EMAIL)
super().tearDown()
def get_app(self):
return serve2.Application(self.db)
class TestDbPolygonHarvest(unittest.TestCase):
def setUp(self):
self.db = RiakDb()
harvest = self.db.create_harvest(1, str(date.today()), 5, "bushels")
self.harvest_id = harvest['id']
def tearDown(self):
self.db.delete_harvest(self.harvest_id)
self.db.delete_user(U_EMAIL)
def test_get_own_harvest(self):
units = self.db.get_harvest(self.harvest_id)['units']
self.assertEqual(units, "bushels")
def test_delete_task(self):
self.db.delete_harvest(self.harvest_id)
self.assertIsNone(self.db.get_harvest(self.harvest_id))
class TestDbPolygonTasks(unittest.TestCase):
def setUp(self):
self.db = RiakDb()
task = self.db.create_task(1, "water tomatoes", str(date.today()), 1,
False, "water the tomatoes")
self.task_id = task['id']
def tearDown(self):
self.db.delete_user(U_EMAIL)
self.db.delete_task(self.task_id)
def test_get_own_task(self):
name = self.db.get_task(self.task_id)['name']
self.assertEqual(name, "water tomatoes")
def test_delete_task(self):
self.db.delete_task(self.task_id)
self.assertIsNone(self.db.get_task(self.task_id))
class TestDbPolygonTypes(unittest.TestCase):
def setUp(self):
self.db = RiakDb()
self.db.create_poly_type("Test", False, None, ["child1", "child2"])
def tearDown(self):
self.db.delete_polygon("Test")
self.db.delete_user(U_EMAIL)
def test_get_poly_type(self):
self.assertEqual(
self.db.get_poly_type("Test")['children'][0], "child1")
# TODO: test_get_deleted_poly_type
def test_delete_poly_type(self):
self.db.delete_poly_type("Test")
self.assertIsNone(self.db.get_poly_type("Test"))
class TestDbPolygonNotes(unittest.TestCase):
def setUp(self):
self.db = RiakDb()
note = self.db.create_note(1, str(date.today()), "NoteTitle",
"NoteContent")
self.note_id = note['id']
def tearDown(self):
self.db.delete_note(self.note_id)
self.db.delete_user(U_EMAIL)
def test_get_note(self):
title = self.db.get_note(self.note_id)['title']
self.assertEqual(title, "NoteTitle")
def test_delete_note(self):
self.db.delete_note(self.note_id)
self.assertIsNone(self.db.get_note(self.note_id))
class TestDbUsers(unittest.TestCase):
def setUp(self):
self.db = RiakDb()
self.db.create_user(U_EMAIL, U_NAME, U_PASS, U_ADDR)
def tearDown(self):
self.db.delete_user(U_EMAIL)
def test_get_user(self):
self.assertEqual(self.db.get_user(U_EMAIL)['name'], U_NAME)
def test_check_correct_password(self):
self.assertTrue(self.db.login(U_EMAIL, U_PASS))
def test_check_incorrect_password(self):
self.assertFalse(self.db.login(U_EMAIL, "bad"))
def test_delete_user(self):
self.db.delete_user(U_EMAIL)
self.assertIsNone(self.db.get_user(U_EMAIL))
