mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-05-18 07:07:18 +00:00
14d73e201e
* Remove plugins Current usage of plugins is not pluggable and includes logging stuff. Move this to conf/logging. * Removed dynamic imports * Add tests for hunters registration
24 lines
712 B
Python
24 lines
712 B
Python
import requests_mock
|
|
import json
|
|
|
|
from kube_hunter.core.events.types import NewHostEvent
|
|
|
|
|
|
# Testing if it doesn't try to run get_cloud if the cloud type is already set.
|
|
# get_cloud(1.2.3.4) will result with an error
|
|
def test_presetcloud():
|
|
expcted = "AWS"
|
|
hostEvent = NewHostEvent(host="1.2.3.4", cloud=expcted)
|
|
assert expcted == hostEvent.cloud
|
|
|
|
|
|
def test_getcloud():
|
|
fake_host = "1.2.3.4"
|
|
expected_cloud = "Azure"
|
|
result = {"cloud": expected_cloud}
|
|
|
|
with requests_mock.mock() as m:
|
|
m.get(f"https://api.azurespeed.com/api/region?ipOrUrl={fake_host}", text=json.dumps(result))
|
|
hostEvent = NewHostEvent(host=fake_host)
|
|
assert hostEvent.cloud == expected_cloud
|