def test_get_connector_config(self): connector = TestConnector(self.dispatcher) self.assertDictEqual( { 'enabled': True }, connector.config) dispatcher = Dispatcher(Config(filename='test/configuration_examples/configuration.json'), auto_load=False) connector = TestConnector(dispatcher) self.assertDictEqual( { 'enabled': True, 'key': 'value' }, connector.config) connector2 = Test2Connector(dispatcher) self.assertDictEqual( { 'enabled': True }, connector2.config) connector3 = Test3Connector(dispatcher) dispatcher.get_connector_config(connector3)
def setUp(self): logging.basicConfig( filename='test/unittest.log', format="%(asctime)s - %(levelname)s - %(name)s:%(funcName)s - %(message)s", level=logging.DEBUG, filemode='w' ) self.dispatcher = Dispatcher( Config(filename='test/configuration_examples/simple.json'), auto_load=False )
def setUp(self): logging.basicConfig( filename='test/unittest.log', format= "%(asctime)s - %(levelname)s - %(name)s:%(funcName)s - %(message)s", level=logging.DEBUG, filemode='w') self.dispatcher = Dispatcher( Config(filename='test/configuration_examples/configuration.json')) self.connector = ObserviumConnector(self.dispatcher)
def setUp(self): logging.basicConfig( filename='test/unittest.log', format= "%(asctime)s - %(levelname)s - %(name)s:%(funcName)s - %(message)s", level=logging.DEBUG, filemode='w') self.dispatcher = Dispatcher( Config(filename='test/configuration_examples/configuration.json')) self.connector = SqliteConnector(self.dispatcher) self.connector.create_ptr_table() ptr = { 'ip_address': u'10.10.10.10', 'hostname': 'cmts-sc-1.domain.example', 'if_name': 'Ethernet0/0/0', 'ptr': 'cmts-sc-1-et0-0-0.domain.example' } self.connector.save_ptr(Ptr(**ptr))
def test_autoload(self): self.dispatcher = Dispatcher( Config(filename='test/configuration_examples/simple.json'), connectors_dir='/../test/connectors/modules' )
class TestDispatcher(unittest.TestCase): def setUp(self): logging.basicConfig( filename='test/unittest.log', format="%(asctime)s - %(levelname)s - %(name)s:%(funcName)s - %(message)s", level=logging.DEBUG, filemode='w' ) self.dispatcher = Dispatcher( Config(filename='test/configuration_examples/simple.json'), auto_load=False ) def test_register_connector(self): self.assertListEqual([], self.dispatcher.get_connector_list()) TestConnector(self.dispatcher) self.assertListEqual(['TestConnector'], self.dispatcher.get_connector_list()) Test2Connector(self.dispatcher) self.assertListEqual(['TestConnector', 'Test2Connector'], self.dispatcher.get_connector_list()) # Test3 doesn't have enabled flag set. Should be ignored Test3Connector(self.dispatcher) self.assertListEqual(['TestConnector', 'Test2Connector'], self.dispatcher.get_connector_list()) def test_connector_load(self): self.assertListEqual([], self.dispatcher.devices.keys()) TestConnector(self.dispatcher) self.dispatcher.load() # No invalid hostnames included self.assertListEqual( sorted(['cmts-1.domain.example.', 'cmts-2.domain.example.']), sorted(self.dispatcher.devices.keys()) ) Test2Connector(self.dispatcher) self.dispatcher.load() # No invalid hostnames included self.assertListEqual( sorted(['cmts-1.domain.example.', 'cmts-2.domain.example.', 'server.domain.example.']), sorted(self.dispatcher.devices.keys()) ) def test_autoload(self): self.dispatcher = Dispatcher( Config(filename='test/configuration_examples/simple.json'), connectors_dir='/../test/connectors/modules' ) def test_save_ptrs(self): TestConnector(self.dispatcher) Test2Connector(self.dispatcher) self.dispatcher.save_ptr(Ptr( ip_address=u'10.10.10.10', hostname='host.domain.example', if_name='Ethernet1/0', ptr='host-eth1/0.domain.example' )) self.dispatcher.save_ptrs({}) def test_get_connector_config(self): connector = TestConnector(self.dispatcher) self.assertDictEqual( { 'enabled': True }, connector.config) dispatcher = Dispatcher(Config(filename='test/configuration_examples/configuration.json'), auto_load=False) connector = TestConnector(dispatcher) self.assertDictEqual( { 'enabled': True, 'key': 'value' }, connector.config) connector2 = Test2Connector(dispatcher) self.assertDictEqual( { 'enabled': True }, connector2.config) connector3 = Test3Connector(dispatcher) dispatcher.get_connector_config(connector3)
action="store_true") parser.add_argument("-t", "--terse", help="terse output - don't display domains", action="store_true") args = parser.parse_args() hostname = args.hostname check_only = args.check diff_only = args.diff terse = args.terse config = Config(check_only=check_only, diff_only=diff_only, terse=terse) dispatcher = Dispatcher(config) print "Loaded connectors: %s" % ', '.join(dispatcher.get_connector_list()) dns = DnsCheck(config=config) output = TabularUtf8Output() fqdn = dns.get_fqdn(hostname) if fqdn: d = Device(hostname=fqdn, config=config) if d.get_interfaces(): d.check_ptrs() print output.display_device_detailed(d) # Filter all PTRs that don't have status equal to STATUS_NOT_UPDATED or STATUS_NOT_CREATED ptrs_for_update = {
args = parser.parse_args() check_only = args.check diff_only = not args.full terse = args.terse ptrs = {} failed_devices = [] interface_number = 0 ip_address_number = 0 config = Config(check_only=check_only, diff_only=diff_only, terse=terse) dispatcher = Dispatcher(config) output = TabularUtf8Output() print "Loaded connectors: %s" % ', '.join(dispatcher.get_connector_list()) dispatcher.load() total_devices = len(dispatcher.devices) print "Loaded %d device(s) from %d connector(s)" % ( total_devices, len(dispatcher.get_connector_list())) print "Fetching data from devices:" # Number of devices we've finished so far iterator = 0 for device in dispatcher.devices.keys(): # Create Device instance from hostname # dispatcher.devices[device] = Device(device, config, dispatcher.dns) # Fetch interfaces if dispatcher.devices[device].get_interfaces():
def test_simple(self): dispatcher = Dispatcher( Config(filename='test/configuration_examples/simple.json')) connector = ObserviumConnector(dispatcher) connector.load_devices()