Renamed TestPillarServer to PillarTestServer

2016-08-31 11:30:55 +02:00
parent 92dd4bf701
commit cf0e6f85de
1 changed files with 5 additions and 4 deletions
--- a/tests/abstract_attract_test.py
+++ b/tests/abstract_attract_test.py
@@ -1,12 +1,13 @@
-from pillar_server.tests import TestPillarServer, AbstractPillarTest
+from pillar.tests import PillarTestServer, AbstractPillarTest
 from attract_server import AttractExtension

-class TestAttractServer(TestPillarServer):
+
+class AttractTestServer(PillarTestServer):
    def __init__(self, *args, **kwargs):
-        super(TestAttractServer, self).__init__(*args, **kwargs)
+        super(AttractTestServer, self).__init__(*args, **kwargs)

        self.load_extension(AttractExtension(), '/attract')


 class AbstractAttractTest(AbstractPillarTest):
-    pillar_server_class = TestAttractServer
+    pillar_server_class = AttractTestServer