test_models.py•1.1 kB
from cycode.cyclient.models import InternalMetadata, K8SResource, ResourcesCollection
from tests import PODS_MOCK
def test_batch_resources_to_json() -> None:
batch = ResourcesCollection('pod', 'default', PODS_MOCK, 77777)
json_dict = batch.to_json()
assert 'resources' in json_dict
assert 'namespace' in json_dict
assert 'total_count' in json_dict
assert 'type' in json_dict
assert json_dict['total_count'] == 77777
assert json_dict['type'] == 'pod'
assert json_dict['namespace'] == 'default'
assert json_dict['resources'][0]['name'] == 'pod_name_1'
def test_internal_metadata_to_json() -> None:
resource = K8SResource('nginx-template-123-456', 'pod', 'cycode', {})
resource.internal_metadata = InternalMetadata('nginx-template', 'deployment')
batch = ResourcesCollection('pod', 'cycode', [resource], 1)
json_dict = batch.to_json()
internal_metadata = json_dict['resources'][0]['internal_metadata']
assert internal_metadata['root_entity_name'] == 'nginx-template'
assert internal_metadata['root_entity_type'] == 'deployment'