def test_process_sleep_specified(hallo_getter):
hallo, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
mood_date = date(2019, 1, 18)
mood_datetime = datetime.combine(mood_date, time(13, 13, 6))
moods = ["Happiness", "Anger", "Tiredness"]
spreadsheet = DailysSpreadsheetMock(test_user, test_chan)
# Setup field
times = [
DailysMoodField.TIME_WAKE,
time(14, 0, 0), DailysMoodField.TIME_SLEEP
]
field = DailysMoodField(spreadsheet, times, moods)
# Send message
evt_mood = EventMessage(test_server, test_chan, test_user,
"HAT sleep 413").with_raw_data(
RawDataTelegram(
get_telegram_time(mood_datetime)))
field.passive_trigger(evt_mood)
# Check mood response is logged
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_SLEEP in notif_dict
assert "message_id" not in notif_dict[DailysMoodField.TIME_SLEEP]
assert notif_dict[DailysMoodField.TIME_SLEEP]["Happiness"] == 4
assert notif_dict[DailysMoodField.TIME_SLEEP]["Anger"] == 1
assert notif_dict[DailysMoodField.TIME_SLEEP]["Tiredness"] == 3
# Check response is given
data_sleep = test_server.get_send_data(1, test_chan, EventMessage)
assert "added" in data_sleep[0].text.lower()
assert DailysMoodField.TIME_SLEEP in data_sleep[0].text
assert mood_date.isoformat() in data_sleep[0].text
assert "413" in data_sleep[0].text
def test_process_no_mood_query(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
mood_date = date(2019, 1, 18)
mood_datetime = datetime.combine(mood_date, time(13, 13, 6))
moods = ["Happiness", "Anger", "Tiredness"]
msg_id = 41212
mood_data = dict()
mood_data[DailysMoodField.TIME_WAKE] = dict()
mood_data[DailysMoodField.TIME_WAKE]["message_id"] = msg_id
for mood in moods:
mood_data[DailysMoodField.TIME_WAKE][mood] = 3
spreadsheet = DailysSpreadsheetMock(
test_user, test_chan, saved_data={"mood": {
mood_date: mood_data
}})
# Setup field
times = [DailysMoodField.TIME_WAKE, time(14, 0, 0)]
field = DailysMoodField(spreadsheet, times, moods)
# Send message
evt_mood = EventMessage(
test_server, test_chan, test_user,
"413").with_raw_data(RawDataTelegram(get_telegram_time(mood_datetime)))
field.passive_trigger(evt_mood)
# Check mood response is not logged
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_WAKE in notif_dict
assert notif_dict[DailysMoodField.TIME_WAKE]["message_id"] == msg_id
assert notif_dict[DailysMoodField.TIME_WAKE]["Happiness"] == 3
assert notif_dict[DailysMoodField.TIME_WAKE]["Anger"] == 3
assert notif_dict[DailysMoodField.TIME_WAKE]["Tiredness"] == 3
assert str(time(14, 0, 0)) not in notif_dict
# Check no response is given
test_server.get_send_data(0)
def test_trigger_sleep_no_query_if_not_in_times(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
spreadsheet = DailysSpreadsheetMock(test_user,
test_chan,
saved_data={"mood": {}})
# Setup field
times = [DailysMoodField.TIME_WAKE, time(14, 0, 0)]
moods = ["Happiness", "Anger", "Tiredness"]
field = DailysMoodField(spreadsheet, times, moods)
# Send message
evt_sleep = EventMessage(test_server, test_chan, test_user, "night")
field.passive_trigger(evt_sleep)
# Check mood query is not sent or added to saved data
assert evt_sleep.get_send_time().date(
) not in spreadsheet.saved_data["mood"]
test_server.get_send_data(0)
def test_process_time_specified(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
mood_date = date(2019, 1, 18)
mood_datetime = datetime.combine(mood_date, time(13, 13, 6))
moods = ["Happiness", "Anger", "Tiredness"]
msg_id = 41212
mood_data = dict()
mood_data[DailysMoodField.TIME_WAKE] = dict()
mood_data[DailysMoodField.TIME_WAKE]["message_id"] = msg_id
for mood in moods:
mood_data[DailysMoodField.TIME_WAKE][mood] = 3
spreadsheet = DailysSpreadsheetMock(
test_user, test_chan, saved_data={"mood": {
mood_date: mood_data
}})
# Setup field
times = [DailysMoodField.TIME_WAKE, time(14, 0, 0)]
field = DailysMoodField(spreadsheet, times, moods)
# Send message
evt_mood = EventMessage(test_server, test_chan, test_user,
"HAT 1400 413").with_raw_data(
RawDataTelegram(
get_telegram_time(mood_datetime)))
field.passive_trigger(evt_mood)
# Check mood response is logged
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_WAKE in notif_dict
assert notif_dict[DailysMoodField.TIME_WAKE]["message_id"] == msg_id
assert notif_dict[DailysMoodField.TIME_WAKE]["Happiness"] == 3
assert notif_dict[DailysMoodField.TIME_WAKE]["Anger"] == 3
assert notif_dict[DailysMoodField.TIME_WAKE]["Tiredness"] == 3
assert str(time(14, 0, 0)) in notif_dict
assert "message_id" not in notif_dict[str(time(14, 0, 0))]
assert notif_dict[str(time(14, 0, 0))]["Happiness"] == 4
assert notif_dict[str(time(14, 0, 0))]["Anger"] == 1
assert notif_dict[str(time(14, 0, 0))]["Tiredness"] == 3
# Check response is given
data_1400 = test_server.get_send_data(1, test_chan, EventMessage)
assert "added" in data_1400[0].text.lower()
assert str(time(14, 0, 0)) in data_1400[0].text
assert mood_date.isoformat() in data_1400[0].text
assert "413" in data_1400[0].text
def test_trigger_morning_query(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
spreadsheet = DailysSpreadsheetMock(test_user, test_chan)
# Setup field
times = [DailysMoodField.TIME_WAKE, time(14, 0, 0)]
moods = ["Happiness", "Anger", "Tiredness"]
field = DailysMoodField(spreadsheet, times, moods)
# Send message
evt_wake = EventMessage(test_server, test_chan, test_user, "morning")
field.passive_trigger(evt_wake)
# Check mood query is sent
notif_dict = spreadsheet.saved_data["mood"][
evt_wake.get_send_time().date()]
assert DailysMoodField.TIME_WAKE in notif_dict
assert "message_id" in notif_dict[DailysMoodField.TIME_WAKE]
# Check query is given
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "how are you feeling" in data_wake[0].text.lower()
assert DailysMoodField.TIME_WAKE in data_wake[0].text
assert all([mood in data_wake[0].text for mood in moods])
def test_create_from_input__no_static_data(hallo_getter, requests_mock):
# Setup stuff
command_name = "setup dailys field"
command_args = "mood"
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
evt = EventMessage(
test_server,
test_chan,
test_user,
"{} {}".format(command_name, command_args),
)
evt.split_command_text(command_name, command_args)
spreadsheet = DailysSpreadsheetMock(test_user, test_chan)
requests_mock.get("{}/stats/mood/static/".format(spreadsheet.dailys_url),
json=[])
# Try and create dailys field
with pytest.raises(hallo.modules.dailys.dailys_field.DailysException) as e:
DailysMoodField.create_from_input(evt, spreadsheet)
assert "mood field static data has not been set up on dailys system" in str(
e.value).lower()
def test_trigger_sleep_after_midnight(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
mood_date = date(2019, 1, 15)
sleep_time = datetime(2019, 1, 16, 0, 34, 15)
times = [
DailysMoodField.TIME_WAKE,
time(14, 0, 0), DailysMoodField.TIME_SLEEP
]
moods = ["Happiness", "Anger", "Tiredness"]
# Setup
saved_data = dict()
saved_data[DailysMoodField.TIME_WAKE] = dict()
saved_data[DailysMoodField.TIME_WAKE]["message_id"] = 1232
saved_data[str(time(14, 0, 0))] = dict()
saved_data[str(time(14, 0, 0))]["message_id"] = 1234
for mood in moods:
saved_data[DailysMoodField.TIME_WAKE][mood] = 3
saved_data[str(time(14, 0, 0))][mood] = 2
spreadsheet = DailysSpreadsheetMock(
test_user, test_chan, saved_data={"mood": {
mood_date: saved_data
}})
# Setup field
field = DailysMoodField(spreadsheet, times, moods)
# Send message
evt_sleep = EventMessage(
test_server, test_chan, test_user,
"night").with_raw_data(RawDataTelegram(get_telegram_time(sleep_time)))
field.passive_trigger(evt_sleep)
# Check mood query is sent for previous day
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_SLEEP in notif_dict
assert "message_id" in notif_dict[DailysMoodField.TIME_SLEEP]
# Check query is given
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "how are you feeling" in data_wake[0].text.lower()
assert DailysMoodField.TIME_SLEEP in data_wake[0].text
assert all([mood in data_wake[0].text for mood in moods])
def test_trigger_sleep_query(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
moods = ["Happiness", "Anger", "Tiredness"]
evt_sleep = EventMessage(test_server, test_chan, test_user, "night")
saved_data = dict()
saved_data[DailysMoodField.TIME_WAKE] = dict()
saved_data[DailysMoodField.TIME_WAKE]["message_id"] = 1232
saved_data[str(time(14, 0, 0))] = dict()
saved_data[str(time(14, 0, 0))]["message_id"] = 1234
for mood in moods:
saved_data[DailysMoodField.TIME_WAKE][mood] = 3
saved_data[str(time(14, 0, 0))][mood] = 2
spreadsheet = DailysSpreadsheetMock(
test_user,
test_chan,
saved_data={"mood": {
evt_sleep.get_send_time().date(): saved_data
}},
)
# Setup field
times = [
DailysMoodField.TIME_WAKE,
time(14, 0, 0), DailysMoodField.TIME_SLEEP
]
field = DailysMoodField(spreadsheet, times, moods)
# Send message
field.passive_trigger(evt_sleep)
# Check mood query is sent
notif_dict = spreadsheet.saved_data["mood"][
evt_sleep.get_send_time().date()]
assert DailysMoodField.TIME_SLEEP in notif_dict
assert "message_id" in notif_dict[DailysMoodField.TIME_SLEEP]
# Check query is given
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "how are you feeling" in data_wake[0].text.lower()
assert DailysMoodField.TIME_SLEEP in data_wake[0].text
assert all([mood in data_wake[0].text for mood in moods])
def test_trigger_time_exactly_once(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
mood_date = date(2019, 1, 18)
# Setup
spreadsheet = DailysSpreadsheetMock(test_user,
test_chan,
saved_data={"mood": {}})
# Setup field
times = [
DailysMoodField.TIME_WAKE,
time(14, 0, 0), DailysMoodField.TIME_SLEEP
]
moods = ["Happiness", "Anger", "Tiredness"]
field = DailysMoodField(spreadsheet, times, moods)
# Prepare events
evt1 = EventMinute()
evt1.send_time = datetime(2019, 1, 18, 13, 59, 11)
evt2 = EventMinute()
evt2.send_time = datetime(2019, 1, 18, 14, 0, 11)
evt3 = EventMinute()
evt3.send_time = datetime(2019, 1, 18, 14, 1, 11)
# Send time before trigger time
field.passive_trigger(evt1)
# Check mood data not updated and query not sent
assert mood_date not in spreadsheet.saved_data["mood"]
test_server.get_send_data(0)
# Send time after trigger time
field.passive_trigger(evt2)
# Check mood query is sent
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert str(time(14, 0, 0)) in notif_dict
assert "message_id" in notif_dict[str(time(14, 0, 0))]
# Check query is given
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "how are you feeling" in data_wake[0].text.lower()
assert str(time(14, 0, 0)) in data_wake[0].text
assert all([mood in data_wake[0].text for mood in moods])
# Set message ID to something
msg_id = "test_message_id"
notif_dict[str(time(14, 0, 0))]["message_id"] = msg_id
spreadsheet.saved_data["mood"][mood_date] = notif_dict
# Send another time after trigger time
field.passive_trigger(evt3)
# Check mood data not updated and query not sent
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert notif_dict[str(time(14, 0, 0))]["message_id"] == msg_id
test_server.get_send_data(0)
def get_field_objects(test_user, test_chan):
spreadsheet = DailysSpreadsheetMock(test_user, test_chan)
field_obs = list()
field_obs.append(
DailysDuolingoField(spreadsheet, "cabinet",
os.getenv("test_duo_password")))
field_obs.append(
DailysMoodField(
spreadsheet,
[DailysMoodField.TIME_SLEEP,
time(14, 00),
time(22, 00)],
["Happiness", "Anger", "Sleepiness"],
))
field_obs.append(DailysFAField(spreadsheet))
field_obs.append(DailysSleepField(spreadsheet))
field_obs.append(DailysDreamField(spreadsheet))
return field_obs
def test_process_most_recent_sleep_query_after_midnight(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
mood_date = date(2019, 1, 18)
sleep_datetime = datetime.combine(mood_date, time(23, 55, 56))
mood_datetime = datetime.combine(mood_date + timedelta(days=1),
time(0, 3, 2))
msg_id = 41212
mood_data = dict()
spreadsheet = DailysSpreadsheetMock(
test_user, test_chan, saved_data={"mood": {
mood_date: mood_data
}})
# Setup field
times = [DailysMoodField.TIME_SLEEP]
moods = ["Happiness", "Anger", "Tiredness"]
field = DailysMoodField(spreadsheet, times, moods)
# Send sleep message, check response
evt_sleep = EventMessage(test_server, test_chan, test_user,
"sleep").with_raw_data(
RawDataTelegram(
get_telegram_time(sleep_datetime)))
field.passive_trigger(evt_sleep)
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_SLEEP in notif_dict
assert "message_id" in notif_dict[DailysMoodField.TIME_SLEEP]
notif_dict[DailysMoodField.TIME_SLEEP]["message_id"] = msg_id
spreadsheet.saved_data["mood"][mood_date] = notif_dict
test_server.get_send_data()
# Send message
evt_mood = EventMessage(
test_server, test_chan, test_user,
"413").with_raw_data(RawDataTelegram(get_telegram_time(mood_datetime)))
field.passive_trigger(evt_mood)
# Check mood response is logged
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_SLEEP in notif_dict
assert "message_id" in notif_dict[DailysMoodField.TIME_SLEEP]
assert notif_dict[DailysMoodField.TIME_SLEEP]["message_id"] == msg_id
assert notif_dict[DailysMoodField.TIME_SLEEP]["Happiness"] == 4
assert notif_dict[DailysMoodField.TIME_SLEEP]["Anger"] == 1
assert notif_dict[DailysMoodField.TIME_SLEEP]["Tiredness"] == 3
# Check response is given
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "added" in data_wake[0].text.lower()
assert DailysMoodField.TIME_SLEEP in data_wake[0].text
assert mood_date.isoformat() in data_wake[0].text
assert "413" in data_wake[0].text
def test_create_from_input(hallo_getter, requests_mock):
dailys_times = ["WakeUpTime", "12:00:00", "SleepTime"]
dailys_moods = ["happiness", "anger", "tiredness", "boisterousness"]
# Setup stuff
command_name = "setup dailys field"
command_args = "mood"
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
evt = EventMessage(
test_server,
test_chan,
test_user,
"{} {}".format(command_name, command_args),
)
evt.split_command_text(command_name, command_args)
spreadsheet = DailysSpreadsheetMock(test_user, test_chan)
requests_mock.get("{}/stats/mood/static/".format(spreadsheet.dailys_url),
json=[{
"date": "static",
"source": "Mock test data",
"stat_name": "mood",
"data": {
"moods": dailys_moods,
"times": dailys_times
}
}])
# Try and create dailys field
field = DailysMoodField.create_from_input(evt, spreadsheet)
assert field.spreadsheet == spreadsheet
assert isinstance(field.times, list)
assert len(field.times) == 3
assert DailysMoodField.TIME_WAKE in field.times
assert DailysMoodField.TIME_SLEEP in field.times
assert time(12, 0, 0) in field.times
assert isinstance(field.moods, list)
assert len(field.moods) == 4
assert field.moods == dailys_moods
def test_no_trigger_wake_after_processed(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
mood_date = date(2019, 1, 18)
mood_datetime = datetime.combine(mood_date, time(13, 13, 6))
wake_datetime = datetime.combine(mood_date, time(13, 15, 7))
moods = ["Happiness", "Anger", "Tiredness"]
spreadsheet = DailysSpreadsheetMock(test_user, test_chan)
# Setup field
times = [DailysMoodField.TIME_WAKE, time(14, 0, 0)]
field = DailysMoodField(spreadsheet, times, moods)
# Send message
evt_mood = EventMessage(test_server, test_chan, test_user,
"HAT wake 413").with_raw_data(
RawDataTelegram(
get_telegram_time(mood_datetime)))
field.passive_trigger(evt_mood)
# Check mood response is logged
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_WAKE in notif_dict
assert "message_id" not in notif_dict[DailysMoodField.TIME_WAKE]
assert notif_dict[DailysMoodField.TIME_WAKE]["Happiness"] == 4
assert notif_dict[DailysMoodField.TIME_WAKE]["Anger"] == 1
assert notif_dict[DailysMoodField.TIME_WAKE]["Tiredness"] == 3
# Check response is given
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "added" in data_wake[0].text.lower()
assert DailysMoodField.TIME_WAKE in data_wake[0].text
assert mood_date.isoformat() in data_wake[0].text
assert "413" in data_wake[0].text
# Send wake message, ensure no response
evt_wake = EventMessage(test_server, test_chan, test_user,
"morning").with_raw_data(
RawDataTelegram(
get_telegram_time(wake_datetime)))
field.passive_trigger(evt_wake)
# Check query isn't logged
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_WAKE in notif_dict
assert "message_id" not in notif_dict[DailysMoodField.TIME_WAKE]
# Check response wasn't given
test_server.get_send_data(0)
def test_no_trigger_sleep_after_processed_sleep_and_midnight(hallo_getter):
hallo_obj, test_server, test_chan, test_user = hallo_getter({"dailys"})
# Setup
mood_date = date(2019, 1, 18)
sleep_datetime = datetime.combine(mood_date, time(23, 13, 6))
mood_datetime = datetime.combine(mood_date, time(23, 15, 7))
sleep2_datetime = datetime.combine(mood_date + timedelta(days=1),
time(0, 3, 15))
msg_id = 123123
moods = ["Happiness", "Anger", "Tiredness"]
saved_data = dict()
saved_data[DailysMoodField.TIME_WAKE] = dict()
saved_data[DailysMoodField.TIME_WAKE]["message_id"] = 1232
saved_data[str(time(14, 0, 0))] = dict()
saved_data[str(time(14, 0, 0))]["message_id"] = 1234
for mood in moods:
saved_data[DailysMoodField.TIME_WAKE][mood] = 3
saved_data[str(time(14, 0, 0))][mood] = 2
spreadsheet = DailysSpreadsheetMock(
test_user, test_chan, saved_data={"mood": {
mood_date: saved_data
}})
# Setup field
times = [
DailysMoodField.TIME_WAKE,
time(14, 0, 0), DailysMoodField.TIME_SLEEP
]
field = DailysMoodField(spreadsheet, times, moods)
# Send sleep query
evt_sleep1 = EventMessage(test_server, test_chan, test_user,
"sleep").with_raw_data(
RawDataTelegram(
get_telegram_time(sleep_datetime)))
field.passive_trigger(evt_sleep1)
# Check mood query is given and stuff
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_SLEEP in notif_dict
assert "message_id" in notif_dict[DailysMoodField.TIME_SLEEP]
notif_dict[DailysMoodField.TIME_SLEEP]["message_id"] = msg_id
spreadsheet.saved_data["mood"][mood_date] = notif_dict
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "how are you feeling" in data_wake[0].text.lower()
assert DailysMoodField.TIME_SLEEP in data_wake[0].text
assert all([mood in data_wake[0].text for mood in moods])
# Then mood response
evt_mood = EventMessage(
test_server, test_chan, test_user, "413").with_raw_data(
RawDataTelegram(get_telegram_time_reply(mood_datetime, msg_id)))
field.passive_trigger(evt_mood)
# Check mood is recorded and response given
notif_dict = spreadsheet.saved_data["mood"][mood_date]
assert DailysMoodField.TIME_SLEEP in notif_dict
assert "message_id" in notif_dict[DailysMoodField.TIME_SLEEP]
assert notif_dict[DailysMoodField.TIME_SLEEP]["message_id"] == msg_id
assert notif_dict[DailysMoodField.TIME_SLEEP]["Happiness"] == 4
assert notif_dict[DailysMoodField.TIME_SLEEP]["Anger"] == 1
assert notif_dict[DailysMoodField.TIME_SLEEP]["Tiredness"] == 3
data_wake = test_server.get_send_data(1, test_chan, EventMessage)
assert "added" in data_wake[0].text.lower()
assert DailysMoodField.TIME_SLEEP in data_wake[0].text
assert mood_date.isoformat() in data_wake[0].text
assert "413" in data_wake[0].text
# Then midnight
# Another sleep query
evt_sleep1 = EventMessage(test_server, test_chan, test_user,
"sleep").with_raw_data(
RawDataTelegram(
get_telegram_time(sleep2_datetime)))
field.passive_trigger(evt_sleep1)
# Check there's no response
test_server.get_send_data(0)
