| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import unittest | 5 import unittest |
| 6 | 6 |
| 7 from google.appengine.ext import ndb | 7 from google.appengine.ext import ndb |
| 8 | 8 |
| 9 from model import analysis_status as status | 9 from model import analysis_status as status |
| 10 from model.wf_culprit import WfCulprit | 10 from model.wf_culprit import WfCulprit |
| 11 | 11 |
| 12 | 12 |
| 13 class WfCulpritTest(unittest.TestCase): | 13 class WfCulpritTest(unittest.TestCase): |
| 14 def testProjectName(self): | 14 def testProjectName(self): |
| 15 culprit = WfCulprit.Create('chromium', 'r1') | 15 culprit = WfCulprit.Create('chromium', 'r1', 123) |
| 16 self.assertEqual('chromium', culprit.project_name) | 16 self.assertEqual('chromium', culprit.project_name) |
| 17 | 17 |
| 18 def testCrNotificationProcessed(self): | 18 def testCrNotificationProcessed(self): |
| 19 culprit = WfCulprit.Create('chromium', 'r1') | 19 culprit = WfCulprit.Create('chromium', 'r1', 123) |
| 20 for s in (status.COMPLETED, status.RUNNING): | 20 for s in (status.COMPLETED, status.RUNNING): |
| 21 culprit.cr_notification_status = s | 21 culprit.cr_notification_status = s |
| 22 self.assertTrue(culprit.cr_notification_processed) | 22 self.assertTrue(culprit.cr_notification_processed) |
| 23 for s in (status.ERROR, None): | 23 for s in (status.ERROR, None): |
| 24 culprit.cr_notification_status = s | 24 culprit.cr_notification_status = s |
| 25 self.assertFalse(culprit.cr_notification_processed) | 25 self.assertFalse(culprit.cr_notification_processed) |
| 26 | 26 |
| 27 def testCrNotified(self): | 27 def testCrNotified(self): |
| 28 culprit = WfCulprit.Create('chromium', 'r1') | 28 culprit = WfCulprit.Create('chromium', 'r1', 123) |
| 29 culprit.cr_notification_status = status.COMPLETED | 29 culprit.cr_notification_status = status.COMPLETED |
| 30 self.assertTrue(culprit.cr_notified) | 30 self.assertTrue(culprit.cr_notified) |
| OLD | NEW |