blob: 7682d79a6b4251b2db8ad0734180edb7955b7bac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import unittest
import nose.selector
from nose.config import Config
from nose.plugins.base import Plugin
from nose.plugins.manager import PluginManager
class TestSelectorPlugins(unittest.TestCase):
def test_rejection(self):
class EvilSelector(Plugin):
def wantFile(self, filename, package=None):
if 'good' in filename:
return False
return None
c = Config(plugins=PluginManager(plugins=[EvilSelector()]))
s = nose.selector.Selector(c)
s2 = nose.selector.Selector(Config())
assert s.wantFile('test_neutral.py')
assert s2.wantFile('test_neutral.py')
assert s.wantFile('test_evil.py')
assert s2.wantFile('test_evil.py')
assert not s.wantFile('test_good.py')
assert s2.wantFile('test_good.py')
if __name__ == '__main__':
unittest.main()
|