AB controller integration test added

1 files changed, 2 insertions, 3 deletions

diff --git a/abcontroller.py b/abcontroller.py
index 9d232c1..e9eace4 100644
--- a/abcontroller.py
+++ b/abcontroller.py
@@ -3,8 +3,8 @@ from collections import namedtuple
 _AB = namedtuple("_AB", ["a", "b"])
 
 class ABController:
-    def __init__(self, enabled=True):
-        self.setPositionCallback = None
+    def __init__(self, enabled=True, callback=None):
+        self.setPositionCallback = callback
         self._limits = dict() # dictionary of all songs
         self._songLimits = None # list of limits for selected song
         self._currentLimits = None # a/b positions of selected limit
@@ -42,7 +42,6 @@ class ABController:
     def positionChanged(self, position):
         if not self._currentLimits:
             return
-
         if position > self._currentLimits.b and self.setPositionCallback and self._enabled:
             self.setPositionCallback(self._currentLimits.a)