Feat: Allow to override FSMonitor class

Will be used to test polling in addition to native FSMonitor.
ljmccarthy · Jul 4, 2018 · 2d2044b · 2d2044b
1 parent 651d64b
commit 2d2044b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fsmonitor/__init__.py b/fsmonitor/__init__.py
@@ -25,9 +25,9 @@
     from .polling import FSMonitor
 
 class FSMonitorThread(threading.Thread):
-    def __init__(self, callback=None, autostart=True):
+    def __init__(self, callback=None, autostart=True, fsmonitor_class=None):
         threading.Thread.__init__(self)
-        self.monitor = FSMonitor()
+        self.monitor = (fsmonitor_class or FSMonitor)()
         self.callback = callback
         self._events = []
         self._events_lock = threading.Lock()