Nosegae单元在Python Google App Engine项目上进行gae-session测试

时间:2011-03-18 06:44:23

标签: python google-app-engine cookies webtest nosetests

当我使用单元测试时,我无法使gae-sessions工作。我不确定这是一个cookie问题还是内部gae-sessions问题(它使用memcache /数据存储区来持久)。当通过浏览器访问时,它在实际应用程序上运行良好。

我在Mac OS 10.6.6,Google App Engine SDK 1.4.2,nosetests版本1.0.0,NoseGAE-0.1.7和gae-sessions v1.06上使用Python 2.5。

这是我的代码:


网络应用实施:

from google.appengine.ext import webapp
from gaesessions import get_current_session

class cookietest(webapp.RequestHandler):
    def get(self):
        s = get_current_session()
        if not 'message' in s:
            self.response.out.write('No session value')
            s['message'] = 'This value is stored in a session'
        else:
            self.response.out.write("Message found: %s" % s['message'])

application = webapp.WSGIApplication([(r'/cookietest', cookietest)])

单元测试实施:

def test_cookietest(self):
    from gaesessions import SessionMiddleware
    app_with_sessions = SessionMiddleware(application, cookie_key=os.urandom(64)) # random key
    self.app = TestApp(app_with_sessions)

    resp = self.app.get('/cookietest')
    cookie = resp.headers['Set-Cookie']
    assert cookie
    resp = self.app.get('/cookietest', headers={'Cookie': cookie})
    print resp
    assert 'This value is stored in a session' in resp

相关输出是:

FAIL: test_cookietest (tests.test.TestUser)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/Asad/.../test.py", line 33, in test_cookietest
    assert 'This value is stored in a session' in resp
AssertionError: 
-------------------- >> begin captured stdout << ---------------------
Response: 200 OK
Cache-Control: no-cache
Content-Type: text/html; charset=utf-8
Expires: Fri, 01 Jan 1990 00:00:00 GMT
Set-Cookie:  DgU00="DQjpTdhQsKYYGhvNr/CLm+v835QV5V1jTX9T4wMULhI=1301034363_89b1c1007a6159e85828ab9a7673759cgAJ9cQB9cQFVB21lc3NhZ2VxAlUhVGhpcyB2YWx1ZSBpcyBzdG9yZWQgaW4gYSBzZXNzaW9ucQNzhnEELg=="; expires=Fri, 25-Mar-2011 06:26:03 GMT; Path=/; HttpOnly
No session value

--------------------- >> end captured stdout << ----------------------
-------------------- >> begin captured logging << --------------------
root: WARNING: Could not read datastore data from /var/folders/pt/ptPaW5O7E-eLIrfFhhczbU+++TI/-Tmp-/nosegae.datastore
root: INFO: zipimporter('/Library/Python/2.5/site-packages/NoseGAE-0.1.7-py2.5.egg', '')
root: INFO: zipimporter('/Library/Python/2.5/site-packages/WebTest-1.2.3-py2.5.egg', '')
root: INFO: zipimporter('/Library/Python/2.5/site-packages/WebOb-1.0-py2.5.egg', '')
root: INFO: zipimporter('/Library/Python/2.5/site-packages/mock-0.7.0b4-py2.5.egg', '')
root: DEBUG: Could not import "strop": Disallowed C-extension or built-in module
--------------------- >> end captured logging << ---------------------

----------------------------------------------------------------------
Ran 10 tests in 0.284s

FAILED (failures=1)

1 个答案:

答案 0 :(得分:4)

我认为您必须在此测试中手动保存会话。

   if not 'message' in s:
        self.response.out.write('No session value')
        s['message'] = 'This value is stored in a session'
        s.save()

更新

gae-sessionsos.environ['HTTP_COOKIE']变量中检索会话Cookie。需要在self.app.get请求之前设置此值,而不是在headers参数中传入Cookie。