← Back to branch summary

~ubuntu-branches/ubuntu/raring/trac-accountmanager/raring

« back to all changes in this revision

Viewing changes to acct_mgr/tests/db.py

  • Committer: Bazaar Package Importer
  • Author(s): Leo Costela
  • Date: 2008-07-15 17:21:11 UTC
  • Revision ID: james.westby@ubuntu.com-20080715172111-ool7wmy573gqolfr
Tags: upstream-0.2.1~vcs20080715
Import upstream version 0.2.1~vcs20080715

Show diffs side-by-side

added added

removed removed

Lines of Context:
acct_mgr/tests/db.py
 
1
# -*- coding: utf-8 -*-
 
2
#
 
3
# Copyright (C) 2007 Matthew Good <trac@matt-good.net>
 
4
#
 
5
# "THE BEER-WARE LICENSE" (Revision 42):
 
6
# <trac@matt-good.net> wrote this file.  As long as you retain this notice you
 
7
# can do whatever you want with this stuff. If we meet some day, and you think
 
8
# this stuff is worth it, you can buy me a beer in return.   Matthew Good
 
9
#
 
10
# Author: Matthew Good <trac@matt-good.net>
 
11
 
 
12
import os.path
 
13
import shutil
 
14
import tempfile
 
15
import unittest
 
16
 
 
17
from trac.test import EnvironmentStub, Mock
 
18
 
 
19
from acct_mgr.db import SessionStore
 
20
 
 
21
class _BaseTestCase(unittest.TestCase):
 
22
    def setUp(self):
 
23
        #self.basedir = os.path.realpath(tempfile.mkdtemp())
 
24
        self.env = EnvironmentStub(enable=['trac.*', 'acct_mgr.*'])
 
25
        self.env.config.set('account-manager', 'password_store',
 
26
                            'SessionStore')
 
27
        self.store = SessionStore(self.env)
 
28
        #self.env.path = os.path.join(self.basedir, 'trac-tempenv')
 
29
        #os.mkdir(self.env.path)
 
30
 
 
31
    def test_get_users(self):
 
32
        db = self.env.get_db_cnx()
 
33
        cursor = db.cursor()
 
34
        cursor.executemany("INSERT INTO session_attribute "
 
35
                       "(sid,authenticated,name,value) "
 
36
                       "VALUES (%s,1,'password',%s)",
 
37
                       [('a', 'a'),
 
38
                        ('b', 'b'),
 
39
                        ('c', 'c')])
 
40
        self.assertEqual(['a', 'b', 'c'], list(self.store.get_users()))
 
41
 
 
42
    def test_has_user(self):
 
43
        db = self.env.get_db_cnx()
 
44
        cursor = db.cursor()
 
45
        cursor.execute("INSERT INTO session_attribute "
 
46
                       "(sid,authenticated,name,value) "
 
47
                       "VALUES (%s,1,'password',%s)",
 
48
                       ('bar', 'bar'))
 
49
 
 
50
        self.assertFalse(self.store.has_user('foo'))
 
51
        self.assertTrue(self.store.has_user('bar'))
 
52
 
 
53
    def test_create_user(self):
 
54
        self.assertFalse(self.store.has_user('foo'))
 
55
        self.store.set_password('foo', 'password')
 
56
        self.assertTrue(self.store.has_user('foo'))
 
57
 
 
58
    def test_update_password(self):
 
59
        self.store.set_password('foo', 'pass1')
 
60
        self.assertFalse(self.store.check_password('foo', 'pass2'))
 
61
        self.store.set_password('foo', 'pass2')
 
62
        self.assertTrue(self.store.check_password('foo', 'pass2'))
 
63
 
 
64
    def test_delete_user(self):
 
65
        self.store.set_password('foo', 'password')
 
66
        self.assertTrue(self.store.has_user('foo'))
 
67
        self.assertTrue(self.store.delete_user('foo'))
 
68
        self.assertFalse(self.store.has_user('foo'))
 
69
 
 
70
    def test_delete_nonexistant_user(self):
 
71
        self.assertFalse(self.store.has_user('foo'))
 
72
        self.assertFalse(self.store.delete_user('foo'))
 
73
 
 
74
 
 
75
class HtDigestTestCase(_BaseTestCase):
 
76
    def setUp(self):
 
77
        _BaseTestCase.setUp(self)
 
78
        self.env.config.set('account-manager', 'hash_method',
 
79
                            'HtDigestHashMethod')
 
80
        self.env.config.set('account-manager', 'htdigest_realm',
 
81
                            'TestRealm')
 
82
 
 
83
 
 
84
class HtPasswdTestCase(_BaseTestCase):
 
85
    def setUp(self):
 
86
        _BaseTestCase.setUp(self)
 
87
        self.env.config.set('account-manager', 'hash_method',
 
88
                            'HtPasswdHashMethod')
 
89
 
 
90
 
 
91
def suite():
 
92
    suite = unittest.TestSuite()
 
93
    suite.addTest(unittest.makeSuite(HtDigestTestCase, 'test'))
 
94
    suite.addTest(unittest.makeSuite(HtPasswdTestCase, 'test'))
 
95
    return suite
 
96
 
 
97
if __name__ == '__main__':
 
98
    unittest.main(defaultTest='suite')
 
99
 
 
100