浏览代码

Added test for confecho.

Mike Naberezny 16 年之前
父节点
当前提交
0be81934fc
共有 2 个文件被更改,包括 23 次插入4 次删除
  1. 3 4
      src/supervisor/confecho.py
  2. 20 0
      src/supervisor/tests/test_confecho.py

+ 3 - 4
src/supervisor/confecho.py

@@ -13,9 +13,8 @@
 ##############################################################################
 
 import pkg_resources
+import sys
 
-def main():
+def main(out=sys.stdout):
     config = pkg_resources.resource_string(__name__, 'skel/sample.conf')
-    print config
-    
-    
+    out.write(config)

+ 20 - 0
src/supervisor/tests/test_confecho.py

@@ -0,0 +1,20 @@
+"""Test suite for supervisor.confecho"""
+
+import unittest
+from StringIO import StringIO
+from supervisor import confecho
+
+class TopLevelFunctionTests(unittest.TestCase):
+    def test_main_writes_data_out_that_looks_like_a_config_file(self):
+        sio = StringIO()
+        confecho.main(out=sio)
+
+        output = sio.getvalue()
+        self.assertTrue("[supervisord]" in output)
+
+
+def test_suite():
+    return unittest.findTestCases(sys.modules[__name__])
+
+if __name__ == '__main__':
+    unittest.main(defaultTest='test_suite')