def test_is_time_daily(self):
interval_min = 40
now = datetime.datetime.now()
send_time_date = now + datetime.timedelta(minutes=interval_min - 1)
send_time = send_time_date.time()
dict_sample_config = {
"interval_min": interval_min,
"daily_send_hour": send_time,
"weekly_send_hour": send_time,
"weekly_send_day": 4,
}
expected_result = True
pintrck = Pintracker(dict_sample_config)
actual_result = pintrck.is_time_daily()
self.assertEqual(actual_result, expected_result)
send_time_date = now - datetime.timedelta(minutes=interval_min + 1)
send_time = send_time_date.time()
dict_sample_config = {
"interval_min": interval_min,
"daily_send_hour": send_time,
"weekly_send_hour": send_time,
"weekly_send_day": 4,
}
expected_result = False
pintrck = Pintracker(dict_sample_config)
actual_result = pintrck.is_time_daily()
self.assertEqual(actual_result, expected_result)
def test_check_dir_total_pines(self):
dict_sample_config = {
"stock_dir": self.base_path,
"smtp_server":"mail.example.com",
"email":"*****@*****.**",
"password":"******",
"interval_min":40,
"daily_send_hour": datetime.time(18,00),
"weekly_send_hour": datetime.time(8,00),
"weekly_send_day": 4,
"min_pines": {
"jazztel (activado)":100,
"orange":20,
"tarjetalia":10
}
}
pintrck = Pintracker(dict_sample_config)
num_pines = pintrck.check_dir_total_pines(
os.path.join(self.base_path, self.dir))
self.assertEqual(num_pines, self.expected_result)
def test_num_pins(self):
dict_sample_config = {
"stock_dir": self.base_path,
"smtp_server":"mail.example.com",
"email":"*****@*****.**",
"password":"******",
"interval_min":40,
"daily_send_hour": datetime.time(18,00),
"weekly_send_hour": datetime.time(8,00),
"weekly_send_day": 4,
"min_pines": {
"jazztel (activado)":100,
"orange":20,
"tarjetalia":10
}
}
pintrck = Pintracker(dict_sample_config)
for file in self.expected_results.iterkeys():
pines_file_path = os.path.join(self.base_path, file)
pines_leidos = pintrck.numero_pines(pines_file_path)
self.assertEqual(self.expected_results[file], pines_leidos)
def test_estado_pines(self):
dict_sample_config = {
"stock_dir": self.base_path,
"smtp_server":"mail.example.com",
"email":"*****@*****.**",
"password":"******",
"interval_min":40,
"daily_send_hour": datetime.time(18,00),
"weekly_send_hour": datetime.time(8,00),
"weekly_send_day": 4,
"min_pines": {
"jazztel (activado)":100,
"orange":20,
"tarjetalia":10
}
}
dict_expected_result = {
"jazztel (activado)": 85,
"orange":85,
"tarjetalia":85
}
pintrck = Pintracker(dict_sample_config)
actual_result = pintrck.estado_pines()
self.assertEqual(actual_result, dict_expected_result)
