1 files changed, 50 insertions, 5 deletions

diff --git a/tests/test_streaming.py b/tests/test_streaming.py
index 8c2160a..883b773 100644
--- a/tests/test_streaming.py
+++ b/tests/test_streaming.py
@@ -154,6 +154,15 @@ def test_unknown_event():
     assert listener.deletes == []
     assert listener.heartbeats == 0
 
+def test_invalid_event():
+    """But not too tolerant"""
+    listener = Listener()
+    with pytest.raises(MastodonMalformedEventError):
+        listener.handle_stream_([
+            'event: whatup',
+            'data: {}',
+            '',
+        ])
 
 def test_missing_event_name():
     listener = Listener()
@@ -257,9 +266,11 @@ def test_stream_user(api, api2):
     
     updates = []
     notifications = []
+    deletes = []
     listener = CallbackStreamListener(
         update_handler = lambda x: updates.append(x),
-        notification_handler = lambda x: notifications.append(x)
+        notification_handler = lambda x: notifications.append(x),
+        delete_handler = lambda x: deletes.append(x)
     )
     
     posted = []
@@ -268,7 +279,9 @@ def test_stream_user(api, api2):
         posted.append(api.status_post("only real cars respond."))
         posted.append(api2.status_post("@mastodonpy_test beep beep I'm a jeep"))
         posted.append(api2.status_post("on the internet, nobody knows you're a plane"))
-        time.sleep(3)
+        time.sleep(1)
+        api.status_delete(posted[0])
+        time.sleep(2)
         streamingClose()
         
     t = threading.Thread(args=(), target=do_activities)
@@ -282,13 +295,45 @@ def test_stream_user(api, api2):
         if stream != None:
             stream.close()
         
-    assert len(updates) == 2
+    assert len(updates) == 1
     assert len(notifications) == 1
+    assert len(deletes) == 1
     
     assert updates[0].id == posted[0].id
-    assert updates[1].id == posted[0].id
+    assert deletes[0] == posted[0].id
     assert notifications[0].status.id == posted[1].id
     
     t.join()
     
-    
\ No newline at end of file
+@pytest.mark.vcr(match_on=['path'])
+def test_stream_user_local(api, api2):
+    patchStreaming()
+    
+    updates = []
+    notifications = []
+    listener = CallbackStreamListener(
+        local_update_handler = lambda x: updates.append(x),
+    )
+    
+    posted = []
+    def do_activities():
+        time.sleep(5)
+        posted.append(api.status_post("it's cool guy"))
+        time.sleep(3)
+        streamingClose()
+        
+    t = threading.Thread(args=(), target=do_activities)
+    t.start()
+    
+    stream = None
+    try:
+        stream = api.stream_user(listener, run_async=True)
+        time.sleep(13)
+    finally:
+        if stream != None:
+            stream.close()
+        
+    assert len(updates) == 1
+    assert updates[0].id == posted[0].id
+    
+    t.join()
\ No newline at end of file