tests/__init__.py

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