def testCheckModelsStatusNoModels(self, containsIssueMock, addIssueMock, removeIssueMock, sendEmailMock): url = "foo.bar" testJson = [] emailParams = {} # Code under test modelsMonitor._checkModelsStatus(testJson, url, emailParams) self.assertFalse(containsIssueMock.called) self.assertFalse(addIssueMock.called) self.assertFalse(removeIssueMock.called) self.assertFalse(sendEmailMock.called)
def testCheckModelsStatusNoModels(self, containsIssueMock, addIssueMock, removeIssueMock, sendEmailMock): url = "foo.bar" testJson = [] emailParams = {} # Code under test modelsMonitor._checkModelsStatus(testJson, url, emailParams) self.assertFalse(containsIssueMock.called) self.assertFalse(addIssueMock.called) self.assertFalse(removeIssueMock.called) self.assertFalse(sendEmailMock.called)
def testCheckModelsStatusGoodModels(self, containsIssueMock, addIssueMock, removeIssueMock, sendEmailMock): url = "foo.bar" testJson = [{u"uid": "123", u"status": MetricStatus.ACTIVE}, {u"uid": "456", u"status": MetricStatus.ACTIVE}] emailParams = {} # Code under test modelsMonitor._checkModelsStatus(testJson, url, emailParams) removeIssueMock.assert_any_call(ANY, "123") removeIssueMock.assert_any_call(ANY, "456") self.assertFalse(containsIssueMock.called) self.assertFalse(addIssueMock.called) self.assertFalse(sendEmailMock.called)
def testCheckModelsStatusErrorModels_AlreadyReported(self, containsIssueMock, addIssueMock, removeIssueMock, sendEmailMock): url = "foo.bar" testJson = [{u"uid": "123", u"status": MetricStatus.ERROR}, {u"uid": "456", u"status": MetricStatus.ERROR}] containsIssueMock.return_value = True emailParams = {} # Code under test modelsMonitor._checkModelsStatus(testJson, url, emailParams) containsIssueMock.assert_any_call(ANY, "123") containsIssueMock.assert_any_call(ANY, "456") self.assertFalse(removeIssueMock.called) self.assertFalse(addIssueMock.called) self.assertFalse(sendEmailMock.called)