A CLI tool that creates locust.io tasks file (locustfile.py) from Swagger/OpenAPI Spec.
- Node JS (4.0 & up)
Best option is:
$ git clone https://github.com/DataGreed/swagger-to-locustfile.git$ cd swagger-to-locustfile$ npm -g install- You are good to go.
- Grabbing the host field for the spec.
- Generating tasks for GET/POST/PUT/UPDATE/DELETE endpoints.
- Replacing Path-Parameters holders with their default value.
- Appending the required query string parameters and their default values.
- Swagger/OAI vendor extensions (
x-locust-importandx-locust-value) to override default values with custom python expressions. - Command Line Options allow overriding
min_wait,min_maxandhost
- Extracting body/params for POST/PUT/DELETE/UPDATE endpoints.
- Support required headers
Basically:
$ swagger2locust /path/to/swagger.json > /tmp/locustfile.py
Or you can pipe the spec:
$ cat /path/to/spec.yaml | swagger2locust
Full Usage:
Usage: swagger2locust [options] <file>
Options:
-h, --help output usage information
-V, --version output the version number
-m, --min <n> minimum time, in milliseconds, that a simulated user will wait between executing each task (default: 1000).
-x, --max <n> maximum time, in milliseconds, that a simulated user will wait between executing each task (default: 3000).
-H, --host <host> The host attribute is a URL prefix (i.e. “http://google.com”) to the host that is to be loaded.
You can add x-locust-import to your root node (same level as host field) to specify extra imports for your locust file. This is useful if you need access to an import when you write an expression in x-locust-value.
So for the following swagger spec:
| JSON | YAML |
{
"swagger" : "2.0",
...
"host" : "subdomain.domain.tld",
"x-locust-import" : ["time"],
...
}
| swagger: "2.0" ... host: "subdomain.domain.tld" x-locust-import: - "time" ... |
from locust import HttpLocust, TaskSet, task
import time
class MyTaskSet(TaskSet):
...x-locust-value allows you to write a python expression that replaces the hardcoded
default value of a field.
Example:
The following spec file
| JSON | YAML |
...
"paths" : {
"/required/qs/params-with-x-locust-value" : {
"get" : {
"parameters" : [ {
"name" : "sreq_timestamp_param",
"in" : "query",
"description" : "Some timestamp.",
"required" : true,
"type" : "number",
"default" : 1455134652,
"x-locust-value" : "str(int(time.time()))"
} ]
}
}
}
...
| ...
paths:
/required/qs/params-with-x-locust-value:
get:
parameters: -
name: "req_timestamp_param"
in: "query"
description: "Some timestamp."
required: true
type: "number"
default: 1455134652
x-locust-value: "str(int(time.time()))"
...
|
Will result in the following line in locustfile.py (x-locust-value
overrrides default):
...
@task
def get_required_qs_params_with_x_locust_value(self):
self.client.get("/api/v1/required/qs/params-with-x-locust-value?some_required_timestamp_param={0}".format(str(int(time.time()))))
...While omitting x-locust-value field will result in the following line:
...
@task
def get_required_qs_params_without_x_locust_value(self):
self.client.get("/api/v1/required/qs/params-with-x-locust-value?some_required_timestamp_param={0}".format("1455134652"))
...- fork
- create a branch on your fork.
- pull-request to master branch here.
- win.
- @DataGreed (PR #13)
Copyright (c) 2016, Liel Dulev
Permission to use, copy, modify, and/or distribute this software for any purpose with or without fee is hereby granted, provided that the above copyright notice and this permission notice appear in all copies.
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.