Differential D134 Diff 284 tests/unit/test-020-auth_cache.py

Changeset View

Standalone View

tests/unit/test-020-auth_cache.py

	# coding: utf8			# coding: utf8

	import unittest			import unittest

	import datetime			import datetime
	import os			import os

	from pykolab.auth.ldap import auth_cache			from pykolab.auth.ldap import auth_cache
	import pykolab			import pykolab
	conf = pykolab.getConf()			conf = pykolab.getConf()
	conf.finalize_conf()			conf.finalize_conf()

	import sqlalchemy			import sqlalchemy
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file

	from sqlalchemy import Column			from sqlalchemy import Column
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy import DateTime			from sqlalchemy import DateTime
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy import Integer			from sqlalchemy import Integer
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy import MetaData			from sqlalchemy import MetaData
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy import String			from sqlalchemy import String
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy import Table			from sqlalchemy import Table
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy import Text			from sqlalchemy import Text
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file

	from sqlalchemy import desc			from sqlalchemy import desc
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy import create_engine			from sqlalchemy import create_engine
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy.ext.declarative import declarative_base			from sqlalchemy.ext.declarative import declarative_base
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file
	from sqlalchemy.orm import mapper			from sqlalchemy.orm import mapper
				Lint: PEP8 E402 Inline Actions module level import not at top of file Lint: PEP8 E402: module level import not at top of file

	try:			try:
	from sqlalchemy.orm import relationship			from sqlalchemy.orm import relationship
	except:			except:
	from sqlalchemy.orm import relation as relationship			from sqlalchemy.orm import relation as relationship

	try:			try:
	from sqlalchemy.orm import sessionmaker			from sqlalchemy.orm import sessionmaker
	except:			except:
	from sqlalchemy.orm import create_session			from sqlalchemy.orm import create_session

	metadata = MetaData()			metadata = MetaData()

	##
	## Classes
	##

	DeclarativeBase = declarative_base()			DeclarativeBase = declarative_base()


	class Entry(DeclarativeBase):			class Entry(DeclarativeBase):
	__tablename__ = 'entries'			__tablename__ = 'entries'

	id = Column(Integer, primary_key=True)			id = Column(Integer, primary_key=True)
	domain = Column(String(256), index=True, nullable=True)			domain = Column(String(256), index=True, nullable=True)
	key = Column(Text, index=True, nullable=False)			key = Column(Text, index=True, nullable=False)
	value = Column(Text, nullable=False)			value = Column(Text, nullable=False)
	last_change = Column(DateTime, nullable=False, default=datetime.datetime.now())			last_change = Column(DateTime, nullable=False, default=datetime.datetime.now())
				Lint: PEP8 E501 Inline Actions line too long (83 > 79 characters) Lint: PEP8 E501: line too long (83 > 79 characters)

	def __init__(self, key, value):			def __init__(self, key, value):
	self.key = key			self.key = key
	if not isinstance(value, unicode):			if not isinstance(value, unicode):
	self.value = unicode(value, 'utf-8')			self.value = unicode(value, 'utf-8')
	else:			else:
	self.value = value			self.value = value

	engine = create_engine('sqlite:////tmp/%s.db' % (os.getpid()), echo=False, echo_pool=False)			engine = create_engine('sqlite:////tmp/%s.db' % (os.getpid()), echo=False, echo_pool=False)
				Lint: PEP8 E501 Inline Actions line too long (91 > 79 characters) Lint: PEP8 E501: line too long (91 > 79 characters)
	DeclarativeBase.metadata.create_all(engine)			DeclarativeBase.metadata.create_all(engine)

	Session = sessionmaker(bind=engine)			Session = sessionmaker(bind=engine)
	db = Session()			db = Session()

	auth_cache.db = db			auth_cache.db = db


	class TestAuthCache(unittest.TestCase):			class TestAuthCache(unittest.TestCase):
	def test_001_plain_insert(self):			def test_001_plain_insert(self):
	auth_cache.set_entry(			auth_cache.set_entry(
	'somekey',			'somekey',
	'ou=People,dc=example,dc=org'			'ou=People,dc=example,dc=org'
	)			)

	result = auth_cache.get_entry('somekey')			result = auth_cache.get_entry('somekey')
	self.assertEqual(result, 'ou=People,dc=example,dc=org')			self.assertEqual(result, 'ou=People,dc=example,dc=org')

	def test_002_plain_encoding_insert(self):			def test_002_plain_encoding_insert(self):
	auth_cache.set_entry(			auth_cache.set_entry(
	'somekey2',			'somekey2',
	'ou=Geschäftsbereich,ou=People,dc=example,dc=org'			'ou=Geschäftsbereich,ou=People,dc=example,dc=org'
	)			)

	result = auth_cache.get_entry('somekey2')			result = auth_cache.get_entry('somekey2')
	self.assertEqual(result, 'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org')			self.assertEqual(result, 'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org')
				Lint: PEP8 E501 Inline Actions line too long (90 > 79 characters) Lint: PEP8 E501: line too long (90 > 79 characters)

	def test_003_unicode_insert(self):			def test_003_unicode_insert(self):
	auth_cache.set_entry(			auth_cache.set_entry(
	'somekey3',			'somekey3',
	u'ou=Geschäftsbereich,ou=People,dc=example,dc=org'			u'ou=Geschäftsbereich,ou=People,dc=example,dc=org'
	)			)

	result = auth_cache.get_entry('somekey3')			result = auth_cache.get_entry('somekey3')
	self.assertEqual(result, 'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org')			self.assertEqual(result, 'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org')
				Lint: PEP8 E501 Inline Actions line too long (90 > 79 characters) Lint: PEP8 E501: line too long (90 > 79 characters)

	@unittest.skip("Double encoding or decoding")			@unittest.skip("Double encoding or decoding")
	def test_004_unicode_escape(self):			def test_004_unicode_escape(self):
	auth_cache.set_entry(			auth_cache.set_entry(
	'somekey4',			'somekey4',
	u'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org'			u'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org'
	)			)

	result = auth_cache.get_entry('somekey4')			result = auth_cache.get_entry('somekey4')
	self.assertEqual(result, u'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org')			self.assertEqual(result, u'ou=Gesch\xc3\xa4ftsbereich,ou=People,dc=example,dc=org')
				Lint: PEP8 E501 Inline Actions line too long (91 > 79 characters) Lint: PEP8 E501: line too long (91 > 79 characters)

	def test_005_longkey(self):			def test_005_longkey(self):
	auth_cache.set_entry(			auth_cache.set_entry(
	'v' + 'e'*512 + 'rylongkey',			'v' + 'e'*512 + 'rylongkey',
	'v' + 'e'*512 + 'rylongvalue'			'v' + 'e'*512 + 'rylongvalue'
	)			)

	result = auth_cache.get_entry('v' + 'e'*512 + 'rylongkey')			result = auth_cache.get_entry('v' + 'e'*512 + 'rylongkey')
	self.assertEqual(result, 'v' + 'e'*512 + 'rylongvalue')			self.assertEqual(result, 'v' + 'e'*512 + 'rylongvalue')