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)