def test_end_to_end_sample_1(_mock):
config = cli.load_config("../config/config_sample_1.py")
collector = SockPuppetCollector(config)
items = list(collector.collect())
for i in items:
for sample in i.samples:
assert sample.labels["flow"] == "ceph-mds"
assert len(items) == len(collector.metric_definitions)
def test_end_to_end_negative(_mock):
config_basic = """
flow_definitions = [
{
"class": "https",
"flows": [
{
"flow": "https-inbound",
"src_port": 1
},
]
},
]
"""
config = mock_module(config_basic)
collector = SockPuppetCollector(config)
items = list(collector.collect())
assert len(items) == 0
def test_end_to_end_dst_set(_mock):
config_basic = """
flow_definitions = [
{
"class": "example",
"flows": [
{
"flow": "in",
"src_port": 1,
},
]
},
]
"""
config = mock_module(config_basic)
collector = SockPuppetCollector(config)
items = list(collector.collect())
assert len(items) == len(collector.metric_definitions)
def test_collect():
config_basic = """
flow_definitions = [
{
"class": "https",
"flows": [
{
"flow": "https-inbound",
"src_port": 443
},
{
"flow": "https-outbound",
"dst_port": 443
},
],
},
]
"""
config = mock_module(config_basic)
collector = SockPuppetCollector(config)
collector.collect()
def test_process_cmd_negative(mock_):
config_basic = """
flow_definitions = [
{
"class": "example",
"flows": [
{
"flow": "in",
"src_port": 4444
},
]
},
]
"""
config = mock_module(config_basic)
collector = SockPuppetCollector(config)
items = list(collector.collect())
for i in items:
for sample in i.samples:
assert sample.labels["process"] == "None"
assert len(items) == len(collector.metric_definitions)
def main(args):
"""Main entry point allowing external calls
Args:
args ([str]): command line parameter list
"""
args = parse_args(args)
setup_logging(args.loglevel)
_logger.info("Listening on: {}:{}".format(args.address, args.port))
start_http_server(args.port, addr=args.address)
config = load_config(args.config_path)
collector = SockPuppetCollector(config=config)
REGISTRY.register(collector)
while True:
time.sleep(10)