本文整理汇总了Python中trac.test.EnvironmentStub.get_known_users方法的典型用法代码示例。如果您正苦于以下问题：Python EnvironmentStub.get_known_users方法的具体用法？Python EnvironmentStub.get_known_users怎么用？Python EnvironmentStub.get_known_users使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类trac.test.EnvironmentStub的用法示例。
在下文中一共展示了EnvironmentStub.get_known_users方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: SessionTestCase
# 需要导入模块: from trac.test import EnvironmentStub [as 别名]
# 或者: from trac.test.EnvironmentStub import get_known_users [as 别名]
#.........这里部分代码省略.........
"""Verify that setting a variable in a session to the default value
removes it from the session.
"""
with self.env.db_transaction as db:
db("INSERT INTO session VALUES ('john', 1, 0)")
db("INSERT INTO session_attribute VALUES ('john', 1, 'foo', 'bar')")
session = DetachedSession(self.env, 'john')
self.assertEqual('bar', session['foo'])
# Setting the variable to the default value removes the variable
with self.env.db_transaction as db:
session.set('foo', 'default', 'default')
session.save()
self.assertEqual(0, self.env.db_query("""
SELECT COUNT(*) FROM session_attribute
WHERE sid='john' AND name='foo'
""")[0][0])
# Setting the variable to a value different from the default sets it
with self.env.db_transaction as db:
session.set('foo', 'something', 'default')
session.save()
self.assertEqual('something', self.env.db_query("""
SELECT value FROM session_attribute
WHERE sid='john' AND name='foo'
""")[0][0])
def test_session_set_email(self):
"""Setting session email invalidates known_users cache."""
session = DetachedSession(self.env, 'user')
session['email'] = '[email protected]'
self.assertEqual([], list(self.env.get_known_users()))
session.save()
email = None
for sid, name, email in self.env.get_known_users():
if sid == 'user':
break
self.assertEqual(session['email'], email)
def test_session_set_name(self):
"""Setting session name invalidates known_users cache."""
session = DetachedSession(self.env, 'user')
session['name'] = 'The User'
self.assertEqual([], list(self.env.get_known_users()))
session.save()
name = None
for sid, name, email in self.env.get_known_users():
if sid == 'user':
break
self.assertEqual(session['name'], name)
def test_session_admin_list(self):
auth_list, anon_list, all_list = _prep_session_table(self.env)
sess_admin = SessionAdmin(self.env)
# Verify the empty case
self.assertRaises(StopIteration, sess_admin._get_list([]).next)
self.assertEqual([i for i in sess_admin._get_list(['authenticated'])],
auth_list)
self.assertEqual([i for i in sess_admin._get_list(['anonymous'])],
anon_list)
self.assertEqual([i for i in sess_admin._get_list(['*'])], all_list)示例2: KnownUsersTestCase
# 需要导入模块: from trac.test import EnvironmentStub [as 别名]
# 或者: from trac.test.EnvironmentStub import get_known_users [as 别名]
class KnownUsersTestCase(unittest.TestCase):
def setUp(self):
self.env = EnvironmentStub()
users = [
('123', None, '[email protected]', 0),
('jane', 'Jane', None, 1),
('joe', None, '[email protected]', 1),
('tom', 'Tom', '[email protected]', 1)
]
self.expected = []
for user in users:
self._insert_user(user)
if user[3] == 1:
self.expected.append(user[:3])
def tearDown(self):
self.env.reset_db()
def _insert_user(self, user):
with self.env.db_transaction as db:
db.execute("""
INSERT INTO session VALUES (%s,%s,0)
""", (user[0], user[3]))
db.executemany("""
INSERT INTO session_attribute VALUES (%s,%s,%s,%s)
""", [(user[0], user[3], 'name', user[1]),
(user[0], user[3], 'email', user[2])])
def test_get_known_users_as_list_of_tuples(self):
users = list(self.env.get_known_users())
i = 0
for i, user in enumerate(users):
self.assertEqual(self.expected[i], user)
else:
self.assertEqual(2, i)
def test_get_known_users_as_dict(self):
users = self.env.get_known_users(as_dict=True)
self.assertEqual(3, len(users))
for exp in self.expected:
self.assertEqual(exp[1:], users[exp[0]])
def test_get_known_users_is_cached(self):
self.env.get_known_users()
self.env.get_known_users(as_dict=True)
self._insert_user(('user4', None, None, 1))
users_list = list(self.env.get_known_users())
users_dict = self.env.get_known_users(as_dict=True)
i = 0
for i, user in enumerate(users_list):
self.assertEqual(self.expected[i], user)
self.assertIn(self.expected[i][0], users_dict)
else:
self.assertEqual(2, i)
self.assertEqual(3, len(users_dict))
def test_invalidate_known_users_cache(self):
self.env.get_known_users()
self.env.get_known_users(as_dict=True)
user = ('user4', 'User Four', '[email protected]', 1)
self._insert_user(user)
self.expected.append(user[:3])
self.env.invalidate_known_users_cache()
users_list = self.env.get_known_users()
users_dict = self.env.get_known_users(as_dict=True)
i = 0
for i, user in enumerate(users_list):
self.assertEqual(self.expected[i], user)
self.assertIn(self.expected[i][0], users_dict)
else:
self.assertEqual(3, i)
self.assertEqual(4, len(users_dict))