def devtree_exaples():
print('DEVTREE #1 : Simple devtree printing')
devtree = DevTree()
print(devtree)
print(devtree.get_text(format='csv'))
print(
devtree.get_text(format='prettytable',
fields=[
'ds_ip', 'client', 'last_time', 'model', 'name',
'parent_name', 'vendor'
]))
print('len : {}'.format(len(devtree)))
print('__iter__ print')
[print(str(f)[:80] + '.[..]') for f in devtree]
# print('repr : {}'.format(repr(devtree)))
print('DEVTREE #2 : Add a DataSource')
# Find the first device that's a receiver to use as the DataSource parent
ds_config = {}
for ds in devtree:
if ds['desc_id'] in ['2', '13']:
ds_config['parent_id'] = ds['ds_id']
ds_config['name'] = 'msiempy_test_datasource_delete_me'
ds_config['ds_ip'] = '0.20.5.5'
ds_config['type_id'] = '65'
print('Adding datasource...')
print('Result ID: ', devtree.add(ds_config))
print('DEVTREE #2.1 : DataSource details...')
ds = devtree.search('msiempy_test_datasource_delete_me')
if ds:
ds.load_details()
else:
print('New datasource not found. Waiting 15 seconds and rechecking...')
time.sleep(15)
devtree.refresh()
ds = devtree.search('msiempy_test_datasource_delete_me')
if ds:
ds.load_details()
print('DETAILS: {}'.format(ds.json))
print('DEVTREE #3 : Deleting DataSource')
for ds in devtree:
if ds['name'] == 'msiempy_test_datasource_delete_me':
print('Test datasource found. Deleting...')
ds = devtree[ds['idx']]
ds.delete()
continue
print('DEVTREE #3.1 : Verifying the datasource is gone')
for ds in devtree:
assert ds[
'name'] != 'msiempy_test_datasource_delete_me', "Looks like the datasource is still here :/"
def devtree_exaples():
print("DEVTREE #1 : Simple devtree printing")
devtree = DevTree()
print(devtree)
print(devtree.get_text(format="csv"))
print(
devtree.get_text(
format="prettytable",
fields=[
"ds_ip",
"client",
"last_time",
"model",
"name",
"parent_name",
"vendor",
],
))
print("len : {}".format(len(devtree)))
print("__iter__ print")
[print(str(f)[:80] + ".[..]") for f in devtree]
# print('repr : {}'.format(repr(devtree)))
print("DEVTREE #2 : Add a DataSource")
# Find the first device that's a receiver to use as the DataSource parent
ds_config = {}
for ds in devtree:
if ds["desc_id"] in ["2", "13"]:
ds_config["parent_id"] = ds["ds_id"]
ds_config["name"] = "msiempy_test_datasource_delete_me"
ds_config["ds_ip"] = "0.20.5.5"
ds_config["type_id"] = "65"
print("Adding datasource...")
print("Result ID: ", devtree.add(ds_config))
print("DEVTREE #2.1 : DataSource details...")
ds = devtree.search("msiempy_test_datasource_delete_me")
if ds:
ds.load_details()
else:
print("New datasource not found. Waiting 15 seconds and rechecking...")
time.sleep(15)
devtree.refresh()
ds = devtree.search("msiempy_test_datasource_delete_me")
if ds:
ds.load_details()
print("DETAILS: {}".format(ds.json))
print("DEVTREE #3 : Deleting DataSource")
for ds in devtree:
if ds["name"] == "msiempy_test_datasource_delete_me":
print("Test datasource found. Deleting...")
ds = devtree[ds["idx"]]
ds.delete()
continue
print("DEVTREE #3.1 : Verifying the datasource is gone")
for ds in devtree:
assert (ds["name"] != "msiempy_test_datasource_delete_me"
), "Looks like the datasource is still here :/"
class T(unittest.TestCase):
def test_esm(self):
esm = ESM()
print(
str(esm.status()) + '\n' + str(esm.buildstamp()) + '\n' +
str(esm.recs()))
def print_devtree(self):
print('DEVTREE #1 : Simple devtree printing')
devtree = DevTree()
print(devtree)
#print(list(devtree))
print(devtree.get_text(format='csv'))
print(
devtree.get_text(format='prettytable',
fields=[
'ds_ip', 'client', 'last_time', 'model',
'name', 'parent_name', 'vendor'
]))
def setUp(self):
print('Refreshing Devtree()')
self.devtree = DevTree()
def test_devtree1_has_devices(self):
print('1. Testing that devtree has some devices...')
self.assertGreater(len(self.devtree), 0)
def test_devtree2_esm_at_top(self):
print('2. Make sure there is an ESM at the top...')
print(self.devtree[0]['name'])
self.assertIn('ESM', self.devtree[0]['name'])
def test_devtree3_datasource_cast(self):
print('3. Make sure datasources are casted as DataSources...')
for ds in self.devtree:
if ds['desc_id'] == '3':
datasource = self.devtree[ds['idx']]
self.assertTrue(type(datasource) is msiempy.device.DataSource)
print('DataSource Type Matches:',
type(datasource) is msiempy.device.DataSource)
break
def test_devtree4_check_old_test_datasource(self):
print('4. Checking for old test datasource...')
for ds in self.devtree:
if ds['name'] == 'msiempy_test_datasource_delete_me':
print('Old test datasource found. Deleting...')
ds = self.devtree[ds['idx']]
ds.delete()
continue
def test_devtree5_add_datasource(self):
ds_config = {}
for ds in self.devtree:
if ds['desc_id'] in ['2', '13']:
ds_config['parent_id'] = ds['ds_id']
ds_config['name'] = 'msiempy_test_datasource_delete_me'
ds_config['ds_ip'] = '0.20.5.5'
ds_config['type_id'] = '65'
print('5. Adding datasource...')
print('Result ID: ', self.devtree.add(ds_config))
def test_devtree6_load_ds_details(self):
print('6. Loading DataSource details...')
ds = self.devtree.search('msiempy_test_datasource_delete_me')
if ds:
ds.load_details()
else:
print(
'New datasource not found. Waiting 15 seconds and rechecking...'
)
time.sleep(15)
self.devtree.refresh()
ds = self.devtree.search('msiempy_test_datasource_delete_me')
if ds:
ds.load_details()
print('DETAILS: {}'.format(ds.json))
def test_devtree7_del_datasource(self):
print('7. Deleting Datasource...')
for ds in self.devtree:
if ds['name'] == 'msiempy_test_datasource_delete_me':
print('Test datasource found. Deleting...')
ds = self.devtree[ds['idx']]
ds.delete()
continue
def test_devtree8_verify_deleted(self):
print('8. Verifying the datasource is gone...')
for ds in self.devtree:
self.assertNotEqual(ds['name'],
'msiempy_test_datasource_delete_me')
class T(unittest.TestCase):
def test_esm(self):
esm = ESM()
print(
str(esm.status()) + "\n" + str(esm.buildstamp()) + "\n" +
str(esm.recs()))
def print_devtree(self):
print("DEVTREE #1 : Simple devtree printing")
devtree = DevTree()
print(devtree)
# print(list(devtree))
print(devtree.get_text(format="csv"))
print(
devtree.get_text(
format="prettytable",
fields=[
"ds_ip",
"client",
"last_time",
"model",
"name",
"parent_name",
"vendor",
],
))
def setUp(self):
print("Refreshing Devtree()")
self.devtree = DevTree()
def test_devtree1_has_devices(self):
print("1. Testing that devtree has some devices...")
self.assertGreater(len(self.devtree), 0)
def test_devtree2_esm_at_top(self):
print("2. Make sure there is an ESM at the top...")
print(self.devtree[0]["name"])
self.assertIn("ESM", self.devtree[0]["name"])
def test_devtree3_datasource_cast(self):
print("3. Make sure datasources are casted as DataSources...")
for ds in self.devtree:
if ds["desc_id"] == "3":
datasource = self.devtree[ds["idx"]]
self.assertTrue(type(datasource) is msiempy.device.DataSource)
print(
"DataSource Type Matches:",
type(datasource) is msiempy.device.DataSource,
)
break
def test_devtree4_check_old_test_datasource(self):
print("4. Checking for old test datasource...")
for ds in self.devtree:
if ds["name"] == "msiempy_test_datasource_delete_me":
print("Old test datasource found. Deleting...")
ds = self.devtree[ds["idx"]]
ds.delete()
continue
def test_devtree5_add_datasource(self):
ds_config = {}
for ds in self.devtree:
if ds["desc_id"] in ["2", "13"]:
ds_config["parent_id"] = ds["ds_id"]
ds_config["name"] = "msiempy_test_datasource_delete_me"
ds_config["ds_ip"] = "0.20.5.5"
ds_config["type_id"] = "65"
print("5. Adding datasource...")
print("Result ID: ", self.devtree.add(ds_config))
def test_devtree6_load_ds_details(self):
print("6. Loading DataSource details...")
ds = self.devtree.search("msiempy_test_datasource_delete_me")
if ds:
ds.load_details()
else:
print(
"New datasource not found. Waiting 15 seconds and rechecking..."
)
time.sleep(15)
self.devtree.refresh()
ds = self.devtree.search("msiempy_test_datasource_delete_me")
if ds:
ds.load_details()
print("DETAILS: {}".format(ds.json))
def test_devtree7_del_datasource(self):
print("7. Deleting Datasource...")
for ds in self.devtree:
if ds["name"] == "msiempy_test_datasource_delete_me":
print("Test datasource found. Deleting...")
ds = self.devtree[ds["idx"]]
ds.delete()
continue
def test_devtree8_verify_deleted(self):
print("8. Verifying the datasource is gone...")
for ds in self.devtree:
self.assertNotEqual(ds["name"],
"msiempy_test_datasource_delete_me")