def test_from_and_to_dict(self):
utl.add_categories_to_db()
category = utl.CATEGORIES[-1]
utl.add_sensors_to_db()
sensor = utl.SENSORS[-1]
r = Reading()
reading = {
"data": 123,
"category": category["name"],
"sensor": sensor["name"],
"timestamp": 123,
}
# Test the from_dict() method
r.from_dict(reading)
self.assertEqual(r.data, reading["data"])
self.assertEqual(r.category.name, reading["category"])
self.assertEqual(r.sensor.name, reading["sensor"])
self.assertEqual(r.timestamp, reading["timestamp"])
# Test the to_dict() method
r_dict = r.to_dict()
self.assertEqual(r_dict["data"], reading["data"])
self.assertEqual(r_dict["category"]["name"], reading["category"])
self.assertEqual(r_dict["sensor"]["name"], reading["sensor"])
self.assertEqual(r_dict["timestamp"], reading["timestamp"])
def test_add_category(self):
utl.add_categories_to_db()
category = utl.CATEGORIES[-1]
r = Reading()
# Test adding a category to a reading
r.add_category(category["name"])
self.assertEqual(r.category.name, category["name"])
def test_add_sensor(self):
utl.add_categories_to_db()
utl.add_sensors_to_db()
sensor = utl.SENSORS[-1]
r = Reading()
# Test adding a sensor to a reading
r.add_sensor(sensor["name"])
self.assertEqual(r.sensor.name, sensor["name"])
def store_readings(self, readings):
db.session.add(Reading(readings))
db.session.commit()