import sys, os __dir = os.path.dirname(os.path.realpath(__file__)) sys.path.append(__dir + "/model") sys.path.append(__dir + "/utils") sys.path.append(__dir + "/collectors") from Config import Config from Comm import Comm import json config = Config(os.path.dirname(os.path.realpath(__file__)) + "/test/config.json") # TODO collectors = config.getCollectors() inventory = [] for collector_name in collectors: try: module_ = __import__(collector_name) collector = getattr(module_, collector_name)() collector.load() dataset = config.getData(collector_name) for data in dataset: inventory.append(collector.retrieveData()) except: print "No " + collector_name + " collector defined!" comm = Comm("localhost") comm.send(json.dumps(inventory))
def test_config(self): config = Config(os.path.dirname(os.path.realpath(__file__)) + '/config.json') self.assertTrue(len(config.getCollectors()) > 0) self.assertTrue(len(config.getData('SystemDataCollector')) > 0)