tests_wtf.py

import http
import os
import json
import pytest
from wtf import APP

ROUTE = '/slack'
TEST_TOKENS = ['token1', 'token2']

@pytest.fixture
def client():
    client = APP.test_client()
    APP.config['TESTING'] = True
    APP.config['SLACK_TOKENS'] = TEST_TOKENS
    yield client

def test_env_vars_present():
    for var in ['SLACK_TOKENS', 'DATA_URL']:
        assert os.getenv(var) != None

@pytest.mark.parametrize("token", TEST_TOKENS)
def test_good_payload_using_valid_token(client, token):
    data = {'text': 'vba','token': token}
    r = client.post(ROUTE, data=data)
    assert b'Veterans Benefits Administration' in r.data
    assert r.status_code == http.HTTPStatus.OK

def test_multi_def_payload(client):
    data = {'text': 'aaa','token': TEST_TOKENS[0]}
    r = client.post(ROUTE, data=data)
    assert b'Abdominal Aortic Aneurysm; or Area Agencies on Aging;' in r.data

def test_comma_in_def(client):
    data = {'text': 'acre','token': TEST_TOKENS[0]}
    r = client.post(ROUTE, data=data)
    assert b'A measure of land 43,560 sq. ft.' in r.data

def test_bad_payload(client):
    data = {'foo': 'bar', 'token': TEST_TOKENS[0]}
    r = client.post(ROUTE, data=data)
    assert b'Improper request' in r.data
    assert r.status_code == http.HTTPStatus.BAD_REQUEST

def test_no_slack_token(client):
    data = {'text': 'vba','token': 'foobar'}
    r = client.post(ROUTE, data=data)
    assert b'Not authorized' in r.data
    assert r.status_code == http.HTTPStatus.UNAUTHORIZED

def test_not_found(client):
    data = {'text': '13231312334','token': TEST_TOKENS[0]}
    r = client.post(ROUTE, data=data)
    assert b'not found!' in r.data
    assert b'13231312334' in r.data
    assert r.status_code == http.HTTPStatus.OK