From 39f62374aaccf70ee3770ab2a8a514bfd3f40b0e Mon Sep 17 00:00:00 2001
From: apahomov <gorblnu4@gmail.com>
Date: Tue, 15 Jan 2013 10:30:48 +0400
Subject: [PATCH] Added tests

---
 tests/functional_tests.py  | 18 ++++++++++++++++++
 tests/test_app/__init__.py |  7 ++++++-
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/tests/functional_tests.py b/tests/functional_tests.py
index 49e9be6..61fc92d 100644
--- a/tests/functional_tests.py
+++ b/tests/functional_tests.py
@@ -169,6 +169,24 @@ class DefaultSecurityTests(SecurityTest):
         self.assertEquals('Basic realm="My Realm"',
                           r.headers['WWW-Authenticate'])
 
+    def test_multi_auth_basic(self):
+        r = self._get('/multi_auth', headers={
+            'Authorization': 'Basic ' + base64.b64encode("joe@lp.com:password")
+        })
+        self.assertIn('Basic', r.data)
+
+    def test_multi_auth_token(self):
+        r = self.json_authenticate()
+        data = json.loads(r.data)
+        token = data['response']['user']['authentication_token']
+        r = self._get('/multi_auth?auth_token=' + token)
+        self.assertIn('Token', r.data)
+
+    def test_multi_auth_session(self):
+        self.authenticate()
+        r = self._get('/multi_auth')
+        self.assertIn('Session', r.data)
+
     def test_user_deleted_during_session_reverts_to_anonymous_user(self):
         self.authenticate()
 
diff --git a/tests/test_app/__init__.py b/tests/test_app/__init__.py
index 5cad502..863db79 100644
--- a/tests/test_app/__init__.py
+++ b/tests/test_app/__init__.py
@@ -4,7 +4,7 @@ from flask import Flask, render_template, current_app
 from flask.ext.mail import Mail
 from flask.ext.security import login_required, roles_required, roles_accepted
 from flask.ext.security.decorators import http_auth_required, \
-     auth_token_required
+     auth_token_required, auth_required
 from flask.ext.security.utils import encrypt_password
 from werkzeug.local import LocalProxy
 
@@ -50,6 +50,11 @@ def create_app(config):
     def token():
         return render_template('index.html', content='Token Authentication')
 
+    @app.route('/multi_auth')
+    @auth_required('session', 'token', 'basic')
+    def multi_auth():
+        return render_template('index.html', content='Session, Token, Basic auth')
+
     @app.route('/post_logout')
     def post_logout():
         return render_template('index.html', content='Post Logout')