Fix plugin tests

mozilla · Aug 1, 2018 · 484da40 · 484da40
1 parent f247218
commit 484da40
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_plugin.py b/tests/test_plugin.py
@@ -2,16 +2,18 @@
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
-from pypom import Region
+from pypom import Region, hookimpl
 
 
 def test_after_wait_for_page_to_load(page):
     log = []
 
     class Plugin:
+        @hookimpl
         def pypom_after_wait_for_page_to_load(self, page):
             log.append(1)
 
+        @hookimpl
         def pypom_after_wait_for_region_to_load(self, region):
             log.append(2)