| 1 |
""" |
|---|
| 2 |
Copyright (c) 2004, CherryPy Team (team@cherrypy.org) |
|---|
| 3 |
All rights reserved. |
|---|
| 4 |
|
|---|
| 5 |
Redistribution and use in source and binary forms, with or without modification, |
|---|
| 6 |
are permitted provided that the following conditions are met: |
|---|
| 7 |
|
|---|
| 8 |
* Redistributions of source code must retain the above copyright notice, |
|---|
| 9 |
this list of conditions and the following disclaimer. |
|---|
| 10 |
* Redistributions in binary form must reproduce the above copyright notice, |
|---|
| 11 |
this list of conditions and the following disclaimer in the documentation |
|---|
| 12 |
and/or other materials provided with the distribution. |
|---|
| 13 |
* Neither the name of the CherryPy Team nor the names of its contributors |
|---|
| 14 |
may be used to endorse or promote products derived from this software |
|---|
| 15 |
without specific prior written permission. |
|---|
| 16 |
|
|---|
| 17 |
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
|---|
| 18 |
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
|---|
| 19 |
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
|---|
| 20 |
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE |
|---|
| 21 |
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
|---|
| 22 |
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
|---|
| 23 |
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
|---|
| 24 |
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
|---|
| 25 |
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|---|
| 26 |
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|---|
| 27 |
""" |
|---|
| 28 |
|
|---|
| 29 |
import cherrypy, time, os |
|---|
| 30 |
|
|---|
| 31 |
class Root: |
|---|
| 32 |
def testGen(self): |
|---|
| 33 |
counter = cherrypy.session.get('counter', 0) + 1 |
|---|
| 34 |
cherrypy.session['counter'] = counter |
|---|
| 35 |
yield str(counter) |
|---|
| 36 |
testGen.exposed = True |
|---|
| 37 |
def testStr(self): |
|---|
| 38 |
counter = cherrypy.session.get('counter', 0) + 1 |
|---|
| 39 |
cherrypy.session['counter'] = counter |
|---|
| 40 |
return str(counter) |
|---|
| 41 |
testStr.exposed = True |
|---|
| 42 |
|
|---|
| 43 |
cherrypy.root = Root() |
|---|
| 44 |
cherrypy.config.update({ |
|---|
| 45 |
'server.logToScreen': False, |
|---|
| 46 |
'server.environment': 'production', |
|---|
| 47 |
'sessionFilter.on': True, |
|---|
| 48 |
'sessionFilter.storageType' : 'file', |
|---|
| 49 |
'sessionFilter.storagePath' : '.', |
|---|
| 50 |
}) |
|---|
| 51 |
|
|---|
| 52 |
import helper |
|---|
| 53 |
|
|---|
| 54 |
class SessionFilterTest(helper.CPWebCase): |
|---|
| 55 |
|
|---|
| 56 |
def testSessionFilter(self): |
|---|
| 57 |
self.getPage('/testStr') |
|---|
| 58 |
self.assertBody('1') |
|---|
| 59 |
self.getPage('/testGen', self.cookies) |
|---|
| 60 |
self.assertBody('2') |
|---|
| 61 |
self.getPage('/testStr', self.cookies) |
|---|
| 62 |
self.assertBody('3') |
|---|
| 63 |
cherrypy.config.update({ |
|---|
| 64 |
'sessionFilter.storageType' : 'file', |
|---|
| 65 |
}) |
|---|
| 66 |
self.getPage('/testStr') |
|---|
| 67 |
self.assertBody('1') |
|---|
| 68 |
self.getPage('/testGen', self.cookies) |
|---|
| 69 |
self.assertBody('2') |
|---|
| 70 |
self.getPage('/testStr', self.cookies) |
|---|
| 71 |
self.assertBody('3') |
|---|
| 72 |
|
|---|
| 73 |
|
|---|
| 74 |
for fname in os.listdir('.'): |
|---|
| 75 |
if fname.startswith('session-'): |
|---|
| 76 |
os.unlink(fname) |
|---|
| 77 |
|
|---|
| 78 |
|
|---|
| 79 |
if __name__ == "__main__": |
|---|
| 80 |
helper.testmain() |
|---|
| 81 |
|
|---|
