Skip to main content

Hypothesis strategies for Open API / Swagger schemas

Project description

Build codecov.io status for master branch Version Python versions License

Schemathesis is a tool for testing your web applications built with Open API / Swagger specifications.

It reads the application schema and generates test cases which will ensure that your application is compliant with its schema.

The application under test could be written in any language, the only thing you need is a valid API schema in a supported format.

Supported specification versions:

  • Swagger 2.0

  • Open API 3.0.x

More API specifications will be added in the future.

Built with:

Inspired by wonderful swagger-conformance project.

Installation

To install Schemathesis via pip run the following command:

pip install schemathesis

Gitter: https://gitter.im/kiwicom/schemathesis

Usage

There are two basic ways to use Schemathesis:

CLI is pretty simple to use and requires no coding, in-code approach gives more flexibility.

Command Line Interface

The schemathesis command can be used to perform Schemathesis test cases:

schemathesis run https://example.com/api/swagger.json
https://github.com/kiwicom/schemathesis/blob/master/img/schemathesis.gif

If your application requires authorization then you can use --auth option for Basic Auth and --header to specify custom headers to be sent with each request.

To filter your tests by endpoint name, HTTP method or Open API tags you could use -E, -M, -T options respectively.

CLI supports passing options to hypothesis.settings. All of them are prefixed with --hypothesis-:

schemathesis run --hypothesis-max-examples=1000 https://example.com/api/swagger.json

For the full list of options, run:

schemathesis --help

Docker

Schemathesis CLI also available as a docker image

docker run kiwicom/schemathesis:latest run http://example.com/schema.json

To run it against localhost server add --network=host parameter:

docker run --network="host" kiwicom/schemathesis:latest run http://127.0.0.1/schema.json

Pre-run CLI hook

Sometimes you need to execute custom code before the CLI run, for example setup an environment, register custom string format strategies or modify Schemathesis behavior in runtime you can use --pre-run hook:

schemathesis --pre-run importable.path.to.module run https://example.com/api/swagger.json

NOTE. This option should be passed before the run part.

The passed value will be processed as an importable Python path, where you can execute your code. An example - https://github.com/kiwicom/schemathesis#custom-string-strategies

Registering custom checks for CLI

To add a new check for the Schemathesis CLI there is a special function

import schemathesis

@schemathesis.register_check
def new_check(response, result):
    # some awesome assertions!
    pass

The registered check should accept a response with requests.Response type and result with schemathesis.models.TestResult type.

After registration, your checks will be available in Schemathesis CLI and you can use them via -c command line option.

schemathesis --pre-run module.with.checks run -c new_check https://example.com/api/swagger.json

In-code

To examine your application with Schemathesis you need to:

  • Setup & run your application, so it is accessible via the network;

  • Write a couple of tests in Python;

  • Run the tests via pytest.

Suppose you have your application running on http://0.0.0.0:8080 and its schema is available at http://0.0.0.0:8080/swagger.json.

A basic test, that will verify that any data, that fit into the schema will not cause any internal server error could look like this:

# test_api.py
import requests
import schemathesis

schema = schemathesis.from_uri("http://0.0.0.0:8080/swagger.json")

@schema.parametrize()
def test_no_server_errors(case):
    # `requests` will make an appropriate call under the hood
    response = case.call()
    assert response.status_code < 500

It consists of four main parts:

  1. Schema preparation; schemathesis package provides multiple ways to initialize the schema - from_path, from_dict, from_uri, from_file.

  2. Test parametrization; @schema.parametrize() generates separate tests for all endpoint/method combination available in the schema.

  3. A network call to the running application; case.call does it.

  4. Verifying a property you’d like to test; In the example, we verify that any app response will not indicate a server-side error (HTTP codes 5xx).

Run the tests:

pytest test_api.py

Other properties that could be tested:

  • Any call will be processed in <50 ms - you can verify the app performance;

  • Any unauthorized access will end with 401 HTTP response code;

Each test function should have the case fixture, that represents a single test case.

Important Case attributes:

  • method - HTTP method

  • formatted_path - full endpoint path

  • headers - HTTP headers

  • query - query parameters

  • body - request body

You can use them manually in network calls or can convert to a dictionary acceptable by requests.request:

import requests

schema = schemathesis.from_uri("http://0.0.0.0:8080/swagger.json")

@schema.parametrize()
def test_no_server_errors(case):
    kwargs = case.as_requests_kwargs()
    response = requests.request(**kwargs)

For each test, Schemathesis will generate a bunch of random inputs acceptable by the schema. This data could be used to verify that your application works in the way as described in the schema or that schema describes expected behavior.

By default, there will be 100 test cases per endpoint/method combination. To limit the number of examples you could use hypothesis.settings decorator on your test functions:

from hypothesis import settings

@schema.parametrize()
@settings(max_examples=5)
def test_something(client, case):
    ...

To narrow down the scope of the schemathesis tests it is possible to filter by method or endpoint:

@schema.parametrize(method="GET", endpoint="/pet")
def test_no_server_errors(case):
    ...

The acceptable values are regexps or list of regexps (matched with re.search).

Explicit examples

If the schema contains parameters examples, then they will be additionally included in the generated cases.

paths:
  get:
    parameters:
    - in: body
      name: body
      required: true
      schema: '#/definitions/Pet'

definitions:
  Pet:
    additionalProperties: false
    example:
      name: Doggo
    properties:
      name:
        type: string
    required:
    - name
    type: object

With this Swagger schema example, there will be a case with body {"name": "Doggo"}. Examples handled with example decorator from Hypothesis, more info about its behavior is here.

NOTE. Schemathesis supports only examples in parameters at the moment, examples of individual properties are not supported.

