From 55a774d90e5ade4176087b1ffae8eabc802d414c Mon Sep 17 00:00:00 2001
From: Aaron Hill <aa1ronham@gmail.com>
Date: Tue, 20 Aug 2013 06:25:37 -0400
Subject: [PATCH] Added sitestream endpoint

---
 tests/test_streaming.py | 16 ++++++++++++----
 tweepy/streaming.py     | 16 ++++++++++++++++
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/tests/test_streaming.py b/tests/test_streaming.py
index fdeb2c9..f796417 100644
--- a/tests/test_streaming.py
+++ b/tests/test_streaming.py
@@ -45,15 +45,23 @@ class TweepyStreamTests(unittest.TestCase):
     def tearDown(self):
         self.stream.disconnect()
 
+    def on_connect():
+        API(self.auth).update_status(mock_tweet())
+
+
     def test_userstream(self):
         # Generate random tweet which should show up in the stream.
-        def on_connect():
-            API(self.auth).update_status(mock_tweet())
-
-        self.listener.connect_cb = on_connect
+        
+        self.listener.connect_cb = self.on_connect
         self.listener.status_stop_count = 1
         self.stream.userstream()
         self.assertEqual(self.listener.status_count, 1)
+    
+    def test_sitestream(self):
+        self.listener.connect_cb = self.on_connect
+        self.listener.status_stop_count = 1
+        self.sitestream(follow=[self.auth.get_username()])
+        self.assertEqual(self.listener.status_count, 1)
 
     def test_sample(self):
         self.listener.status_stop_count = 10
diff --git a/tweepy/streaming.py b/tweepy/streaming.py
index cd6bffb..6f9a506 100644
--- a/tweepy/streaming.py
+++ b/tweepy/streaming.py
@@ -296,6 +296,22 @@ class Stream(object):
         self.parameters['delimited'] = 'length'
         self._start(async)
 
+    def sitestream(self, follow, stall_warnings=False, with_='user', replies=False, async=False):
+        self.parameters = {}
+        if self.running:
+            raise TweepError('Stream object already connected!')
+        self.url = '/%s/site.json' % STREAM_VERSION
+        self.parameters['follow'] = ','.join(map(str, follow))
+        self.parameters['delimited'] = 'length'
+        if stall_warnings:
+            self.parameters['stall_warnings'] = stall_warnings
+        if with_:
+            self.parameters['with'] = with_
+        if replies:
+            self.parameters['replies'] = replies
+        self.body = urlencode_noplus(self.parameters)
+        self._start(async)
+
     def disconnect(self):
         if self.running is False:
             return
-- 
2.25.1