def testStatsDebugMsg(self, golo=False): """We hide debug messages from stats module when not in golo.""" stats.StatsUploader._Upload.side_effect = EnvironmentError() with cros_test_lib.LoggingCapturer() as logs: upload_command_stats.main(self.argv) self.AssertLogsContain( logs, stats.StatsUploader.ENVIRONMENT_ERROR, inverted=(not golo))
def testStatsDebugMsg(self, golo=False): """We hide debug messages from stats module when not in golo.""" stats.StatsUploader._Upload.side_effect = EnvironmentError() with cros_test_lib.LoggingCapturer() as logs: upload_command_stats.main(self.argv) self.AssertLogsContain(logs, stats.StatsUploader.ENVIRONMENT_ERROR, inverted=(not golo))
def testGoloRun(self): """Test when running in the golo.""" cros_build_lib.GetHostDomain.return_value = constants.GOLO_DOMAIN upload_command_stats.main(self.argv) self.assertEquals(stats.StatsUploader._Upload.call_count, 1) self.testStatsDebugMsg(golo=True)
def testNormalRun(self): """Going for code coverage.""" upload_command_stats.main(self.argv) self.assertEquals(stats.StatsUploader._Upload.call_count, 1)
def testGoloRun(self): """Test when running in the golo.""" cros_build_lib.GetHostDomain.return_value = constants.GOLO_DOMAIN upload_command_stats.main(self.argv) self.assertEquals(stats.StatsUploader._Upload.call_count, 1) self.testStatsDebugMsg(golo=True)
def testNormalRun(self): """Going for code coverage.""" upload_command_stats.main(self.argv) self.assertEquals(stats.StatsUploader._Upload.call_count, 1)