diff --git a/.travis.yml b/.travis.yml index 46419ea..a430a71 100644 --- a/.travis.yml +++ b/.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 diff --git a/tests/nose_plugin.py b/tests/nose_plugin.py new file mode 100644 index 0000000..1db6351 --- /dev/null +++ b/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()]) \ No newline at end of file