class Cafeteira:
def __init__(self):
self.twitter = TwitterManager()
self.scheduled_times = self.setup_schedule()
self.schedule = ScheduleManager(self.scheduled_times)
self.machine = MachineManager()
self.machine.listen_button(self.button_callback)
self.start_schedule_process()
self.start_api_process()
def get_machine_status(self):
return self.machine.machine_status
def get_schedule_times(self):
return self.scheduled_times
def button_callback(self, pin):
logger.debug("Button panic activaded!")
if self.machine.machine_status is MachineStatus.stand_by:
self.start_coffee_routine_async()
else:
self.twitter.tweet_turn_off_machine()
self.machine.go_back_stand_by()
def setup_schedule(self):
FILE_PATH = "schedule_coffee.txt"
reader = ScheduleReader(FILE_PATH)
times = reader.read_scheduled_times()
return times
def check_schedule_are_ok(self):
while True:
if self.schedule.its_time() and self.machine.machine_status is MachineStatus.stand_by:
logger.debug("Schedule time is Ok!")
self.start_coffee_routine_async()
time.sleep(30)
def start_api_process(self):
logger.debug("Create process for API")
self.api = ApiManager(self)
self.api.start()
thread_api = Thread(target=self.api.start)
thread_api.start()
def start_schedule_process(self):
logger.debug("Create process for Schedule Reader")
thread_reading_schedule = Thread(target=self.check_schedule_are_ok)
thread_reading_schedule.start()
def start_coffee_routine_async(self):
if self.machine.machine_status is MachineStatus.stand_by:
logger.debug("Create thread for coffee")
self.twitter.tweet_turn_on_machine()
thread_making_coffee = Thread(target=self.machine.start_coffee_routine)
thread_making_coffee.start()
class MachineManagerTest(unittest.TestCase):
def setUp(self):
self.target = MachineManager()
self.adapter_start_mock = MagicMock()
self.adapter_stop_mock = MagicMock()
self.target.machine_adapter.start = self.adapter_start_mock
self.target.machine_adapter.stop = self.adapter_stop_mock
MachineManager.wait_one_minute = MagicMock()
def test_machine_routine_should_make_and_keep_coffee_hot(self):
self.target.make_coffee = Mock(wraps=self.target.make_coffee)
self.target.keep_coffee_hot = Mock(wraps=self.target.keep_coffee_hot)
self.target.go_back_stand_by = Mock(wraps=self.target.go_back_stand_by)
self.target.start_coffee_routine()
self.assertTrue(self.target.make_coffee.called)
self.assertTrue(self.target.keep_coffee_hot.called)
self.assertTrue(self.target.go_back_stand_by.called)
def test_machine_should_not_make_coffee_when_machine_st_warming(self):
self.target.machine_status = MachineStatus.warming_coffee
self.target.make_coffee()
self.assertNotEquals(
self.target.machine_status, MachineStatus.making_coffee
)
self.assertFalse(self.adapter_start_mock.called)
self.assertNotEquals(MachineManager.wait_one_minute.call_count, 30)
def test_machine_should_no_make_coffee_when_machine_st_making_coffee(self):
self.target.machine_status = MachineStatus.warming_coffee
self.target.make_coffee()
self.assertNotEquals(
self.target.machine_status, MachineStatus.making_coffee
)
self.assertFalse(self.adapter_start_mock.called)
self.assertNotEquals(MachineManager.wait_one_minute.call_count, 30)
def test_machine_should_make_coffee(self):
self.target.machine_status = MachineStatus.stand_by
self.target.make_coffee()
self.assertEquals(
self.target.machine_status, MachineStatus.making_coffee
)
self.assertTrue(self.adapter_start_mock.called)
self.assertEquals(MachineManager.wait_one_minute.call_count, 30)
def test_machine_should_keep_coffee_hot(self):
self.target.machine_status = MachineStatus.making_coffee
self.target.keep_coffee_hot()
self.assertEquals(
self.target.machine_status, MachineStatus.warming_coffee
)
self.assertEquals(self.adapter_stop_mock.call_count, 30)
self.assertEquals(self.adapter_start_mock.call_count, 30)
self.assertEquals(MachineManager.wait_one_minute.call_count, 60)
def test_machine_should_not_keep_coffee_hot_when_machine_st_stand_by(self):
self.target.machine_status = MachineStatus.stand_by
self.target.keep_coffee_hot()
self.assertNotEquals(
self.target.machine_status, MachineStatus.warming_coffee
)
self.assertNotEquals(self.adapter_stop_mock.call_count, 30)
self.assertNotEquals(self.adapter_start_mock.call_count, 30)
self.assertNotEquals(MachineManager.wait_one_minute.call_count, 60)
def test_machine_should_not_keep_coffee_hot_when_machine_st_warming(self):
self.target.machine_status = MachineStatus.warming_coffee
self.target.keep_coffee_hot()
self.assertNotEquals(self.adapter_stop_mock.call_count, 30)
self.assertNotEquals(self.adapter_start_mock.call_count, 30)
self.assertNotEquals(MachineManager.wait_one_minute.call_count, 60)
def test_machine_should_set_machine_standby_when_making_coffee(self):
self.target.machine_status = MachineStatus.making_coffee
self.target.go_back_stand_by()
self.assertEquals(
self.target.machine_status, MachineStatus.stand_by
)
self.assertEquals(self.adapter_stop_mock.call_count, 1)
def test_machine_should_set_machine_standby_when_warming_coffee(self):
self.target.machine_status = MachineStatus.warming_coffee
self.target.go_back_stand_by()
self.assertEquals(self.target.machine_status, MachineStatus.stand_by)
self.assertEquals(self.adapter_stop_mock.call_count, 1)