From 061744b69f2976448f423b84804d64b809e38808 Mon Sep 17 00:00:00 2001 From: VirtuBox Date: Wed, 23 Oct 2019 12:43:19 +0200 Subject: [PATCH] Fix tests --- tests/cli/13_test_stack.py | 9 +++++---- tests/cli/plugins/test_example.py | 2 +- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/cli/13_test_stack.py b/tests/cli/13_test_stack.py index 63939cb..c6b19b3 100644 --- a/tests/cli/13_test_stack.py +++ b/tests/cli/13_test_stack.py @@ -10,10 +10,11 @@ class CliTestCaseStack(test.WOTestCase): self.app.close() def test_wo_cli_stack_install(self): - self.app = get_test_app(argv=['stack', 'install']) - self.app.setup() - self.app.run() - self.app.close() + argv = ['stack', 'install'] + with self.make_app(argv=argv) as app: + app.run() + self.eq(app.pargs.stack.install) + def test_wo_cli_stack_install_web(self): self.app = get_test_app(argv=['stack', 'install', '--web']) diff --git a/tests/cli/plugins/test_example.py b/tests/cli/plugins/test_example.py index 5de3c2d..dc101f8 100644 --- a/tests/cli/plugins/test_example.py +++ b/tests/cli/plugins/test_example.py @@ -1,6 +1,6 @@ """Tests for Example Plugin.""" -from wo.utils import WOTestCase +from wo.utils.test import WOTestCase class ExamplePluginTestCase(WOTestCase):