Fixing on_ticker handler

yahoofinancelive · Mar 31, 2021 · 4a4edbf · 4a4edbf
1 parent 1c96b16
commit 4a4edbf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/yliveticker_test.py b/tests/yliveticker_test.py
@@ -21,7 +21,7 @@ def on_msg(ws, msg):
             } == msg
 
         yticker = YLiveTicker(on_ticker=on_msg)
-        yticker.on_message(
+        yticker.on_message(None,
             "CgRNU0ZUFYVrMUMY0KLMjcNcKgNOTVMwCDgBRZ\
                 +B3L9IuPn8CVU0IjRDXYVrMUNlQApHwNgBBA=="
         )
diff --git a/yliveticker/__init__.py b/yliveticker/__init__.py
@@ -61,7 +61,7 @@ def on_message(self, ws, message):
         if self.on_ticker is None:
             print(json.dumps(data))
         else:
-            self.on_ticker(data)
+            self.on_ticker(ws, data)
 
     def on_error(self, ws, error):
         if self.on_custom_error is None: