# along with IPyPBX. If not, see <http://www.gnu.org/licenses/>.
from ipypbx import main
+from ipypbx.tests import db
class InitDatabase(object):
- _tables = (
- 'connection', 'sipprofile', 'domain', 'gateway', 'endpoint',
- 'extension')
-
- @classmethod
- def setUpClass(cls):
- cls.db, _created = main.setupDb(dbname=':memory:')
-
def setUp(self):
for table in self._tables:
- self.db.exec_('delete from ipypbxweb_%s' % table)
+ db.exec_('delete from ipypbxweb_%s' % table)