Skip to main content

Creates and drops temporary databases for testing purposes.

Project description

Creates and drops temporary databases for testing purposes.

This package has been tested with Python 2.6, 2.7, 3.3 up to 3.5.

Copyright (c) 2008-2015 gocept gmbh & co. kg and contributors.

All Rights Reserved.

This software is subject to the provisions of the Zope Public License, Version 2.1 (ZPL). A copy of the ZPL should accompany this distribution. THIS SOFTWARE IS PROVIDED “AS IS” AND ANY AND ALL EXPRESS OR IMPLIED WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS FOR A PARTICULAR PURPOSE.

gocept.testdb provides small helper classes that create and drop temporary databases.

>>> import gocept.testdb
>>> import os.path
>>> import shutil
>>> import sqlalchemy
>>> import tempfile
>>> sql_dir = tempfile.mkdtemp()
>>> schema = os.path.join(sql_dir, 'sample.sql')
>>> with open(schema, 'w') as f:
...     _ = f.write('CREATE TABLE foo (dummy int);')

We’ll use a custom prefix specific to the current process whenever we create fixed-name databases during this test, in order to allow concurrent test runs on the same machine (such as a CI server):

>>> import os
>>> pid_prefix = 'gocept.testdb.tests-PID%s-' % os.getpid()
>>> db_template = pid_prefix + 'templatetest'

MySQL

gocept.testdb expects the usage of PyMySQL as database driver.

First, instantiate a test database object and have it create the database on the server. You can specify the name or the prefix of the database:

>>> gocept.testdb.MySQL(schema_path=schema, db_name='mytestdb').dsn
'mysql+pymysql://.../mytestdb'
>>> db = gocept.testdb.MySQL(schema_path=schema, prefix='my-tests')
>>> db.dsn
'mysql+pymysql://.../my-tests-...'
>>> db.create()

This will use the appropriate command-line tools to create a database with a random name.

You can use the following environment variables to customize the DSN:

MYSQL_HOST

hostname, defaults to localhost

MYSQL_USER

username, defaults to None which means to use the name of the user logged into the operating system.

MYSQL_PASS

password, defaults to None which means no password required.

MYSQL_COMMAND_POSTFIX

attach this postfix to MySQL commands, defaults to an empty string. You need this variable if your MySQL commands are named like mysql5 instead of mysql.

The dbapi DSN can then be used to connect to the database:

>>> engine = sqlalchemy.create_engine(db.dsn)

The database is marked as a testing database by creating a table called tmp_functest in it:

>>> conn = engine.connect()
>>> ignore = conn.execute('SELECT * from tmp_functest')

The database object also offers convenience methods for determining the status of the database:

>>> db.exists
True
>>> db.is_testing
True

If you passed a schema_path to the constructor, the SQL code in this file is executed, e. g. to set up tables:

>>> ignore = conn.execute('SELECT * from foo')

When done, simply drop the database:

>>> conn.close()
>>> try:
...     db.drop()
... except RuntimeError:
...     pass  # Jenkins fails to drop databases in Python 3, *sigh*

PostgreSQL

General

The same procedure also works for PostgreSQL. You can use the following environment variables to customize the DSN:

POSTGRES_HOST

hostname, defaults to localhost

POSTGRES_USER

username, defaults to None which means to use the name of the user logged into the operating system.

POSTGRES_PASS

password, defaults to None which means no password required. Note: Instead of using POSTGRES_PASS, use the ~/.pgpass mechanism provided by postgres itself.

>>> db = gocept.testdb.PostgreSQL(schema_path=schema)
>>> db.create()
>>> engine = sqlalchemy.create_engine(db.dsn)
>>> conn = engine.connect()
>>> ignore = conn.execute('SELECT * from tmp_functest')
>>> ignore = conn.execute('SELECT * from foo')
>>> conn.invalidate()
>>> db.drop()

Encoding

For PostgreSQL an optional encoding, database name and database name prefix parameters can be specified in the constructor. They are used when creating the database.

>>> gocept.testdb.PostgreSQL(encoding='UTF8', db_name='mytestdb').dsn
'postgresql://...localhost/mytestdb'
>>> gocept.testdb.PostgreSQL(prefix='my-tests').dsn
'postgresql://...localhost/my-tests-...'

Templates

For PostgreSQL, an optional template parameter can be passed to the constructor. It specifies the name of a template database which is used for the creation of the test database. If the template database does not exist, it is created with the specified schema.

