def test_add_bite(self): table_name = "volleyball_single" table_col = 0 data = {'table': table_name, 'subject_col_id': table_col, 'slice': 0, 'total': 1, 'technique': 'found-majority'} Bite.delete().execute() # delete all Bites Apple.delete().execute() # delete all Apples result = self.app.post('/add', data=data, content_type='multipart/form-data') self.assertIn('apple', result.json) apple_id = result.json['apple'] self.assertEqual(result.status_code, 200, msg=result.data) database.connect(reuse_if_open=True) self.assertEqual(len(Bite.select()), 1) sleep(sleep_time) result = self.app.get('/status') self.assertEqual(result.status_code, 200, msg=result.data) self.assertTrue(result.is_json) j = { "apples": [ { "apple": table_name, "status": STATUS_COMPLETE, "complete": True, "agreement": 1.0, "elected": 0, } ] } self.assertEqual(elect(apple_id, 'majority'), None) # just for the coverage self.assertEqual(elect(apple_id, 'found-majority'), None) # just for the coverage
def test_add_multiple_bite(self): table_name = "volleyball_double" table_col = 0 data = {'table': table_name, 'subject_col_id': table_col, 'slice': 0, 'total': 2, 'technique': 'majority'} Bite.delete().execute() # delete all Bites Apple.delete().execute() # delete all Apples result = self.app.post('/add', data=data, content_type='multipart/form-data') self.assertEqual(result.status_code, 200, msg=result.data) database.connect(reuse_if_open=True) self.assertEqual(len(Bite.select()), 1) sleep(sleep_time) result = self.app.get('/status') self.assertEqual(result.status_code, 200, msg=result.data) self.assertTrue(result.is_json) j = { "apples": [ { "apple": table_name, "status": STATUS_NEW, "complete": False, "elected": None, "agreement": None, } ] } self.assertDictEqual(result.get_json(), j) data = {'table': table_name, 'subject_col_id': table_col, 'slice': 1, 'total': 2, 'technique': 'majority'} result = self.app.post('/add', data=data, content_type='multipart/form-data') self.assertEqual(result.status_code, 200, msg=result.data) database.connect(reuse_if_open=True) self.assertEqual(len(Bite.select()), 2) sleep(sleep_time) result = self.app.get('/status') self.assertEqual(result.status_code, 200, msg=result.data) self.assertTrue(result.is_json) j = { "apples": [ { "apple": table_name, "status": STATUS_COMPLETE, "complete": True, "agreement": 1.0, "elected": table_col, } ] }