def add_data_to_database(self, data): with self.app.app_context(): self.db = SQLAlchemy() self.db.init_app(self.app) for datapoint in data: test = Test() test.name = "Test" + str(datapoint['test']) self.db.session.add(test) user = User() user.auth0_id = "User" + str(datapoint['user']) user.name = "User" + str(datapoint['user']) user.email = "User" + str(datapoint['user']) self.db.session.add(user) self.db.session.commit() for j in range(datapoint['data']): result = TestResult() result.test_id = test.id result.user_id = user.id result.value = j result.time = datetime.datetime.utcnow() self.db.session.add(result) self.db.session.commit()
def setUp(self): user = User() user.id = 1 user.username = '******' user.email = '*****@*****.**' self.user = user