Add a bit more to the python test driver
BUG=skia:
GOLD_TRYBOT_URL= https://gold.skia.org/search2?unt=true&query=source_type%3Dgm&master=false&issue=1773263002
Review URL: https://codereview.chromium.org/1773263002
diff --git a/tools/skiaserve/tester/tester.py b/tools/skiaserve/tester/tester.py
index b6c04bd..b99529c 100644
--- a/tools/skiaserve/tester/tester.py
+++ b/tools/skiaserve/tester/tester.py
@@ -12,12 +12,13 @@
from os import listdir
from os.path import isfile, join
-ops = [
+default_ops = [
"enable_gpu",
"post",
"info",
"cmd",
- "img"
+ "img",
+ "batchList"
]
def Check(request):
@@ -80,17 +81,23 @@
r = self.cmd()
return len(r.json()['commands']) - 1 # why the minus 1 here?
+ def batchList(self):
+ path = self.output_dir + '/' + self.skp_name + '.batches.json'
+ return WriteJson(Check(requests.get(self.url + '/batches')), path)
+
def main():
parser = argparse.ArgumentParser(description='Tester for SkiaServe')
parser.add_argument('--skp_dir', default='skps', type=str)
parser.add_argument('--url', default='http://localhost:8888', type=str)
parser.add_argument('--output_dir', default='results', type=str)
parser.add_argument('--match', default='.*', type=str)
+ parser.add_argument('--ops', nargs='+', default=default_ops)
args = parser.parse_args()
skp_dir = args.skp_dir
url = args.url
output_dir = args.output_dir
+ ops = args.ops
if not os.path.isdir(output_dir):
os.makedirs(output_dir)