mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-05-19 23:59:28 +00:00
0f1739262f
Fix linting issues with flake8 and black. Add pre-commit congifuration, update documnetation for it. Apply linting check in Travis CI.
34 lines
833 B
Python
34 lines
833 B
Python
from kube_hunter.modules.report import get_reporter, get_dispatcher
|
|
from kube_hunter.modules.report.factory import (
|
|
YAMLReporter,
|
|
JSONReporter,
|
|
PlainReporter,
|
|
HTTPDispatcher,
|
|
STDOUTDispatcher,
|
|
)
|
|
|
|
|
|
def test_reporters():
|
|
test_cases = [
|
|
("plain", PlainReporter),
|
|
("json", JSONReporter),
|
|
("yaml", YAMLReporter),
|
|
("notexists", PlainReporter),
|
|
]
|
|
|
|
for report_type, expected in test_cases:
|
|
actual = get_reporter(report_type)
|
|
assert type(actual) is expected
|
|
|
|
|
|
def test_dispatchers():
|
|
test_cases = [
|
|
("stdout", STDOUTDispatcher),
|
|
("http", HTTPDispatcher),
|
|
("notexists", STDOUTDispatcher),
|
|
]
|
|
|
|
for dispatcher_type, expected in test_cases:
|
|
actual = get_dispatcher(dispatcher_type)
|
|
assert type(actual) is expected
|