Falcon, wsgiref: модульные тесты - PullRequest
Новая
       38

Falcon, wsgiref: модульные тесты

0 голосов
/ 25 октября 2019

У меня есть следующий код: 

master.py    

def create():
    master = falcon.API(middleware=Auth())
    msg = Message()
    master.add_route('/message', msg)


master = create()

if __name__ == '__main__':
    httpd = simple_server.make_server("127.0.0.1", 8989, master)
    process = Thread(target=httpd.serve_forever, name="master_process")
    process.start()

    #Some logic happens here
    s_process = Thread(target=httpd.shutdown(), name="shut_process")
    s_process.start()
    s.join()

Я попытался создать приведенный ниже тестовый пример для следующего: 

from falcon import testing
from master import create

@pytest.fixture(scope='module')
def client():
   return testing.TestClient(create())

def test_post_message(client):
   result = client.simulate_post('/message', headers={'token': "ubxuybcwe"}, body='{"message": "I'm here!"}') --> This line throws the error
   assert result.status_code == 200

Я попытался запустить выше, но получаю следующую ошибку:

TypeError: 'NoneType' object is not callable

Я на самом деле не могу понять, как мне написать сценарий для этого.

1 Ответ

0 голосов
/ 26 октября 2019

Исходя из того, что сказал @hoefling, нижеследующее исправило это: 

master.py    

def create():
     master = falcon.API(middleware=Auth())
     msg = Message()
     master.add_route('/message', msg)
     return master


master = create()


if __name__ == '__main__':
    httpd = simple_server.make_server("127.0.0.1", 8989, master)
    process = Thread(target=httpd.serve_forever, name="master_process")
    process.start()

    #Some logic happens here
    s_process = Thread(target=httpd.shutdown(), name="shut_process")
    s_process.start()
    s.join()

И тогда тестовый пример работает: 

from falcon import testing
from master import create

@pytest.fixture(scope='module')
def client():
    return testing.TestClient(create())

def test_post_message(client):
    result = client.simulate_post('/message', headers={'token': "ubxuybcwe"}, 
    body='{"message": "I'm here!"}') 
    assert result.status_code == 200

Большое спасибо @hoefling!

...