|
| 1 | +var Koa = require('koa'); |
| 2 | +var bodyParser = require('koa-bodyparser'); |
| 3 | +var request = require('supertest'); |
| 4 | +var should = require('should'); |
| 5 | + |
| 6 | +const AV = require('../..'); |
| 7 | +const appInfo = require('../fixtures/app-info'); |
| 8 | + |
| 9 | +var app = new Koa(); |
| 10 | + |
| 11 | +if (process.env.KOA_VER === '1') { |
| 12 | + app.use(AV.koa()); |
| 13 | + app.use(bodyParser()); |
| 14 | + app.use(AV.Cloud.CookieSession({framework: 'koa', secret: 'my secret', maxAge: 3600000, fetchUser: true})); |
| 15 | + |
| 16 | + app.use(function *(next) { |
| 17 | + try { |
| 18 | + yield next; |
| 19 | + } catch (err) { |
| 20 | + this.status = err.status || 500; |
| 21 | + this.body = err.message; |
| 22 | + this.app.emit('error', err, this); |
| 23 | + } |
| 24 | + }); |
| 25 | + |
| 26 | + app.use(function *(next) { |
| 27 | + var method = this.request.method; |
| 28 | + var url = this.request.url; |
| 29 | + |
| 30 | + if (method === 'GET' && url === '/') { |
| 31 | + this.status = 200; |
| 32 | + this.body = '<p>Hello world</p>'; |
| 33 | + } else if (method === 'POST' && url === '/login') { |
| 34 | + return AV.User.logIn(this.request.body.username, this.request.body.password).then( user => { |
| 35 | + this.saveCurrentUser(user); |
| 36 | + this.response.redirect('/profile'); |
| 37 | + }); |
| 38 | + } else if (method === 'GET' && url === '/profile') { |
| 39 | + this.status = 200; |
| 40 | + this.body = this.currentUser; |
| 41 | + } else if (method === 'POST' && url === '/logout') { |
| 42 | + this.status = 200; |
| 43 | + this.saveCurrentUser(null); |
| 44 | + } else { |
| 45 | + yield next; |
| 46 | + } |
| 47 | + }); |
| 48 | +} else { |
| 49 | + app.use(AV.koa2()); |
| 50 | + app.use(bodyParser()); |
| 51 | + app.use(AV.Cloud.CookieSession({framework: 'koa2', secret: 'my secret', maxAge: 3600000, fetchUser: true})); |
| 52 | + |
| 53 | + app.use(async (ctx, next) => { |
| 54 | + try { |
| 55 | + await next(); |
| 56 | + } catch (err) { |
| 57 | + ctx.status = err.status || 500; |
| 58 | + ctx.body = err.message; |
| 59 | + ctx.app.emit('error', err, this); |
| 60 | + } |
| 61 | + }); |
| 62 | + |
| 63 | + app.use(async (ctx, next) => { |
| 64 | + var method = ctx.request.method; |
| 65 | + var url = ctx.request.url; |
| 66 | + |
| 67 | + if (method === 'GET' && url === '/') { |
| 68 | + ctx.status = 200; |
| 69 | + ctx.body = '<p>Hello world</p>'; |
| 70 | + } else if (method === 'POST' && url === '/login') { |
| 71 | + return AV.User.logIn(ctx.request.body.username, ctx.request.body.password).then( user => { |
| 72 | + ctx.saveCurrentUser(user); |
| 73 | + ctx.response.redirect('/profile'); |
| 74 | + }); |
| 75 | + } else if (method === 'GET' && url === '/profile') { |
| 76 | + ctx.status = 200; |
| 77 | + ctx.body = ctx.currentUser; |
| 78 | + } else if (method === 'POST' && url === '/logout') { |
| 79 | + ctx.status = 200; |
| 80 | + ctx.saveCurrentUser(null); |
| 81 | + } else { |
| 82 | + return next(); |
| 83 | + } |
| 84 | + }); |
| 85 | +} |
| 86 | + |
| 87 | +var server = app.listen(); |
| 88 | + |
| 89 | +describe('koa/cookie-session', function() { |
| 90 | + it('index', function(done) { |
| 91 | + request(server).get('/') |
| 92 | + .expect(200, function(err, res) { |
| 93 | + res.headers['content-type'].should.be.startWith('text/html'); |
| 94 | + res.text.should.be.equal('<p>Hello world</p>'); |
| 95 | + done(err); |
| 96 | + }); |
| 97 | + }); |
| 98 | + |
| 99 | + it('loign', function(done) { |
| 100 | + request(server).post('/login') |
| 101 | + .send({ |
| 102 | + username: 'admin', |
| 103 | + password: 'admin' |
| 104 | + }) |
| 105 | + .expect(302, function(err, res) { |
| 106 | + res.headers.location.should.equal('/profile'); |
| 107 | + res.headers['set-cookie'][0].indexOf('avos:sess=eyJfdWlkIjoiNTRmZDZhMDNlNGIwNmM0MWUwMGIxZjQwIiwiX3Nlc3Npb25Ub2tlbiI6IncyanJ0a2JlaHAzOG90cW1oYnF1N3liczkifQ==; path=/; expires=').should.equal(0); |
| 108 | + res.headers['set-cookie'][1].indexOf('avos:sess.sig=jMYF3Iwhmw903-K1K12MVdAFOh0; path=/; expires=').should.equal(0); |
| 109 | + done(err); |
| 110 | + }); |
| 111 | + }); |
| 112 | + |
| 113 | + it('profile', function(done) { |
| 114 | + request(server).get('/profile') |
| 115 | + .set('Cookie', 'avos:sess=eyJfdWlkIjoiNTRmZDZhMDNlNGIwNmM0MWUwMGIxZjQwIiwiX3Nlc3Npb25Ub2tlbiI6IncyanJ0a2JlaHAzOG90cW1oYnF1N3liczkifQ==; avos:sess.sig=jMYF3Iwhmw903-K1K12MVdAFOh0') |
| 116 | + .expect(200, function(err, res) { |
| 117 | + should.exist(res.body.objectId); |
| 118 | + res.body.username.should.be.equal('admin'); |
| 119 | + done(err); |
| 120 | + }); |
| 121 | + }); |
| 122 | + |
| 123 | + it('profile without cookie', function(done) { |
| 124 | + request(server).get('/profile') |
| 125 | + .expect(204, function(err, res) { |
| 126 | + res.body.should.be.empty(); |
| 127 | + done(err); |
| 128 | + }); |
| 129 | + }); |
| 130 | + |
| 131 | + it('logout', function(done) { |
| 132 | + request(server).post('/logout') |
| 133 | + .set('Cookie', 'avos:sess=eyJfdWlkIjoiNTRmZDZhMDNlNGIwNmM0MWUwMGIxZjQwIiwiX3Nlc3Npb25Ub2tlbiI6IncyanJ0a2JlaHAzOG90cW1oYnF1N3liczkifQ==; avos:sess.sig=jMYF3Iwhmw903-K1K12MVdAFOh0') |
| 134 | + .expect(200, function(err, res) { |
| 135 | + res.headers['set-cookie'][0].indexOf('avos:sess=; path=/; expires=').should.equal(0); |
| 136 | + done(err); |
| 137 | + }); |
| 138 | + }) |
| 139 | +}); |
0 commit comments