| 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 | |
| 8 | |
| 9 from model import analysis_status as status | 7 from model import analysis_status as status |
| 10 from model.wf_culprit import WfCulprit | 8 from model.wf_culprit import WfCulprit |
| 11 | 9 |
| 12 | 10 |
| 13 class WfCulpritTest(unittest.TestCase): | 11 class WfCulpritTest(unittest.TestCase): |
| 14 def testProjectName(self): | |
| 15 culprit = WfCulprit.Create('chromium', 'r1', 123) | |
| 16 self.assertEqual('chromium', culprit.project_name) | |
| 17 | |
| 18 def testCrNotificationProcessed(self): | 12 def testCrNotificationProcessed(self): |
| 19 culprit = WfCulprit.Create('chromium', 'r1', 123) | 13 culprit = WfCulprit.Create('chromium', 'r1', 123) |
| 20 for s in (status.COMPLETED, status.RUNNING): | 14 for s in (status.COMPLETED, status.RUNNING): |
| 21 culprit.cr_notification_status = s | 15 culprit.cr_notification_status = s |
| 22 self.assertTrue(culprit.cr_notification_processed) | 16 self.assertTrue(culprit.cr_notification_processed) |
| 23 for s in (status.ERROR, None): | 17 for s in (status.ERROR, None): |
| 24 culprit.cr_notification_status = s | 18 culprit.cr_notification_status = s |
| 25 self.assertFalse(culprit.cr_notification_processed) | 19 self.assertFalse(culprit.cr_notification_processed) |
| 26 | 20 |
| 27 def testCrNotified(self): | 21 def testCrNotified(self): |
| 28 culprit = WfCulprit.Create('chromium', 'r1', 123) | 22 culprit = WfCulprit.Create('chromium', 'r1', 123) |
| 29 culprit.cr_notification_status = status.COMPLETED | 23 culprit.cr_notification_status = status.COMPLETED |
| 30 self.assertTrue(culprit.cr_notified) | 24 self.assertTrue(culprit.cr_notified) |
| OLD | NEW |