Added test, some error checking in Qt interface

1 files changed, 24 insertions, 2 deletions

diff --git a/midi_launchpad_mini_integrationtest.py b/midi_launchpad_mini_integrationtest.py
index beeee51..0bb9a82 100644
--- a/midi_launchpad_mini_integrationtest.py
+++ b/midi_launchpad_mini_integrationtest.py
@@ -64,8 +64,7 @@ def test_startStopAndPauseButtons(uut, midiWrapperMock, playerMock):
     
     midiWrapperMock.simulateInput(playPauseButton)
     assert playerMock.state == PlayerMock.PLAYING
-    msg = midiWrapperMock.getLatestMessage()
-    assert msg == (playPauseButton, MidiController.LED_GREEN, 0)
+    assert midiWrapperMock.getLatestMessage() == (playPauseButton, MidiController.LED_GREEN, 0)
 
     midiWrapperMock.simulateInput(stopButton)
     assert playerMock.state == PlayerMock.STOPPED
@@ -88,6 +87,29 @@ def test_startStopAndPauseButtons(uut, midiWrapperMock, playerMock):
     midiWrapperMock.simulateInput(stopButton)
     assert playerMock.state == PlayerMock.STOPPED
 
+def test_startPauseButtonLed(uut, midiWrapperMock, playerMock, soloTool):
+    playPauseButton = 112
+    stopButton = 96
+    uut.connect()
+
+    assert playerMock.state == PlayerMock.STOPPED
+    
+    soloTool.play()
+    assert playerMock.state == PlayerMock.PLAYING
+    assert midiWrapperMock.getLatestMessage() == (playPauseButton, MidiController.LED_GREEN, 0)
+
+    soloTool.stop()
+    assert playerMock.state == PlayerMock.STOPPED
+    assert midiWrapperMock.getLatestMessage() == (playPauseButton, MidiController.LED_YELLOW, 0)
+
+    soloTool.pause()
+    assert midiWrapperMock.getLatestMessage() == (playPauseButton, MidiController.LED_YELLOW, 0)
+    assert playerMock.state == PlayerMock.PAUSED
+
+    soloTool.play()
+    assert midiWrapperMock.getLatestMessage() == (playPauseButton, MidiController.LED_GREEN, 0)
+    assert playerMock.state == PlayerMock.PLAYING
+
 def test_jumpToAButton(uut, midiWrapperMock, soloTool, playerMock):
     ab = (0.5, 0.6)
     uut.connect()