Changeset View
Changeset View
Standalone View
Standalone View
tests/unit/test-020-auth_cache.py
| Show All 31 Lines | |||||
| 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()) | ||||
| 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) | ||||
| 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') | ||||
| Show All 38 Lines | |||||