update unit test

.travis.yml

@@ -13,7 +13,7 @@ before_install:
   - sudo tests/socksify/install.sh
 script:
   - pep8 .
-  - nosetests shadowsocks/*
+  - python tests/nose_plugin.py
   - python tests/test.py -c tests/table.json
   - python tests/test.py -c tests/aes.json
   - python tests/test.py -c tests/aes-ctr.json

tests/nose_plugin.py

@@ -0,0 +1,27 @@
+import sys
+import nose
+from nose.plugins.base import Plugin
+
+class ExtensionPlugin(Plugin):
+
+    name = "ExtensionPlugin"
+
+    def options(self, parser, env):
+        Plugin.options(self,parser,env)
+
+    def configure(self, options, config):
+        Plugin.configure(self, options, config)
+        self.enabled = True
+
+    def wantFile(self, file):
+        return file.endswith('.py')
+
+    def wantDirectory(self,directory):
+        return True
+
+    def wantModule(self,file):
+        return True
+
+
+if __name__ == '__main__':
+    nose.main(addplugins=[ExtensionPlugin()])