Stripped down Users Test file

gistfile1.py

from application import db
from app.models import User
from common.tests import BaseTestCase
from common.util import rand_string_gen

class UsersTest(BaseTestCase):

  def setUp(self):
    with self.app.app_context():
      db.create_all()

  def tearDown(self):
    with self.app.app_context():
      db.session.remove()
      db.drop_all()

  @staticmethod
  def create_user(username, password):
    """Creates a user with hashed password in the database

    :return: user object
    """
    user = User(username=username)
    user.hash_password(password)
    db.session.add(user)
    db.session.commit()
    return user

  @staticmethod
  def generate_test_users(n=1):
    """Returns a list of n test users
    """
    rv = list()
    for i in xrange(0, n):
      rv.append({'username': rand_string_gen(), 'password': rand_string_gen()})
    return rv

  @staticmethod
  def create_basic_auth_header(username, password):
    """
    :return: Basic auth header entry
    """
    return {
      'Authorization': 'Basic %s' % b64encode("{0}:{1}".format(username, password))
    }

  ### Tests ###
  def test_get_all_users(self):
    num_users = 5
    users = UsersTest.generate_test_users(num_users)
    for user in users:
      UsersTest.create_user(user['username'], user['password'])

    auth_headers = UsersTest.create_basic_auth_header(
      username=users[0]['username'],
      password=users[0]['password']
    )
    response = self.client.get(
      '/api/v0/users',
      headers=auth_headers
    )
    self.assertStatus(response, 200)
    data = json.loads(response.data)
    users = data['data']
    self.assertEqual(len(users), num_users)

  def test_get_user_by_id(self):
    # create user
    test_username = 'test_user'
    test_password = 'test_password'
    test_user = UsersTest.create_user(
      username=test_username,
      password=test_password
    )

    auth_headers = UsersTest.create_basic_auth_header(
      username=test_username,
      password=test_password
    )
    response = self.client.get(
      '/api/v0/users/%s' % (test_user.id,),
      headers=auth_headers
    )
    self.assertEqual(response.status_code, 200)
    user = json.loads(response.data)
    self.assertEqual(user['username'], test_username)