def __init__(self):
self.config = get_config()
self.rpcserver = RPCServer()
self.core = Core(self)
self.db_session = init_model()
self.rpcserver.add_receiver(JSONReceiver())
self.rpcserver.register_object(self.core)
self.rpcserver.register_object(Health())
self.rpcserver.register_object(Qpsmtpd())
self.running = False
def __init__(self):
self.config = get_config()
self.rpcserver = RPCServer()
self.core = Core(self)
self.db_session = init_model()
self.rpcserver.add_receiver(JSONReceiver())
self.rpcserver.register_object(self.core)
self.rpcserver.register_object(Health())
self.rpcserver.register_object(Qpsmtpd())
self.running = False
def setUp(self):
super(DatabaseUnitTest, self).setUp()
from vmail.model import init_model, init_rw_model
engine = self._create_engine()
# Create the database schema
meta.bind = engine
meta.create_all()
# Insert some test data into the database
for (name, quota, account_limit) in testdata.packages:
packages.insert().values(name=name,
quota=quota,
account_limit=account_limit).execute()
for (domain, package, package_id, quota,
account_limit) in testdata.domains:
domains.insert().values(domain=domain,
package=package,
package_id=package_id,
quota=quota,
account_limit=account_limit).execute()
for (domain_id, email, se, name, password, cleartext, quota, enabled,
admin) in testdata.users:
users.insert().values(domain_id=domain_id,
email=email,
secondary_email=se,
name=name,
password=password,
cleartext=cleartext,
quota=quota,
enabled=enabled,
admin=admin).execute()
for (email, bytes, messages) in testdata.user_quotas:
user_quotas.insert().values(email=email,
bytes=bytes,
messages=messages).execute()
for (email, subject, body, created, active) in testdata.vacation:
vacation.insert().values(email=email,
subject=subject,
body=body,
created=created,
active=active).execute()
for (on_vacation, notified,
notified_at) in testdata.vacation_notification:
vacation_notification.insert().values(
on_vacation=on_vacation,
notified=notified,
notified_at=notified_at).execute()
for (domain_id, source, destination) in testdata.forwardings:
forwardings.insert().values(domain_id=domain_id,
source=source,
destination=destination).execute()
for (domain_id, source, destination) in testdata.forwards:
forwards.insert().values(domain_id=domain_id,
source=source,
destination=destination).execute()
for (domain_id, source, transport_) in testdata.transport:
transport.insert().values(domain_id=domain_id,
source=source,
transport=transport_).execute()
for (email, user_id, method, local_addr, remote_addr,
date) in testdata.logins:
logins.insert().values(email=email,
user_id=user_id,
method=method,
local_addr=local_addr,
remote_addr=remote_addr,
date=date).execute()
for (ip_address, action, comment) in testdata.hosts:
hosts.insert().values(ip_address=ip_address,
action=action,
comment=comment).execute()
for address in testdata.blacklist:
blacklist.insert().values(address=address).execute()
for address in testdata.whitelist:
whitelist.insert().values(address=address).execute()
mysql_sucks.insert().values(test=1).execute()
# Initialize the sessions
init_model(engine)
init_rw_model(engine)
# Set the db and rw_db as local variables to avoid auto-connecting
from vmail.model import db, rw_db
self.db = db
self.rw_db = rw_db
def setUp(self):
super(DatabaseUnitTest, self).setUp()
from vmail.model import init_model
engine = self._create_engine()
# Create the database schema
meta.bind = engine
meta.create_all()
# Insert some test data into the database
for (name, quota, account_limit) in testdata.packages:
packages.insert().values(
name = name,
quota = quota,
account_limit = account_limit
).execute()
for (domain, package, package_id, quota, account_limit) in testdata.domains:
domains.insert().values(
domain = domain,
package = package,
package_id = package_id,
quota = quota,
account_limit = account_limit
).execute()
for (domain_id, email, se, name, password, cleartext, quota, enabled, admin) in testdata.users:
users.insert().values(
domain_id = domain_id,
email = email,
secondary_email = se,
name = name,
password = password,
cleartext = cleartext,
quota = quota,
enabled = enabled,
admin = admin
).execute()
for (email, bytes, messages) in testdata.user_quotas:
user_quotas.insert().values(
email = email,
bytes = bytes,
messages = messages
).execute()
for (email, subject, body, created, active) in testdata.vacation:
vacation.insert().values(
email = email,
subject = subject,
body = body,
created = created,
active = active
).execute()
for (on_vacation, notified, notified_at) in testdata.vacation_notification:
vacation_notification.insert().values(
on_vacation = on_vacation,
notified = notified,
notified_at = notified_at
).execute()
for (domain_id, source, destination) in testdata.forwardings:
forwardings.insert().values(
domain_id = domain_id,
source = source,
destination = destination
).execute()
for (domain_id, source, destination) in testdata.forwards:
forwards.insert().values(
domain_id = domain_id,
source = source,
destination = destination
).execute()
for (domain_id, source, transport_) in testdata.transport:
transport.insert().values(
domain_id = domain_id,
source = source,
transport = transport_
).execute()
for (email, user_id, method, local_addr, remote_addr, date) in testdata.logins:
logins.insert().values(
email = email,
user_id = user_id,
method = method,
local_addr = local_addr,
remote_addr = remote_addr,
date = date
).execute()
for (ip_address, action, comment) in testdata.hosts:
hosts.insert().values(
ip_address = ip_address,
action = action,
comment = comment
).execute()
for address in testdata.blacklist:
blacklist.insert().values(address=address).execute()
for address in testdata.whitelist:
whitelist.insert().values(address=address).execute()
mysql_sucks.insert().values(test=1).execute()
# Initialize the sessions
self.db_session = init_model(engine)
self.db = self.db_session()
