mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-02 17:50:39 +00:00
39 lines
1.0 KiB
Python
Executable File
39 lines
1.0 KiB
Python
Executable File
#!/usr/bin/python
|
|
import sys, string, json, urllib
|
|
import requests
|
|
|
|
BASE_URL="http://positive-cocoa-90213.appspot.com"
|
|
|
|
def test_time(test_name, runtime):
|
|
r = requests.post(BASE_URL + "/record/%s/%f" % (urllib.quote(test_name, safe=""), runtime))
|
|
print r.text
|
|
assert r.status_code == 204
|
|
|
|
def test_sched(test_run, shard_count, shard_id):
|
|
tests = json.dumps({'tests': string.split(sys.stdin.read())})
|
|
r = requests.post(BASE_URL + "/schedule/%s/%d/%d" % (test_run, shard_count, shard_id), data=tests)
|
|
assert r.status_code == 200
|
|
result = r.json()
|
|
for test in sorted(result['tests']):
|
|
print test
|
|
|
|
def usage():
|
|
print "%s <cmd> <args..>" % sys.argv[0]
|
|
print " time <test name> <run time>"
|
|
print " sched <test run> <num shards> <shard id>"
|
|
|
|
def main():
|
|
if len(sys.argv) < 4:
|
|
usage()
|
|
sys.exit(1)
|
|
|
|
if sys.argv[1] == "time":
|
|
test_time(sys.argv[2], float(sys.argv[3]))
|
|
elif sys.argv[1] == "sched":
|
|
test_sched(sys.argv[2], int(sys.argv[3]), int(sys.argv[4]))
|
|
else:
|
|
usage()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|