Lodge It

import unittest
import re
import os

from werkzeug import BaseResponse, Client, Request

import kay
from kay.app import get_application
from kay.auth.models import GoogleUser

class MyTestCase(unittest.TestCase):

  def setUp(self):
    app = get_application()
    self.client = Client(app, BaseResponse)
    self.email = 'test@example.com'
    self.userid = '123'
    self.user = GoogleUser(key_name='_%s' % self.userid,
                           email=self.email, first_name='Given',
                           last_name='Family', display_name='Given Family')
    self.user.put()

  def tearDown(self):
    pass

  def test_success(self):
    """Test with normal CapabilityServiceStub"""
    response = self.client.get('/')
    os.environ['USER_EMAIL'] = ''
    os.environ['USER_ID'] = ''
    self.assertEqual(response.status_code, 302)
    os.environ['USER_EMAIL'] = self.email
    os.environ['USER_ID'] = self.userid
    response = self.client.get('/')
    self.assertEqual(response.status_code, 200)
    users = GoogleUser.all().fetch(1000)
    self.assertEqual(len(users), 1)