The first time you create the database with the db_template argument, the template database is created (if it does not exist already) along with the requested database:

>>> db = gocept.testdb.PostgreSQL(schema_path=schema, db_template=db_template)

Now with the template available, the schema is not used any more to create the database (it’s copied from the template database).

When creating the database, we can, however, force the template database to be created afresh from the schema. Doing so now will leave us with both a test database and a template database according to the modified schema:

>>> db = gocept.testdb.PostgreSQL(
...     schema_path=schema, db_template=db_template, force_template=True)

The template database (and with it, the test database) is also created anew if the schema file is newer than the existing template database.

If, however, the template database cannot be set up properly, it is removed altogether to avoid a broken template database interfering with subsequent tests.

The drop-all command-line script

The Database classes’ drop_all functionality is available independently through a command-line script named drop-all. The script drops any test databases from both the PostgreSQL and MySQL servers that match the test-database naming convention with any of the prefixes passed as command-line arguments. Usage:

$ bin/drop-all "<prefix>"

Test clean up:

>>> shutil.rmtree(sql_dir)

Development

To run the buildout of this package copy local.cfg.example to local.cfg and edit it to match your needs:

  • MYSQL_COMMAND_POSTFIX is needed if your MySQL commands look like mysql5 instead of mysql

  • MySQL has to open a port for the tests to connect to. Configure this in your my.cnf.

Running tests

Install tox and run the tests calling tox.

Source code

The source code is available at https://bitbucket.org/gocept/gocept.testdb

Bug reports

Please report any bugs you find to https://bitbucket.org/gocept/gocept.testdb/issues

Changelog

1.3 (2015-10-07)

  • Drop support of Python 3.2.

  • Streamline documentation.

1.3a1 (2015-09-24)

  • Officially support Python 3.2 up to Python 3.4.

  • Switch PyMySQL driver to support Python 3 for MySQL.

  • Add environment variable MYSQL_COMMAND_POSTFIX to use MySQL commands like mysql5 instead of mysql.

1.2.1 (2013-09-03)

  • Improve retry logic for dropping databases: ensure that db actually exists before trying to drop it (#12706).

1.2 (2013-06-21)

  • Provisional compatibility to Python 3 (no compatible mysql driver exists yet, though).

  • Fixed test code that made implicit assumptions about existing databases on the PostgreSQL server used or depended on timing conditions.

1.1.1 (2013-03-19)

  • Use template0 when changing LC_COLLATE which was introduced in v1.1.

1.1 (2013-03-19)

  • Add possibility to set the LC_COLLATE when creating a database.

1.0 (2013-03-12)

  • Allow a postfix for mysql-commands.

1.0b5 (2011-12-22)

  • Use timestamp for randomizing database names to avoid collisions.

1.0b4 (2011-09-23)

  • drop-all entry point now also works, if PostgreSQL or MySQL is not installed/running.

1.0b3 (2011-05-13)

  • Added connect convenience method.

1.0b2 (2011-05-04)

  • Added is_testing property as a convenience API.

  • Added exists property as a convenience API.

1.0b1 (2011-04-12)

  • Changed the protocol for using test databases: separated instantiating a Database instance from creating the database on the server. Creating Database instances is cheap now so they can be interacted with and passed around, deferring the expensive database creation on the server to the moment db.create() is called explicitly. This is also more symmetric with respect to db.drop().

  • Added functionality for dropping all databases matching the testdb naming scheme which may have been left on the server by earlier test runs. Optionally also drops the template db.

  • Added option to specify the name of the database for MySQL and PostgreSQL.

  • Added an option to specify a template database for PostgreSQL. If it does not exist, it is created from the specified schema. It is also possible to force the creation of the template even if it exists (dropping the current template database). Whenever the schema has changed since the template db was last created, the template db is updated automatically.

0.4 (2010-12-15)

  • Added option to specify the encoding for the PostgreSQL database.

  • Updated PostgreSQL protocol from postgres: to postgresql: as the former one is deprecated in SQLAlchemy 0.6, thus requiring atleast version 0.5.6 of SQLAlchemy.

  • Added documentation how to develop this package further.

  • Added doumentation about usage of Database prefix.

0.3 (2010-10-15)

  • PostgreSQL: Don’t call createdb/dropdb with --quite but only psql.

0.2 (2009-02-26)

  • implemented authentication with password for mysql. Passwords for postgres are still not supported, though.

0.1 (2008-09-26)

  • first release

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

gocept.testdb-1.3.tar.gz (22.5 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page