def load_tests(loader, tests, pattern): # pylint: disable=W0613 try: log.info('Start ndb tests') suite = unittest.TestSuite() test_list = get_test_list(TEST_DIR) log.info('Test list is %s', test_list) for test in test_list: log.info('Starting test harness %s', test) harness = yaml.load(open(test)) if harness.get('debug'): enable_handler_console() tests = harness.get('tests', ['topology']) name = harness.get('ndb', 'neighbordb') filename = name.split('/')[-1] if filename in harness: ndb = harness.get(filename) else: assert os.path.exists(name) ndb = yaml.load(open(name)) kwargs = dict() kwargs['valid_patterns'] = harness.get('valid_patterns', dict()) kwargs['invalid_patterns'] = harness.get('invalid_patterns') kwargs['tag'] = harness.get('tag', filename) kwargs['filename'] = test for test in tests: assert test in ['pattern', 'topology'] suite.addTest(NeighbordbTest('neighbordb_%s' % test, ndb, **kwargs)) if 'nodes' in harness and 'topology' in tests: for key in harness['nodes'].keys(): assert key in ['pass', 'fail'] entries = harness['nodes'][key] if not entries is None: for entry in entries: filename = entry['name'].split('/')[-1] kwargs['node'] = harness.get(filename, entry['name']) kwargs['match'] = entry.get('match') kwargs['tag'] = '%s:%s' % (harness.get('tag'), filename) log.info('Adding node %s', name) suite.addTest(NeighbordbTest('node_%s' % key, ndb, **kwargs)) except Exception as exc: # pylint: disable=W0703 log.exception('Unexpected error trying to execute load_tests: %s' % exc) else: return suite
def enable_logging(): enable_handler_console()
def load_tests(loader, tests, pattern): #pylint: disable=W0613 suite = unittest.TestSuite() test_list = os.environ.get('TESTS', None) if not test_list: test_list = [f for f in os.listdir(TEST_DIR) if os.path.join(TEST_DIR, f).endswith('_test')] else: test_list = test_list.split(',') print test_list for test in test_list: print 'Starting test %s' % test definition = yaml.load(open(os.path.join(TEST_DIR, test))) nodes = definition.get('nodes', []) for node in nodes: print 'Adding test: %s' % node['name'] suite.addTest(NodeTest(test, node, definition['neighbordb'])) if definition.get('configured_neighbordb', None): suite.addTest(NeighbordbTest(test, definition['neighbordb'], definition['configured_neighbordb'])) return suite if __name__ == '__main__': enable_handler_console() unittest.main()
import ztpserver.neighbordb import ztpserver.topology import ztpserver.controller import ztpserver.repository import ztpserver.config from ztpserver.app import enable_handler_console from ztpserver.serializers import SerializerError from server_test_lib import remove_all, random_string from server_test_lib import ztp_headers, write_file from server_test_lib import create_definition, create_attributes, create_node enable_handler_console(level='DEBUG') def reload_mocked_modules(): reload(ztpserver.neighbordb) reload(ztpserver.config) reload(ztpserver.repository) reload(ztpserver.topology) reload(ztpserver.controller) class RouterTests(unittest.TestCase): def match_routes(self, url, valid, invalid): request = Request.blank(url) router = ztpserver.controller.Router()