From 61f00f4110b1d33827e3bd19ffef34f10be8273a Mon Sep 17 00:00:00 2001 From: maxbr Date: Mon, 17 Oct 2016 14:56:11 +0200 Subject: Add unit tests JIRA: RELENG-12 This commit increases test statement coverage to 84%. It also fixes the bugs that emerged while testing. Change-Id: I696091f1a07f8b7647872c7cb15f4368a4690619 Signed-off-by: maxbr --- pharos-dashboard/src/notification/models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'pharos-dashboard/src/notification/models.py') diff --git a/pharos-dashboard/src/notification/models.py b/pharos-dashboard/src/notification/models.py index 2d19918..0ee275d 100644 --- a/pharos-dashboard/src/notification/models.py +++ b/pharos-dashboard/src/notification/models.py @@ -26,7 +26,8 @@ class BookingNotification(models.Model): } def save(self, *args, **kwargs): - notifications = self.booking.bookingnotification_set.filter(type=self.type) - if notifications.count() > 1: + notifications = self.booking.bookingnotification_set.filter(type=self.type).exclude( + id=self.id) + if notifications.count() > 0: raise ValueError('Doubled Notification') return super(BookingNotification, self).save(*args, **kwargs) \ No newline at end of file -- cgit 1.2.3-korg