def testGsdUrlNoMatch(self): """Verify None returned when MatchUrl() returns None.""" with open('testdata/test_page_many_links.html', 'r') as test_page: test_result = CommandResult() test_result.output = test_page.read() test_result.returncode = 0 cb_url_lib.RunCommand(['gsutil', 'ls', self.gsd_url], redirect_stdout=True, redirect_stderr=True).AndReturn(test_result) self.mox.ReplayAll() actual = cb_url_lib.DetermineUrl(self.http_url, self.gsd_pattern) self.assertEqual(None, actual)
def testGsdUrlGood(self): """Verify GSD URL returned when page opens properly.""" with open('testdata/test_page_gsd_links.html', 'r') as test_page: test_result = CommandResult() test_result.output = test_page.read() test_result.returncode = 0 cb_url_lib.RunCommand(['gsutil', 'ls', self.gsd_url], redirect_stdout=True, redirect_stderr=True).AndReturn(test_result) self.mox.ReplayAll() expected = ( 'gs://chromeos-releases/stable-channel/x86-alex/0.12.433.269/' 'chromeos_0.12.433.269_x86-alex_recovery_stable-channel_mp-v3.bin') actual = cb_url_lib.DetermineUrl(self.http_url, self.gsd_pattern) self.assertEqual(expected, actual)