build.tests.unit.test_db_models

Complexity

Total Complexity

5

Size/Duplication

Total Lines	103
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

5 Functions

Rating	Name	Duplication	Size	Complexity
A	test_switch_doc_preset_interfaces()	0	26	1
A	test_document_base_model_dict()	0	8	1
A	test_interface_detail_doc()	0	10	1
A	test_switch_doc_preset_interfaces_speed_none()	0	27	1
A	test_switch_doc_no_preset_interfaces()	0	13	1

"""This test suite cover the db models, however, most have been
indirectly tested on test_main.py, so for now it's mostly to cover
the remaning parts."""

from datetime import datetime

from napps.kytos.topology.db.models import (DocumentBaseModel,
                                            InterfaceDetailDoc, SwitchDoc)


def test_document_base_model_dict() -> None:
    """test_document_base_model_dict."""
    _id = "some_id"
    utcnow = datetime.utcnow()
    payload = {"_id": _id, "inserted_at": utcnow, "updated_at": utcnow}
    model = DocumentBaseModel(**payload)
    assert model.model_dump() == {**payload, **{"id": _id}}
    assert "_id" not in model.model_dump(exclude={"_id"})


def test_switch_doc_preset_interfaces() -> None:
    """test_switch_doc_preset_interfaces."""
    dpid = "00:00:00:00:00:00:00:01"
    interface_id = f"{dpid}:1"
    interfaces = {
        interface_id: {
            "id": interface_id,
            "port_number": 1,
            "lldp": True,
            "enabled": True,
            "active": True,
            "mac": "some_mac",
            "speed": 0,
            "name": "some_name",
            "switch": dpid,
        }
    }
    payload = {
        "_id": dpid,
        "enabled": True,
        "active": True,
        "interfaces": interfaces,
    }
    model = SwitchDoc(**payload)
    assert model
    assert interface_id == model.interfaces[0].id


def test_switch_doc_preset_interfaces_speed_none() -> None:
    """test_switch_doc_preset_interfaces speed none."""
    dpid = "00:00:00:00:00:00:00:01"
    interface_id = f"{dpid}:1"
    interfaces = {
        interface_id: {
            "id": interface_id,
            "port_number": 1,
            "lldp": True,
            "enabled": True,
            "active": True,
            "mac": "some_mac",
            "speed": None,
            "name": "some_name",
            "switch": dpid,
        }
    }
    payload = {
        "_id": dpid,
        "enabled": True,
        "active": True,
        "interfaces": interfaces,
    }
    model = SwitchDoc(**payload)
    assert model
    assert interface_id == model.interfaces[0].id
    assert model.interfaces[0].speed is None


def test_switch_doc_no_preset_interfaces() -> None:
    """test_switch_doc_no_preset_interfaces."""
    dpid = "00:00:00:00:00:00:00:01"
    interfaces = []
    payload = {
        "_id": dpid,
        "enabled": True,
        "active": True,
        "interfaces": interfaces,
    }
    model = SwitchDoc(**payload)
    assert model
    assert model.interfaces == interfaces


def test_interface_detail_doc() -> None:
    """Test InterfaceDetailDoc"""
    payload = {
        "available_tags": {"vlan": [[200, 300], [500, 550]]},
        "tag_ranges": {"vlan": [[100, 4095]]},
        "special_available_tags": {"vlan": ["any"]},
        "special_tags": {"vlan": ["any", "untagged"]}
    }
    model = InterfaceDetailDoc(**payload)
    assert model