Fix broken tests

endpoints/notificationmethod.py

@@ -6,7 +6,7 @@ import re
 from flask.ext.mail import Message
 from app import mail, app
 from data import model
-from workers.worker import JobException
+from workers.queueworker import JobException
 
 logger = logging.getLogger(__name__)
 

workers/repositoryactioncounter.py

@@ -10,14 +10,7 @@ POLL_PERIOD_SECONDS = 10
 
 logger = logging.getLogger(__name__)
 
-class RepositoryActionCountWorker(Worker):
-  def __init__(self):
-    super(RepositoryActionCountWorker, self).__init__()
-    self.add_operation(self._count_repository_actions, POLL_PERIOD_SECONDS)
-
-  def _count_repository_actions(self):
-    """ Counts actions for a random repository for the previous day. """
-
+def count_repository_actions():
   with UseThenDisconnect(app.config):
     try:
       # Get a random repository to count.
@@ -46,6 +39,16 @@ class RepositoryActionCountWorker(Worker):
     except Repository.DoesNotExist:
       logger.debug('No further repositories to count')
 
+
+class RepositoryActionCountWorker(Worker):
+  def __init__(self):
+    super(RepositoryActionCountWorker, self).__init__()
+    self.add_operation(self._count_repository_actions, POLL_PERIOD_SECONDS)
+
+  def _count_repository_actions(self):
+    """ Counts actions for a random repository for the previous day. """
+    count_repository_actions()
+
 if __name__ == "__main__":
   worker = RepositoryActionCountWorker()
   worker.start()