def setUp(self):
"""
set up DAS core module
"""
self.db = 'test_views.db'
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config['logger'] = logger
config['verbose'] = debug
config['views_dir'] = os.getcwd()
config['views_engine'] = 'sqlite:///%s' % self.db
try:
del config['sum_views']
except:
pass
self.view = DASViewManager(config)
def setUp(self):
"""
set up DAS core module
"""
self.db = 'test_views.db'
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config['logger'] = logger
config['verbose'] = debug
config['views_dir'] = os.getcwd()
config['views_engine'] = 'sqlite:///%s' % self.db
try:
del config['sum_views']
except:
pass
self.view = DASViewManager(config)
class testDAS(unittest.TestCase):
"""
A test class for the DASViewManager class
"""
def setUp(self):
"""
set up DAS core module
"""
self.db = 'test_views.db'
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config['logger'] = logger
config['verbose'] = debug
config['views_dir'] = os.getcwd()
config['views_engine'] = 'sqlite:///%s' % self.db
try:
del config['sum_views']
except:
pass
self.view = DASViewManager(config)
def tearDown(self):
"""Invoke after each test"""
try:
os.remove(self.db)
except:
pass
def test_view1(self):
"""test view creation"""
self.view.create_db()
expect = 'find dataset, count(file)'
query = expect + ' where bla'
self.view.create('dataset', query)
result = self.view.get('dataset')
self.assertEqual(expect, result)
# test non-existing view
self.assertRaises(Exception, self.view.get, 'lll')
# test to check that we add the same view twice
query = 'find dataset, count(file), run'
self.assertRaises(Exception, self.view.create, ('dataset', query))
# test view update
query0 = 'find dataset, count(file), run'
expect = 'find dataset, count(file), sum(file.size)'
query = expect + ' where bla'
# first we should get exception that view doesn't exists
self.assertRaises(Exception, self.view.update, 'dataset')
self.view.create('dataset_test', query0)
self.view.update('dataset_test', query)
result = self.view.get('dataset_test')
self.assertEqual(expect, result)
def test_view2(self):
"""test to checkout all views"""
self.view.create_db()
query1 = u'find dataset, count(file) where bla'
self.view.create('dataset1', query1)
query2 = u'find dataset, count(file), sum(file) where bla'
self.view.create('dataset2', query2)
result = self.view.all()
expect = {u'dataset1': strip_query(query1),
u'dataset2': strip_query(query2)}
self.assertEqual(expect, result)
class testDAS(unittest.TestCase):
"""
A test class for the DASViewManager class
"""
def setUp(self):
"""
set up DAS core module
"""
self.db = 'test_views.db'
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config['logger'] = logger
config['verbose'] = debug
config['views_dir'] = os.getcwd()
config['views_engine'] = 'sqlite:///%s' % self.db
try:
del config['sum_views']
except:
pass
self.view = DASViewManager(config)
def tearDown(self):
"""Invoke after each test"""
try:
os.remove(self.db)
except:
pass
def test_view1(self):
"""test view creation"""
self.view.create_db()
expect = 'find dataset, count(file)'
query = expect + ' where bla'
self.view.create('dataset', query)
result = self.view.get('dataset')
self.assertEqual(expect, result)
# test non-existing view
self.assertRaises(Exception, self.view.get, 'lll')
# test to check that we add the same view twice
query = 'find dataset, count(file), run'
self.assertRaises(Exception, self.view.create, ('dataset', query))
# test view update
query0 = 'find dataset, count(file), run'
expect = 'find dataset, count(file), sum(file.size)'
query = expect + ' where bla'
# first we should get exception that view doesn't exists
self.assertRaises(Exception, self.view.update, 'dataset')
self.view.create('dataset_test', query0)
self.view.update('dataset_test', query)
result = self.view.get('dataset_test')
self.assertEqual(expect, result)
def test_view2(self):
"""test to checkout all views"""
self.view.create_db()
query1 = u'find dataset, count(file) where bla'
self.view.create('dataset1', query1)
query2 = u'find dataset, count(file), sum(file) where bla'
self.view.create('dataset2', query2)
result = self.view.all()
expect = {
u'dataset1': strip_query(query1),
u'dataset2': strip_query(query2)
}
self.assertEqual(expect, result)
