[401678c] | 1 | import pytest
|
---|
| 2 | from flowtimer.Schedule import Schedule
|
---|
| 3 | from flowtimer.Phase import Phase
|
---|
| 4 |
|
---|
| 5 | class TestSchedule:
|
---|
| 6 |
|
---|
| 7 | def test_schedule_initialization(self):
|
---|
| 8 | phase1 = Phase("Warm-up", 300)
|
---|
| 9 | phase2 = Phase("Workout", 600)
|
---|
| 10 | schedule = Schedule([phase1, phase2])
|
---|
| 11 |
|
---|
| 12 | assert schedule.phase_list == [phase1, phase2]
|
---|
| 13 | assert schedule.current_phase == phase1
|
---|
| 14 | assert schedule.state == "initial"
|
---|
| 15 |
|
---|
| 16 | def test_schedule_start(self):
|
---|
| 17 | phase1 = Phase("Warm-up", 300)
|
---|
| 18 | schedule = Schedule([phase1])
|
---|
| 19 | schedule.start()
|
---|
| 20 |
|
---|
| 21 | assert schedule.state == "running"
|
---|
| 22 | assert schedule.current_phase.running() is True
|
---|
| 23 |
|
---|
| 24 | def test_schedule_pause(self):
|
---|
| 25 | phase1 = Phase("Warm-up", 300)
|
---|
| 26 | schedule = Schedule([phase1])
|
---|
| 27 | schedule.start()
|
---|
| 28 | schedule.pause()
|
---|
| 29 |
|
---|
| 30 | assert schedule.state == "paused"
|
---|
| 31 | assert schedule.is_paused() is True
|
---|
| 32 |
|
---|
| 33 | def test_schedule_abort(self):
|
---|
| 34 | phase1 = Phase("Warm-up", 300)
|
---|
| 35 | schedule = Schedule([phase1])
|
---|
| 36 | schedule.start()
|
---|
| 37 | schedule.abort()
|
---|
| 38 |
|
---|
| 39 | assert schedule.state == "finished"
|
---|
| 40 | assert schedule.current_phase.finished() is True
|
---|
| 41 |
|
---|
| 42 | def test_schedule_finished(self):
|
---|
| 43 | phase1 = Phase("Warm-up", 300)
|
---|
| 44 | phase2 = Phase("Workout", 600)
|
---|
| 45 | schedule = Schedule([phase1, phase2])
|
---|
| 46 |
|
---|
| 47 | assert schedule.finished() is False
|
---|
| 48 |
|
---|
| 49 | phase1.start()
|
---|
| 50 | phase1.tick(300)
|
---|
| 51 | assert phase1.finished() is True
|
---|
| 52 | assert schedule.finished() is False
|
---|
| 53 |
|
---|
| 54 | schedule.skip()
|
---|
| 55 | schedule.current_phase.tick(600)
|
---|
| 56 | assert schedule.finished() is True
|
---|
| 57 |
|
---|
| 58 | def test_schedule_skip(self):
|
---|
| 59 | phase1 = Phase("Warm-up", 300)
|
---|
| 60 | phase2 = Phase("Workout", 600)
|
---|
| 61 | schedule = Schedule([phase1, phase2])
|
---|
| 62 |
|
---|
| 63 | schedule.start()
|
---|
| 64 | schedule.skip()
|
---|
| 65 |
|
---|
| 66 | assert schedule.current_phase == phase2
|
---|
| 67 |
|
---|
| 68 | def test_schedule_skip_at_final_phase(self):
|
---|
| 69 | phase1 = Phase("Warm-up", 300)
|
---|
| 70 | schedule = Schedule([phase1])
|
---|
| 71 |
|
---|
| 72 | schedule.start()
|
---|
| 73 | schedule.skip()
|
---|
| 74 |
|
---|
| 75 | assert schedule.state == "finished"
|
---|
| 76 |
|
---|
| 77 | def test_schedule_tick(self):
|
---|
| 78 | phase1 = Phase("Warm-up", 300)
|
---|
| 79 | phase2 = Phase("Workout", 600)
|
---|
| 80 | schedule = Schedule([phase1, phase2])
|
---|
| 81 |
|
---|
| 82 | schedule.start()
|
---|
| 83 | schedule.tick(300)
|
---|
| 84 |
|
---|
| 85 | assert phase1.finished() is True
|
---|
| 86 | assert schedule.current_phase == phase2
|
---|
| 87 | assert schedule.state == "running"
|
---|
| 88 |
|
---|
| 89 | def test_schedule_tick_to_completion(self):
|
---|
| 90 | phase1 = Phase("Warm-up", 300)
|
---|
| 91 | phase2 = Phase("Workout", 600)
|
---|
| 92 | schedule = Schedule([phase1, phase2])
|
---|
| 93 |
|
---|
| 94 | schedule.start()
|
---|
| 95 | schedule.tick(300) # Should finish phase1 and move to phase2
|
---|
| 96 | schedule.tick(600) # Should finish phase2 and complete the schedule
|
---|
| 97 |
|
---|
| 98 | assert phase2.finished() is True
|
---|
| 99 | assert schedule.finished() is True
|
---|
| 100 |
|
---|
| 101 | def test_schedule_current_phase_is_final(self):
|
---|
| 102 | phase1 = Phase("Warm-up", 300)
|
---|
| 103 | phase2 = Phase("Workout", 600)
|
---|
| 104 | schedule = Schedule([phase1, phase2])
|
---|
| 105 |
|
---|
| 106 | assert schedule.current_phase_is_final() is False
|
---|
| 107 |
|
---|
| 108 | schedule.skip()
|
---|
| 109 | assert schedule.current_phase_is_final() is True
|
---|
| 110 |
|
---|