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)
