def setUp(self):
session.commit()
drop_all()
create_all()
session.add(Host(id=1)) # Contains alerts 1,2,4
session.add(Host(id=2)) # Contains alert 6,7
for i in range(1,8):
session.add(Alert(id=i, name=i))
session.add(HostGroup(id=1))
session.add(HostGroup(id=2))
session.add(HostGroupAssignment(member_host_id=1, host_group_id=1))
session.add(Service(id=1))
session.add(Service(id=2))
session.add(Service(id=3))
session.add(ServiceDependency(host_id=1, service_id=1))
session.add(RedundancyGroup(id=1, service_id=3))
session.add(RedundancyGroupComponent(host_id=2, redundancy_group_id=1))
session.add(ServiceDependency(redundancy_group_id=1, service_id=3))
session.add(AlertCheckEntity(host_id=1, alert_id=1))
session.add(AlertCheckEntity(host_group_id=1, alert_id=2))
session.add(AlertCheckEntity(host_group_id=2, alert_id=3))
session.add(AlertCheckEntity(service_id=1, alert_id=4))
session.add(AlertCheckEntity(service_id=2, alert_id=5))
session.add(AlertCheckEntity(host_id=2, alert_id=6))
session.add(AlertCheckEntity(service_id=3, alert_id=7))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
session.add(Host(id=1)) # Contains alerts 1,2,4
session.add(Host(id=2)) # Contains alert 6,7
for i in range(1, 8):
session.add(Alert(id=i, name=i))
session.add(HostGroup(id=1))
session.add(HostGroup(id=2))
session.add(HostGroupAssignment(member_host_id=1, host_group_id=1))
session.add(Service(id=1))
session.add(Service(id=2))
session.add(Service(id=3))
session.add(ServiceDependency(host_id=1, service_id=1))
session.add(RedundancyGroup(id=1, service_id=3))
session.add(RedundancyGroupComponent(host_id=2, redundancy_group_id=1))
session.add(ServiceDependency(redundancy_group_id=1, service_id=3))
session.add(AlertCheckEntity(host_id=1, alert_id=1))
session.add(AlertCheckEntity(host_group_id=1, alert_id=2))
session.add(AlertCheckEntity(host_group_id=2, alert_id=3))
session.add(AlertCheckEntity(service_id=1, alert_id=4))
session.add(AlertCheckEntity(service_id=2, alert_id=5))
session.add(AlertCheckEntity(host_id=2, alert_id=6))
session.add(AlertCheckEntity(service_id=3, alert_id=7))
session.commit()
def setUp(self):
create_all()
# Insert all plugins
for letter in ["A", "B", "C", "D"]:
p = Plugin(id=letter, name=letter)
session.add(p)
g = HostGroup(name="G")
session.add(g)
g2 = HostGroup(name="G2")
session.add(g2)
# Insert all hosts
for number in range(1, 8):
h = Host(name=number)
session.add(h)
session.flush()
if number in [3, 4]:
hga = HostGroupAssignment(member_host_id=h.id,
host_group_id=g.id)
session.add(hga)
if number == 7:
hga = HostGroupAssignment(member_host_id=h.id,
host_group_id=g2.id)
session.add(hga)
hga = HostGroupAssignment(member_host_group_id=g.id,
host_group_id=g2.id)
session.add(hga)
# Insert all checks and do assignments
for number in range(1, 6):
c = Check(name=number)
session.add(c)
session.add(CheckAssignment(check_id=1, host_id=1))
session.add(CheckAssignment(check_id=1, host_id=2))
session.add(CheckAssignment(check_id=1, host_id=4))
session.add(CheckAssignment(check_id=2, host_id=3))
session.add(CheckAssignment(check_id=4, host_id=5))
session.add(CheckAssignment(check_id=3, host_group_id=1))
session.add(CheckAssignment(check_id=5, host_group_id=2))
session.add(CheckPlugin(check_id=1, plugin_id=2))
session.add(CheckPlugin(check_id=1, plugin_id=3))
session.add(CheckPlugin(check_id=2, plugin_id=2))
session.add(CheckPlugin(check_id=3, plugin_id=1))
session.add(CheckPlugin(check_id=4, plugin_id=1))
session.add(CheckPlugin(check_id=4, plugin_id=2))
session.add(CheckPlugin(check_id=4, plugin_id=3))
session.add(CheckPlugin(check_id=5, plugin_id=4))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
session.add(Alert(id=1))
session.add(Alert(id=2))
session.add(Check(id=1))
session.add(Plugin(id=1))
session.add(AlertRestrictToEntity(alert_id=1, plugin_id=1))
session.add(AlertRestrictToEntity(alert_id=1, check_id=1))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
session.add(Alert(id=1))
session.add(Alert(id=2))
session.add(Check(id=1))
session.add(Plugin(id=1))
session.add(AlertRestrictToEntity(alert_id=1, plugin_id=1))
session.add(AlertRestrictToEntity(alert_id=1, check_id=1))
session.commit()
def setUp(self):
create_all()
# Insert all plugins
for letter in ["A", "B", "C", "D"]:
p = Plugin(id=letter, name=letter)
session.add(p)
g = HostGroup(name="G")
session.add(g)
g2 = HostGroup(name="G2")
session.add(g2)
# Insert all hosts
for number in range(1, 8):
h = Host(name=number)
session.add(h)
session.flush()
if number in [3, 4]:
hga = HostGroupAssignment(member_host_id=h.id, host_group_id=g.id)
session.add(hga)
if number == 7:
hga = HostGroupAssignment(member_host_id=h.id, host_group_id=g2.id)
session.add(hga)
hga = HostGroupAssignment(member_host_group_id=g.id, host_group_id=g2.id)
session.add(hga)
# Insert all checks and do assignments
for number in range(1, 6):
c = Check(name=number)
session.add(c)
session.add(CheckAssignment(check_id=1, host_id=1))
session.add(CheckAssignment(check_id=1, host_id=2))
session.add(CheckAssignment(check_id=1, host_id=4))
session.add(CheckAssignment(check_id=2, host_id=3))
session.add(CheckAssignment(check_id=4, host_id=5))
session.add(CheckAssignment(check_id=3, host_group_id=1))
session.add(CheckAssignment(check_id=5, host_group_id=2))
session.add(CheckPlugin(check_id=1, plugin_id=2))
session.add(CheckPlugin(check_id=1, plugin_id=3))
session.add(CheckPlugin(check_id=2, plugin_id=2))
session.add(CheckPlugin(check_id=3, plugin_id=1))
session.add(CheckPlugin(check_id=4, plugin_id=1))
session.add(CheckPlugin(check_id=4, plugin_id=2))
session.add(CheckPlugin(check_id=4, plugin_id=3))
session.add(CheckPlugin(check_id=5, plugin_id=4))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
# Add Hosts
for i in range(1,5):
session.add(Host(id=i, name="Host {}".format(i)))
# Add Groups
for i in range(1,9):
session.add(HostGroup(id=i, name="Group {}".format(i)))
session.flush()
# Assign hosts to groups
session.add(HostGroupAssignment(member_host_id=1, host_group_id=1))
session.add(HostGroupAssignment(member_host_id=1, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=2, host_group_id=1))
session.add(HostGroupAssignment(member_host_id=2, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=2, host_group_id=3))
session.add(HostGroupAssignment(member_host_id=3, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=3, host_group_id=3))
session.add(HostGroupAssignment(member_host_id=3, host_group_id=5))
session.add(HostGroupAssignment(member_host_id=4, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=4, host_group_id=6))
session.add(HostGroupAssignment(member_host_id=4, host_group_id=5))
# Assign groups to groups
session.add(HostGroupAssignment(member_host_group_id=4,
host_group_id=5))
session.add(HostGroupAssignment(member_host_group_id=3,
host_group_id=4))
session.add(HostGroupAssignment(member_host_group_id=6,
host_group_id=7))
session.add(HostGroupAssignment(member_host_group_id=7,
host_group_id=8))
session.add(HostGroupAssignment(member_host_group_id=8,
host_group_id=6))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
# Add Hosts
for i in range(1, 5):
session.add(Host(id=i, name="Host {}".format(i)))
# Add Groups
for i in range(1, 9):
session.add(HostGroup(id=i, name="Group {}".format(i)))
session.flush()
# Assign hosts to groups
session.add(HostGroupAssignment(member_host_id=1, host_group_id=1))
session.add(HostGroupAssignment(member_host_id=1, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=2, host_group_id=1))
session.add(HostGroupAssignment(member_host_id=2, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=2, host_group_id=3))
session.add(HostGroupAssignment(member_host_id=3, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=3, host_group_id=3))
session.add(HostGroupAssignment(member_host_id=3, host_group_id=5))
session.add(HostGroupAssignment(member_host_id=4, host_group_id=2))
session.add(HostGroupAssignment(member_host_id=4, host_group_id=6))
session.add(HostGroupAssignment(member_host_id=4, host_group_id=5))
# Assign groups to groups
session.add(
HostGroupAssignment(member_host_group_id=4, host_group_id=5))
session.add(
HostGroupAssignment(member_host_group_id=3, host_group_id=4))
session.add(
HostGroupAssignment(member_host_group_id=6, host_group_id=7))
session.add(
HostGroupAssignment(member_host_group_id=7, host_group_id=8))
session.add(
HostGroupAssignment(member_host_group_id=8, host_group_id=6))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
# Insert all plugins
for letter in ["A", "B", "C"]:
session.add(Plugin(id=letter, name=letter))
# Insert all hosts
for number in range(1,9):
session.add(Host(name=number))
# Mock the assigned_plugins method
Host.assigned_plugins = property(lambda s: Plugin.query.all())
# All plugins OK
session.add(PluginResult(host_id=1, plugin_id="A",
health_status="ok"))
session.add(PluginResult(host_id=1, plugin_id="B",
health_status="ok"))
session.add(PluginResult(host_id=1, plugin_id="C",
health_status="ok"))
# All plugins critical
session.add(PluginResult(host_id=2, plugin_id="A",
health_status="critical"))
session.add(PluginResult(host_id=2, plugin_id="B",
health_status="critical"))
session.add(PluginResult(host_id=2, plugin_id="C",
health_status="critical"))
# All plugins unknown
session.add(PluginResult(host_id=3, plugin_id="A",
health_status="unknown"))
session.add(PluginResult(host_id=3, plugin_id="B",
health_status="unknown"))
session.add(PluginResult(host_id=3, plugin_id="C",
health_status="unknown"))
# One plugin critical
session.add(PluginResult(host_id=4, plugin_id="A",
health_status="ok"))
session.add(PluginResult(host_id=4, plugin_id="B",
health_status="critical"))
session.add(PluginResult(host_id=4, plugin_id="C",
health_status="ok"))
# One plugin unknown
session.add(PluginResult(host_id=5, plugin_id="A",
health_status="ok"))
session.add(PluginResult(host_id=5, plugin_id="B",
health_status="ok"))
session.add(PluginResult(host_id=5, plugin_id="C",
health_status="unknown"))
# All plugins no_data (Host 6)
# One plugin no data
session.add(PluginResult(host_id=7, plugin_id="A",
health_status="ok"))
session.add(PluginResult(host_id=7, plugin_id="C",
health_status="minor"))
# Mixed results
session.add(PluginResult(host_id=8, plugin_id="A",
health_status="ok"))
session.add(PluginResult(host_id=8, plugin_id="B",
health_status="major"))
session.add(PluginResult(host_id=8, plugin_id="C",
health_status="minor"))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
session.add(Plugin(id="Plugin", name="Plugin"))
# Insert all hosts
for number in range(1,7):
session.add(Host(name=number))
# Mock the assigned_plugins method
Host.assigned_plugins = property(lambda s: Plugin.query.all())
session.add(PluginResult(host_id=1, plugin_id="Plugin",
health_status="ok"))
session.add(PluginResult(host_id=2, plugin_id="Plugin",
health_status="minor"))
session.add(PluginResult(host_id=3, plugin_id="Plugin",
health_status="major"))
session.add(PluginResult(host_id=4, plugin_id="Plugin",
health_status="critical"))
session.add(PluginResult(host_id=5, plugin_id="Plugin",
health_status="unknown"))
# Host ID 6 has no data
session.add(HostGroup(id=1))
session.add(HostGroupAssignment(host_group_id=1, member_host_id=1))
session.add(HostGroupAssignment(host_group_id=1, member_host_id=4))
for number in range(1,11):
session.add(Service(id=number, name=number))
# test_failed_dependency
session.add(ServiceDependency(service_id=1, host_id=1))
session.add(ServiceDependency(service_id=1, host_id=1))
session.add(ServiceDependency(service_id=1, host_id=3))
# test_ok_dependency
session.add(ServiceDependency(service_id=2, host_id=1))
session.add(ServiceDependency(service_id=2, host_id=1))
session.add(ServiceDependency(service_id=2, host_id=1))
# test_failure_in_redundancy_group
session.add(RedundancyGroup(id=1, service_id=3))
session.add(RedundancyGroupComponent(redundancy_group_id=1, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=1, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=1, host_id=4))
session.add(ServiceDependency(service_id=3, redundancy_group_id=1))
# test_failed_redundancy_group
session.add(RedundancyGroup(id=2, service_id=4))
session.add(RedundancyGroupComponent(redundancy_group_id=2, host_id=2))
session.add(RedundancyGroupComponent(redundancy_group_id=2, host_id=2))
session.add(RedundancyGroupComponent(redundancy_group_id=2, host_id=4))
session.add(ServiceDependency(service_id=4, redundancy_group_id=2))
# redundancy_group_ok
session.add(RedundancyGroup(id=3, service_id=5))
session.add(RedundancyGroupComponent(redundancy_group_id=3, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=3, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=3, host_id=1))
session.add(ServiceDependency(service_id=5, redundancy_group_id=3))
# test_failed_dependency_with_redundancy_group
session.add(RedundancyGroup(id=4, service_id=6))
session.add(RedundancyGroupComponent(redundancy_group_id=4, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=4, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=4, host_id=1))
session.add(ServiceDependency(service_id=6, redundancy_group_id=4))
session.add(ServiceDependency(service_id=6, host_id=3))
# test_all_dependency_failed
session.add(ServiceDependency(service_id=7, host_id=3))
session.add(ServiceDependency(service_id=7, host_id=3))
session.add(ServiceDependency(service_id=7, host_id=4))
# test_no_data
session.add(ServiceDependency(service_id=8, host_id=1))
session.add(ServiceDependency(service_id=8, host_id=1))
session.add(ServiceDependency(service_id=8, host_id=6))
# test_host_group_dependency
session.add(ServiceDependency(service_id=9, host_group_id=1))
# test_host_group_redundancy
session.add(RedundancyGroup(id=5, service_id=10))
session.add(RedundancyGroupComponent(redundancy_group_id=5,
host_group_id=1))
session.add(ServiceDependency(service_id=10, redundancy_group_id=5))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
# Insert all plugins
for letter in ["A", "B", "C"]:
session.add(Plugin(id=letter, name=letter))
# Insert all hosts
for number in range(1, 9):
session.add(Host(name=number))
# Mock the assigned_plugins method
Host.assigned_plugins = property(lambda s: Plugin.query.all())
# All plugins OK
session.add(PluginResult(host_id=1, plugin_id="A", health_status="ok"))
session.add(PluginResult(host_id=1, plugin_id="B", health_status="ok"))
session.add(PluginResult(host_id=1, plugin_id="C", health_status="ok"))
# All plugins critical
session.add(
PluginResult(host_id=2, plugin_id="A", health_status="critical"))
session.add(
PluginResult(host_id=2, plugin_id="B", health_status="critical"))
session.add(
PluginResult(host_id=2, plugin_id="C", health_status="critical"))
# All plugins unknown
session.add(
PluginResult(host_id=3, plugin_id="A", health_status="unknown"))
session.add(
PluginResult(host_id=3, plugin_id="B", health_status="unknown"))
session.add(
PluginResult(host_id=3, plugin_id="C", health_status="unknown"))
# One plugin critical
session.add(PluginResult(host_id=4, plugin_id="A", health_status="ok"))
session.add(
PluginResult(host_id=4, plugin_id="B", health_status="critical"))
session.add(PluginResult(host_id=4, plugin_id="C", health_status="ok"))
# One plugin unknown
session.add(PluginResult(host_id=5, plugin_id="A", health_status="ok"))
session.add(PluginResult(host_id=5, plugin_id="B", health_status="ok"))
session.add(
PluginResult(host_id=5, plugin_id="C", health_status="unknown"))
# All plugins no_data (Host 6)
# One plugin no data
session.add(PluginResult(host_id=7, plugin_id="A", health_status="ok"))
session.add(
PluginResult(host_id=7, plugin_id="C", health_status="minor"))
# Mixed results
session.add(PluginResult(host_id=8, plugin_id="A", health_status="ok"))
session.add(
PluginResult(host_id=8, plugin_id="B", health_status="major"))
session.add(
PluginResult(host_id=8, plugin_id="C", health_status="minor"))
session.commit()
def setUp(self):
session.commit()
drop_all()
create_all()
session.add(Plugin(id="Plugin", name="Plugin"))
# Insert all hosts
for number in range(1, 7):
session.add(Host(name=number))
# Mock the assigned_plugins method
Host.assigned_plugins = property(lambda s: Plugin.query.all())
session.add(
PluginResult(host_id=1, plugin_id="Plugin", health_status="ok"))
session.add(
PluginResult(host_id=2, plugin_id="Plugin", health_status="minor"))
session.add(
PluginResult(host_id=3, plugin_id="Plugin", health_status="major"))
session.add(
PluginResult(host_id=4,
plugin_id="Plugin",
health_status="critical"))
session.add(
PluginResult(host_id=5,
plugin_id="Plugin",
health_status="unknown"))
# Host ID 6 has no data
session.add(HostGroup(id=1))
session.add(HostGroupAssignment(host_group_id=1, member_host_id=1))
session.add(HostGroupAssignment(host_group_id=1, member_host_id=4))
for number in range(1, 11):
session.add(Service(id=number, name=number))
# test_failed_dependency
session.add(ServiceDependency(service_id=1, host_id=1))
session.add(ServiceDependency(service_id=1, host_id=1))
session.add(ServiceDependency(service_id=1, host_id=3))
# test_ok_dependency
session.add(ServiceDependency(service_id=2, host_id=1))
session.add(ServiceDependency(service_id=2, host_id=1))
session.add(ServiceDependency(service_id=2, host_id=1))
# test_failure_in_redundancy_group
session.add(RedundancyGroup(id=1, service_id=3))
session.add(RedundancyGroupComponent(redundancy_group_id=1, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=1, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=1, host_id=4))
session.add(ServiceDependency(service_id=3, redundancy_group_id=1))
# test_failed_redundancy_group
session.add(RedundancyGroup(id=2, service_id=4))
session.add(RedundancyGroupComponent(redundancy_group_id=2, host_id=2))
session.add(RedundancyGroupComponent(redundancy_group_id=2, host_id=2))
session.add(RedundancyGroupComponent(redundancy_group_id=2, host_id=4))
session.add(ServiceDependency(service_id=4, redundancy_group_id=2))
# redundancy_group_ok
session.add(RedundancyGroup(id=3, service_id=5))
session.add(RedundancyGroupComponent(redundancy_group_id=3, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=3, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=3, host_id=1))
session.add(ServiceDependency(service_id=5, redundancy_group_id=3))
# test_failed_dependency_with_redundancy_group
session.add(RedundancyGroup(id=4, service_id=6))
session.add(RedundancyGroupComponent(redundancy_group_id=4, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=4, host_id=1))
session.add(RedundancyGroupComponent(redundancy_group_id=4, host_id=1))
session.add(ServiceDependency(service_id=6, redundancy_group_id=4))
session.add(ServiceDependency(service_id=6, host_id=3))
# test_all_dependency_failed
session.add(ServiceDependency(service_id=7, host_id=3))
session.add(ServiceDependency(service_id=7, host_id=3))
session.add(ServiceDependency(service_id=7, host_id=4))
# test_no_data
session.add(ServiceDependency(service_id=8, host_id=1))
session.add(ServiceDependency(service_id=8, host_id=1))
session.add(ServiceDependency(service_id=8, host_id=6))
# test_host_group_dependency
session.add(ServiceDependency(service_id=9, host_group_id=1))
# test_host_group_redundancy
session.add(RedundancyGroup(id=5, service_id=10))
session.add(
RedundancyGroupComponent(redundancy_group_id=5, host_group_id=1))
session.add(ServiceDependency(service_id=10, redundancy_group_id=5))
session.commit()
