|
3 | 3 |
|
4 | 4 |
|
5 | 5 | class TestCLI(TestCase):
|
6 |
| - # change dir before running tests |
7 |
| - |
8 |
| - os.chdir('..') |
9 |
| - |
10 |
| - def test_newevent(self): |
11 |
| - from tvwb import newevent |
12 |
| - assert newevent(name='TestEvent') |
13 |
| - self.assertRaises(ValueError, newevent, name='!') |
14 |
| - self.assertRaises(ValueError, newevent, name='@') |
15 |
| - self.assertRaises(ValueError, newevent, name='#') |
16 |
| - self.assertRaises(ValueError, newevent, name='test event') |
17 |
| - self.assertRaises(ValueError, newevent, name='test-event') |
18 |
| - self.assertRaises(ValueError, newevent, name='test_event') |
19 |
| - |
20 |
| - def test_newaction(self): |
21 |
| - from tvwb import newaction |
22 |
| - assert newaction(name='TestAction') |
23 |
| - self.assertRaises(ValueError, newaction, name='!') |
24 |
| - self.assertRaises(ValueError, newaction, name='@') |
25 |
| - self.assertRaises(ValueError, newaction, name='#') |
26 |
| - self.assertRaises(ValueError, newaction, name='test action') |
27 |
| - self.assertRaises(ValueError, newaction, name='test-action') |
28 |
| - self.assertRaises(ValueError, newaction, name='test_action') |
| 6 | + |
| 7 | + def setUp(self) -> None: |
| 8 | + self.prefix = 'python tvwb.py' |
| 9 | + self.initial_settings = open('settings.py', 'r').read() |
| 10 | + |
| 11 | + def test_create_and_register_action(self): |
| 12 | + |
| 13 | + # create action |
| 14 | + cmd = f'{self.prefix} action:create CustomAction --no-register' |
| 15 | + try: |
| 16 | + os.system(cmd) |
| 17 | + except Exception as e: |
| 18 | + self.fail(e) |
| 19 | + |
| 20 | + # register action |
| 21 | + cmd = f'{self.prefix} action:register CustomAction' |
| 22 | + try: |
| 23 | + os.system(cmd) |
| 24 | + except Exception as e: |
| 25 | + self.fail(e) |
| 26 | + |
| 27 | + def tearDown(self) -> None: |
| 28 | + os.remove('components/actions/custom_action.py') |
| 29 | + |
| 30 | + # restore settings.py |
| 31 | + with open('settings.py', 'w') as settings_file: |
| 32 | + settings_file.write(self.initial_settings) |
| 33 | + settings_file.close() |
0 commit comments