tests/__init__.py

   1
   2 __all__ = ['TestCase', 'main']
   3
   4
   5 import os
   6 import unittest
   7 from m_librarian.db import open_db, init_db
   8 from m_librarian.inp import import_inpx
   9
  10
  11 class TestCase(unittest.TestCase):
  12     def setUp(self):
  13         try:
  14             os.remove('/tmp/m_librarian-test.sqlite')
  15         except OSError:
  16             pass
  17         open_db('sqlite:///tmp/m_librarian-test.sqlite')
  18         init_db()
  19
  20     def tearDown(self):
  21         try:
  22             os.remove('/tmp/m_librarian-test.sqlite')
  23         except OSError:
  24             pass
  25
  26     def import_inpx(self, inpx):
  27         import_inpx(os.path.join(os.path.dirname(__file__), inpx))
  28
  29
  30 def main():
  31     try:
  32         unittest.main(testRunner=unittest.TextTestRunner())
  33     except SystemExit, msg:
  34         result = msg.args[0]
  35     else:
  36         result = 0
  37     raise SystemExit(result)