Fix bugs in relative keypoint handler

1 files changed, 32 insertions, 0 deletions

diff --git a/solo-tool-project/test/handlers_integrationtest.py b/solo-tool-project/test/handlers_integrationtest.py
new file mode 100644
index 0000000..85050d6
--- /dev/null
+++ b/solo-tool-project/test/handlers_integrationtest.py
@@ -0,0 +1,32 @@
+import pytest
+
+from fixtures import soloTool, testSongs, mockPlayer, songPool
+
+from solo_tool.handlers import keyPointRelative
+
+testCases = [
+    ([0.1, 0.3], 0.0, +1, 0.1, "Start +1"),
+    ([0.1, 0.3], 0.1, +1, 0.3, "First +1"),
+    ([0.1, 0.3], 0.2, +1, 0.3, "Between +1"),
+    ([0.1, 0.3], 0.3, +1, 0.3, "Second +1"),
+    ([0.1, 0.3], 0.4, +1, 0.4, "End +1"),
+
+    ([0.1, 0.3], 0.0, -1, 0.0, "Start -1"),
+    ([0.1, 0.3], 0.1, -1, 0.1, "First -1"),
+    ([0.1, 0.3], 0.2, -1, 0.1, "Between -1"),
+    ([0.1, 0.3], 0.3, -1, 0.1, "Second -1"),
+    ([0.1, 0.3], 0.4, -1, 0.3, "End -1"),
+
+    ([0.0, 0.3], 0.0, -1, 0.0, "0.0 -1"),
+]
+
+@pytest.mark.parametrize("keyPoints,current,delta,expected,description", testCases)
+def test_keyPointRelativeEdgeCases(soloTool, testSongs, keyPoints, current, delta, expected, description):
+    soloTool.addSong(testSongs[0])
+    soloTool.keyPoints = keyPoints
+    soloTool.keyPoint = current
+
+    handler = keyPointRelative(soloTool, delta)
+    handler()
+
+    assert soloTool.keyPoint == expected, description