test_server.py

import pytest

from server import *

@pytest.fixture(scope='module')
def client():
    app.config['TESTING'] = True
    print(app)

    with app.test_client() as client:
        with app.app_context():
            print('blah')
        yield client

''' Test that the root route of "/" works
    so that we know that client connected properly
    in the fixture '''
def test_root(client):
    rv = client.get('/')
    assert b'Hello World' in rv.data

''' USER ENDPOINTS '''

''' Test that when we register a new user that a string
    of the userid is returned '''
def test_register(client):
    response = client.get('/register_user', data={'name': 'delta', 
                                                  'password': '1234'})
    assert b'\"USER_EXISTS\"' not in response.data 
    assert b'\"NO_DB\"' not in response.data

''' Test that when we try to register the same user that we
    get back "USER_EXISTS" '''
def test_register_existing_user(client):
    response = client.get('/register_user', data={'name': 'delta', 
                                                  'password': '1234'})
    assert b'\"USER_EXISTS\"' in response.data 

''' Test that when we try to authenticate an existing user 
    with the correct username and password that we
    get back a string of their userid '''
def test_authenticate_user_success(client):
    response = client.get('/authenticate_user', data={'name': 'delta', 
                                                  'password': '1234'})
    assert b'\"USER_DNE\"' not in response.data 
    assert b'\"NO_DB\"' not in response.data
    assert b'\"INCORRECT_PASSWORD\"' not in response.data 

''' Test that when we try to authenticate an existing user 
    with the wrong password that we get back the string 
    of "INCORRECT PASSWORD" '''