1 files changed, 13 insertions, 14 deletions
diff --git a/deluge/plugins/WebUi/deluge_webui/tests/test_plugin_webui.py b/deluge/plugins/WebUi/deluge_webui/tests/test_plugin_webui.py
index 56e1cc023..1badedca0 100644
--- a/deluge/plugins/WebUi/deluge_webui/tests/test_plugin_webui.py
+++ b/deluge/plugins/WebUi/deluge_webui/tests/test_plugin_webui.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 #
 # Copyright (C) 2016 bendikro <bro.devel+deluge@gmail.com>
 #
@@ -6,44 +5,44 @@
 # the additional special exception to link portions of this program with the OpenSSL library.
 # See LICENSE for more details.
 #
-
-from __future__ import unicode_literals
-
-from twisted.trial import unittest
+import pytest
+import pytest_twisted
 
 import deluge.component as component
 from deluge.core.core import Core
 from deluge.core.rpcserver import RPCServer
 from deluge.tests import common
-from deluge.tests.basetest import BaseTestCase
 
 common.disable_new_release_check()
 
 
-class WebUIPluginTestCase(BaseTestCase):
-    def set_up(self):
-        common.set_tmp_config_dir()
+@pytest.mark.usefixtures('component')
+class TestWebUIPlugin:
+    @pytest_twisted.async_yield_fixture(autouse=True)
+    async def set_up(self, request):
+        self = request.instance
         self.rpcserver = RPCServer(listen=False)
         self.core = Core()
-        return component.start()
+        await component.start()
+
+        yield
 
-    def tear_down(self):
         def on_shutdown(result):
             del self.rpcserver
             del self.core
 
-        return component.shutdown().addCallback(on_shutdown)
+        await component.shutdown().addCallback(on_shutdown)
 
     def test_enable_webui(self):
         if 'WebUi' not in self.core.get_available_plugins():
-            raise unittest.SkipTest('WebUi plugin not available for testing')
+            pytest.skip('WebUi plugin not available for testing')
 
         d = self.core.enable_plugin('WebUi')
 
         def result_cb(result):
             if 'WebUi' not in self.core.get_enabled_plugins():
                 self.fail('Failed to enable WebUi plugin')
-            self.assertTrue(result)
+            assert result
 
         d.addBoth(result_cb)
         return d