def test_status_raises_not_enough_data(delimiter, mocker): p = Parser( delimiter=delimiter, status_parser=mocker.stub(), header_parser=mocker.stub(), body_parser=mocker.stub() ) with pytest.raises(NotEnoughDataError): p.status()
def test_status_transitions_to_header_state(delimiter, mocker): p = Parser( delimiter=delimiter, status_parser=mocker.Mock(return_value={'status': 'status success'}), header_parser=mocker.stub(), body_parser=mocker.stub() ) p.buffer = b'left\r\nright' p.status() assert p.result['status'] == 'status success' assert p.state == States.Header