Direct strategies access

For convenience you can explore the schemas and strategies manually:

>>> import schemathesis
>>> schema = schemathesis.from_uri("http://0.0.0.0:8080/petstore.json")
>>> endpoint = schema["/v2/pet"]["POST"]
>>> strategy = endpoint.as_strategy()
>>> strategy.example()
Case(
    path='/v2/pet',
    method='POST',
    path_parameters={},
    headers={},
    cookies={},
    query={},
    body={
        'name': '\x15.\x13\U0008f42a',
        'photoUrls': ['\x08\U0009f29a', '\U000abfd6\U000427c4', '']
    },
    form_data={}
)

Schema instances implement Mapping protocol.

Lazy loading

If you have a schema that is not available when the tests are collected, for example it is build with tools like apispec and requires an application instance available, then you can parametrize the tests from a pytest fixture.

# test_api.py
import schemathesis

schema = schemathesis.from_pytest_fixture("fixture_name")

@schema.parametrize()
def test_api(case):
    ...

In this case the test body will be used as a sub-test via pytest-subtests library.

NOTE: the used fixture should return a valid schema that could be created via schemathesis.from_dict or other schemathesis.from_ variations.

Extending schemathesis

If you’re looking for a way to extend schemathesis or reuse it in your own application, then runner module might be helpful for you. It can run tests against the given schema URI and will do some simple checks for you.

from schemathesis import runner

runner.execute("http://127.0.0.1:8080/swagger.json")

The built-in checks list includes the following:

  • Not a server error. Asserts that response’s status code is less than 500;

  • Status code conformance. Asserts that response’s status code is listed in the schema;

  • Content type conformance. Asserts that response’s content type is listed in the schema;

  • Response schema conformance. Asserts that response’s content conforms to the declared schema;

You can provide your custom checks to the execute function, the check is a callable that accepts one argument of requests.Response type.

from datetime import timedelta
from schemathesis import runner, models

def not_too_long(response, result: models.TestResult):
    assert response.elapsed < timedelta(milliseconds=300)

runner.execute("http://127.0.0.1:8080/swagger.json", checks=[not_too_long])

Custom string strategies

Some string fields could use custom format and validators, e.g. card_number and Luhn algorithm validator.

For such cases it is possible to register custom strategies:

  1. Create hypothesis.strategies.SearchStrategy object

  2. Optionally provide predicate function to filter values

  3. Register it via schemathesis.register_string_format

strategy = strategies.from_regex(r"\A4[0-9]{15}\Z").filter(luhn_validator)
schemathesis.register_string_format("visa_cards", strategy)

Unittest support

Schemathesis supports Python’s built-in unittest framework out of the box, you only need to specify strategies for hypothesis.given:

from unittest import TestCase
from hypothesis import given
import schemathesis

schema = schemathesis.from_uri("http://0.0.0.0:8080/petstore.json")
new_pet_strategy = schema["/v2/pet"]["POST"].as_strategy()

class TestSchema(TestCase):

    @given(case=new_pet_strategy)
    def test_pets(self, case):
        response = case.call()
        assert response.status_code < 500

Documentation

For the full documentation, please see https://schemathesis.readthedocs.io/en/latest/ (WIP)

Or you can look at the docs/ directory in the repository.

Local development

First, you need to prepare a virtual environment with poetry. Install poetry (check out the installation guide) and run this command inside the project root:

poetry install

For simpler local development Schemathesis includes a aiohttp-based server with 3 endpoints in Swagger 2.0 schema:

  • /api/success - always returns {"success": true}

  • /api/failure - always returns 500

  • /api/slow - always returns {"slow": true} after 250 ms delay

  • /api/unsatisfiable - parameters for this endpoint are impossible to generate

  • /api/invalid - invalid parameter definition. Uses int instead of integer

  • /api/flaky - returns 1/1 ratio of 200/500 responses

  • /api/multipart - accepts multipart data

  • /api/teapot - returns 418 status code, that is not listed in the schema

  • /api/text - returns plain/text responses, which are not declared in the schema

  • /api/malformed_json - returns malformed JSON with application/json content type header

To start the server:

./test_server.sh 8081

It is possible to configure available endpoints via --endpoints option. The value is expected to be a comma separated string with endpoint names (success, failure or slow):

./test_server.sh 8081 --endpoints=success,slow

Then you could use CLI against this server:

schemathesis run http://127.0.0.1:8081/swagger.yaml
================================== Schemathesis test session starts =================================
platform Linux -- Python 3.7.4, schemathesis-0.12.2, hypothesis-4.39.0, hypothesis_jsonschema-0.9.8
rootdir: /
hypothesis profile 'default' -> database=DirectoryBasedExampleDatabase('/.hypothesis/examples')
Schema location: http://127.0.0.1:8081/swagger.yaml
Base URL: http://127.0.0.1:8081
Specification version: Swagger 2.0
collected endpoints: 2

GET /api/slow .                                                                               [ 50%]
GET /api/success .                                                                            [100%]

============================================== SUMMARY ==============================================

not_a_server_error            2 / 2 passed          PASSED

========================================= 2 passed in 0.29s =========================================

Running tests

You could run tests via tox:

tox -p all -o

or pytest in your current environment:

pytest test/ -n auto

Contributing

Any contribution in development, testing or any other area is highly appreciated and useful to the project.

Please, see the CONTRIBUTING.rst file for more details.

Python support

Schemathesis supports Python 3.6, 3.7 and 3.8.

License

The code in this project is licensed under MIT license. By contributing to schemathesis, you agree that your contributions will be licensed under its MIT license.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

schemathesis-0.15.0-py3-none-any.whl (34.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page