vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Change notificationworker test to pytest

Browse source
This commit is contained in:
Joseph Schorr 2017-07-12 15:10:58 +03:00
parent 50c2f1fde8
commit 8ec198228c
1 changed files with 25 additions and 38 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
workers/notificationworker/test/test_notificationworker.py
View file

@ -1,20 +1,11 @@
import unittest
from data import model from data import model
from workers.notificationworker import NotificationWorker from workers.notificationworker.notificationworker import NotificationWorker
from initdb import setup_database_for_testing, finished_database_for_testing from test.fixtures import *
class NotificationWorkerTests(unittest.TestCase): def test_basic_notification(initialized_db):
def setUp(self):
setup_database_for_testing(self)
def tearDown(self):
finished_database_for_testing(self)
def test_basic_notification(self):
# Ensure the public user doesn't have any notifications. # Ensure the public user doesn't have any notifications.
target_user = model.user.get_user('public') target_user = model.user.get_user('public')
self.assertEquals(0, len(list(model.notification.list_notifications(target_user)))) assert len(list(model.notification.list_notifications(target_user))) == 0
# Add a basic build notification. # Add a basic build notification.
repo = model.repository.get_repository('devtable', 'simple') repo = model.repository.get_repository('devtable', 'simple')
@ -38,8 +29,4 @@ class NotificationWorkerTests(unittest.TestCase):
}) })
# Ensure the notification was handled. # Ensure the notification was handled.
self.assertEquals(1, len(list(model.notification.list_notifications(target_user)))) assert len(list(model.notification.list_notifications(target_user))) == 1
if __name__ == '__main__':
unittest.main()