def test_schedule():
from src.services.database import schedule
start = datetime.datetime(2017, 1, 1, 1)
end = datetime.datetime(2017, 1, 2, 1)
StateChange(start, end, 0, 0, True).save()
start = datetime.datetime(2017, 1, 2, 1)
end = datetime.datetime(2017, 1, 2, 1)
StateChange(start, end, 0, 0, False).save()
now = datetime.datetime(2017, 1, 1, 1)
assert StateChange.get_current(now).fan is True
schedule.remove()
start = datetime.datetime(2017, 1, 2, 1)
end = datetime.datetime(2017, 1, 2, 1)
StateChange(start, end, 0, 0, False).save()
start = datetime.datetime(2017, 1, 1, 1)
end = datetime.datetime(2017, 1, 2, 1)
StateChange(start, end, 0, 0, True).save()
now = datetime.datetime(2017, 1, 1, 1)
assert StateChange.get_current(now).fan is True
schedule.remove()
print("schedule passed")
def remove_schedule():
to_remove = StateChange.from_dictionary(request.get_json(force=True))
result = schedule.remove({"_id": to_remove.id})
return Response(json.dumps({"data": result}, default=json_util.default), mimetype='application/json')