Rename test_generator.py to test_stream_generator.py

Author: m4dcoder

st2common/tests/unit/test_generator.py renamed to st2common/tests/unit/test_stream_generator.py

@@ -15,8 +15,14 @@
 
 import mock
 import unittest2
+
 from st2common.stream import listener
 
+INCLUDE = "test"
+END_EVENT = "test_end_event"
+END_ID = "test_end_id"
+EVENTS = [(INCLUDE, MockBody("notend")), (END_EVENT, MockBody(END_ID))]
+
 
 class MockBody(object):
 
@@ -25,12 +31,6 @@ def __init__(self, id):
         self.status = "succeeded"
 
 
-INCLUDE = "test"
-END_EVENT = "test_end_event"
-END_ID = "test_end_id"
-EVENTS = [(INCLUDE, MockBody("notend")), (END_EVENT, MockBody(END_ID))]
-
-
 class MockQueue():
 
     def __init__(self):
@@ -71,7 +71,3 @@ def test_generator(self, mock_queue,
         events = EVENTS.append('')
         for index, val in enumerate(app_iter):
             self.assertEquals(val, events[index])
-
-
-if __name__ == "__main__":
-    unittest2.main()