def get_upload_algorithms():
"""
:return: List of StoredAdapter entities
"""
categories = dao.get_uploader_categories()
categories_ids = [categ.id for categ in categories]
return dao.get_adapters_from_categories(categories_ids)
def test_introspect(self):
"""
Test that expected categories and groups are found in DB after introspection.
We also check algorithms introspected during base_testcase.init_test_env
"""
all_categories = dao.get_algorithm_categories()
category_ids = [
cat.id for cat in all_categories
if cat.displayname == "AdaptersTest"
]
adapters = dao.get_adapters_from_categories(category_ids)
self.assertEqual(8, len(adapters), "Introspection failed!")
nr_adapters_mod2 = 0
for algorithm in adapters:
self.assertTrue(
algorithm.module in [
'tvb.tests.framework.adapters.testadapter1',
'tvb.tests.framework.adapters.testadapter2',
'tvb.tests.framework.adapters.testadapter3',
'tvb.tests.framework.adapters.ndimensionarrayadapter',
'tvb.tests.framework.adapters.testgroupadapter'
], "Unknown Adapter module:" + str(algorithm.module))
self.assertTrue(
algorithm.classname in [
"TestAdapter1", "TestAdapterDatatypeInput", "TestAdapter2",
"TestAdapter22", "TestAdapterHugeMemoryRequired",
"TestAdapter3", "TestAdapterHDDRequired",
"NDimensionArrayAdapter"
], "Unknown Adapter Class:" + str(algorithm.classname))
if algorithm.module == 'tvb.tests.framework.adapters.testadapter2':
nr_adapters_mod2 += 1
self.assertEqual(nr_adapters_mod2, 2)
def get_upload_algorithms():
"""
:return: List of StoredAdapter entities
"""
categories = dao.get_uploader_categories()
categories_ids = [categ.id for categ in categories]
return dao.get_adapters_from_categories(categories_ids)
def test_introspect(self):
"""
Test that expected categories and groups are found in DB after introspection.
We also check algorithms introspected during base_testcase.init_test_env
"""
all_categories = dao.get_algorithm_categories()
category_ids = [cat.id for cat in all_categories if cat.displayname == "AdaptersTest"]
adapters = dao.get_adapters_from_categories(category_ids)
self.assertEqual(8, len(adapters), "Introspection failed!")
nr_adapters_mod2 = 0
for algorithm in adapters:
self.assertTrue(algorithm.module in ['tvb.tests.framework.adapters.testadapter1',
'tvb.tests.framework.adapters.testadapter2',
'tvb.tests.framework.adapters.testadapter3',
'tvb.tests.framework.adapters.ndimensionarrayadapter',
'tvb.tests.framework.adapters.testgroupadapter'],
"Unknown Adapter module:" + str(algorithm.module))
self.assertTrue(algorithm.classname in ["TestAdapter1", "TestAdapterDatatypeInput",
"TestAdapter2", "TestAdapter22", "TestAdapterHugeMemoryRequired",
"TestAdapter3", "TestAdapterHDDRequired",
"NDimensionArrayAdapter"
],
"Unknown Adapter Class:" + str(algorithm.classname))
if algorithm.module == 'tvb.tests.framework.adapters.testadapter2':
nr_adapters_mod2 += 1
self.assertEqual(nr_adapters_mod2, 2)
def test_default_cancel(self):
"""
On cancel we should get a redirect to the back page link.
"""
cherrypy.request.method = "POST"
categories = dao.get_algorithm_categories()
algo_groups = dao.get_adapters_from_categories([categories[0].id])
self._expect_redirect('/project/viewoperations/%i' % self.test_project.id, self.flow_c.default,
categories[0].id, algo_groups[0].id, cancel=True, back_page='operations')
def test_default_cancel(self):
"""
On cancel we should get a redirect to the back page link.
"""
cherrypy.request.method = "POST"
categories = dao.get_algorithm_categories()
algo_groups = dao.get_adapters_from_categories([categories[0].id])
self._expect_redirect('/project/viewoperations/%i' % self.test_project.id, self.flow_c.default,
categories[0].id, algo_groups[0].id, cancel=True, back_page='operations')
def teardown_method(self):
all_categories = dao.get_algorithm_categories()
category_ids = [
cat.id for cat in all_categories
if cat.displayname == TestCategory.category_name
]
adapters = dao.get_adapters_from_categories(category_ids)
for algorithm in adapters:
dao.remove_entity(Algorithm, algorithm.id)
dao.remove_entity(AlgorithmCategory, category_ids[0])
def test_default(self):
"""
Test default method from step controllers. Check that the submit link is ok, that a mainContent
is present in result dict and that the isAdapter flag is set to true.
"""
cherrypy.request.method = "GET"
categories = dao.get_algorithm_categories()
for categ in categories:
algo_groups = dao.get_adapters_from_categories([categ.id])
for algo in algo_groups:
result_dict = self.flow_c.default(categ.id, algo.id)
assert result_dict[common.KEY_SUBMIT_LINK] == '/flow/%i/%i' % (categ.id, algo.id)
assert 'mainContent' in result_dict
assert result_dict['isAdapter']
def test_default(self):
"""
Test default method from step controllers. Check that the submit link is ok, that a mainContent
is present in result dict and that the isAdapter flag is set to true.
"""
cherrypy.request.method = "GET"
categories = dao.get_algorithm_categories()
for categ in categories:
algo_groups = dao.get_adapters_from_categories([categ.id])
for algo in algo_groups:
result_dict = self.flow_c.default(categ.id, algo.id)
self.assertEqual(result_dict[common.KEY_SUBMIT_LINK], '/flow/%i/%i' % (categ.id, algo.id))
self.assertTrue('mainContent' in result_dict)
self.assertTrue(result_dict['isAdapter'])
def get_analyze_groups(self):
"""
:return: list of AlgorithmTransientGroup entities
"""
categories = dao.get_launchable_categories(elimin_viewers=True)
categories_ids = [categ.id for categ in categories]
stored_adapters = dao.get_adapters_from_categories(categories_ids)
groups_list = []
for adapter in stored_adapters:
# For empty groups, this time, we fill the actual adapter
group = AlgorithmTransientGroup(adapter.group_name or adapter.displayname,
adapter.group_description or adapter.description)
group = self._find_group(groups_list, group)
group.children.append(adapter)
return categories[0], groups_list
def get_analyze_groups(self):
"""
:return: list of AlgorithmTransientGroup entities
"""
categories = dao.get_launchable_categories(elimin_viewers=True)
categories_ids = [categ.id for categ in categories]
stored_adapters = dao.get_adapters_from_categories(categories_ids)
groups_list = []
for adapter in stored_adapters:
# For empty groups, this time, we fill the actual adapter
group = AlgorithmTransientGroup(adapter.group_name or adapter.displayname,
adapter.group_description or adapter.description)
group = self._find_group(groups_list, group)
group.children.append(adapter)
return categories[0], groups_list
def test_default(self):
"""
Test default method from step controllers. Check that the submit link is ok, that a mainContent
is present in result dict and that the isAdapter flag is set to true.
"""
cherrypy.request.method = "GET"
categories = dao.get_algorithm_categories()
for categ in categories:
# Ignore creators, as those won't go through this flow
if categ.displayname in [CreateAlgorithmCategoryConfig.category_name]:
continue
algo_groups = dao.get_adapters_from_categories([categ.id])
for algo in algo_groups:
result_dict = self.flow_c.default(categ.id, algo.id)
assert result_dict[common.KEY_SUBMIT_LINK] == '/flow/%i/%i' % (categ.id, algo.id)
assert 'mainContent' in result_dict
assert result_dict['isAdapter']
def test_introspect(self):
"""
Test that expected categories and groups are found in DB after introspection.
We also check algorithms introspected during base_testcase.init_test_env
"""
introspector = Introspector()
introspector.introspection_registry.ADAPTERS[
TestCategory] = import_adapters(tvb.tests.framework.adapters,
ALL_TEST_ADAPTERS)
introspector.introspect()
all_categories = dao.get_algorithm_categories()
category_ids = [
cat.id for cat in all_categories
if cat.displayname == TestCategory.category_name
]
adapters = dao.get_adapters_from_categories(category_ids)
assert 5 == len(adapters), "Introspection failed!"
nr_adapters_mod3 = 0
for algorithm in adapters:
assert algorithm.module in [
'tvb.tests.framework.adapters.testadapter1',
'tvb.tests.framework.adapters.testadapter2',
'tvb.tests.framework.adapters.testadapter3'
], "Unknown Adapter Module:" + str(algorithm.module)
assert algorithm.classname in [
"TestAdapter1", "TestAdapter2",
"TestAdapterHugeMemoryRequired", "TestAdapter3",
"TestAdapterHDDRequired"
], "Unknown Adapter Class:" + str(algorithm.classname)
if algorithm.module == 'tvb.tests.framework.adapters.testadapter3':
nr_adapters_mod3 += 1
assert nr_adapters_mod3 == 3
