Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(2394)

Unified Diff: recipe_modules/url/example.py

Issue 2868333004: Add URL recipe module from "depot_tools". (Closed)
Patch Set: response api object Created 3 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: recipe_modules/url/example.py
diff --git a/recipe_modules/url/example.py b/recipe_modules/url/example.py
new file mode 100644
index 0000000000000000000000000000000000000000..c546aff91b0f5494bbfba6233a79d75c52d92143
--- /dev/null
+++ b/recipe_modules/url/example.py
@@ -0,0 +1,78 @@
+# Copyright 2017 The LUCI Authors. All rights reserved.
+# Use of this source code is governed under the Apache License, Version 2.0
+# that can be found in the LICENSE file.
+
+DEPS = [
+ 'path',
+ 'step',
+ 'url',
+]
+
+def RunSteps(api):
+ api.step('step1',
+ ['/bin/echo', api.url.join('foo', 'bar', 'baz')])
+ api.step('step2',
+ ['/bin/echo', api.url.join('foo/', '/bar/', '/baz')])
+ api.step('step3',
+ ['/bin/echo', api.url.join('//foo/', '//bar//', '//baz//')])
+ api.step('step4',
+ ['/bin/echo', api.url.join('//foo/bar//', '//baz//')])
+
+ # get_file
+ api.url.must_get_file('fake://foo/bar', api.path['start_dir'],
+ headers={'Authorization': 'thing'})
+
+ raised = False
+ try:
+ api.url.must_get_file('fake://foo/bar/error', api.path['start_dir'])
+ except ValueError:
+ raised = True
+ assert raised
+
+ # get_text
+ v = api.url.must_get_text('fake://foo/bar/text')
+ assert v.method == 'GET'
+ assert v.output == '<head>RESULT!</head>'
+ assert v.status_code == 200
+
+ raised = False
+ try:
+ api.url.must_get_text('fake://foo/bar/text/error')
+ except ValueError:
+ raised = True
+ assert raised
+
+ v = api.url.get_text('fake://foo/bar/text/error2')
+ assert v.status_code == 403
+
+ v = api.url.get_text('fake://foo/bar/text (w/ auth)',
+ headers={'Authorization': 'thing'}, transient_retry=False)
+ assert v.output == '<head>AUTH!</head>'
+ assert v.status_code == 200
+
+ # get_json
+ v = api.url.must_get_json('fake://foo/bar/json', log=True,
+ strip_prefix=api.url.GERRIT_JSON_PREFIX)
+ assert v.output == {'foo': 'bar', 'baz': 'qux'}
+ assert v.status_code == 200
+
+ raised = False
+ try:
+ api.url.must_get_json('fake://foo/bar/json/error')
+ except ValueError:
+ raised = True
+ assert raised
+
+
+def GenTests(api):
+ yield (
+ api.test('basic') +
+ api.url.text('GET fake://foo/bar', 'ohai') +
+ api.url.text('GET fake://foo/bar/text', '<head>RESULT!</head>') +
+ api.url.error('GET fake://foo/bar/error', 403) +
+ api.url.text('GET fake://foo/bar/text (w/ auth)', '<head>AUTH!</head>') +
+ api.url.error('GET fake://foo/bar/text/error', 502) +
+ api.url.error('GET fake://foo/bar/text/error2', 403) +
+ api.url.json('GET fake://foo/bar/json', {'foo': 'bar', 'baz': 'qux'}) +
+ api.url.error('GET fake://foo/bar/json/error', 401)
+ )

Powered by Google App Engine
This is Rietveld 408576698