netbox-agent/tests/server.py

from netbox_agent.dmidecode import parse
from netbox_agent.server import ServerBase
from netbox_agent.vendors.hp import HPHost
from netbox_agent.vendors.qct import QCTHost
from netbox_agent.vendors.supermicro import SupermicroHost
from tests.conftest import parametrize_with_fixtures


@parametrize_with_fixtures('dmidecode/')
def test_init(fixture):
    dmi = parse(fixture)
    server = ServerBase(dmi)
    assert server


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'HP_SL4540_Gen8',
        'HP_BL460c_Gen9',
        'HP_DL380p_Gen8',
        'HP_SL4540_Gen8'
        'HP_ProLiant_BL460c_Gen10_Graphics_Exp'
    ])
def test_hp_service_tag(fixture):
    dmi = parse(fixture)
    server = HPHost(dmi)
    assert server.get_service_tag() == '4242'


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'HP_ProLiant_m710x'
    ])
def test_moonshot_blade(fixture):
    dmi = parse(fixture)
    server = HPHost(dmi)
    assert server.get_service_tag() == 'CN66480BLA'
    assert server.get_chassis_service_tag() == 'CZ3702MD5K'
    assert server.is_blade() is True
    assert server.own_expansion_slot() is False


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'SYS-5039MS-H12TRF-OS012.txt'
    ])
def test_supermicro_blade(fixture):
    dmi = parse(fixture)
    server = SupermicroHost(dmi)
    assert server.get_service_tag() == 'E235735X6B01665'
    assert server.get_chassis_service_tag() == 'C9390AF40A20098'
    assert server.get_chassis() == 'SYS-5039MS-H12TRF-OS012'
    assert server.is_blade() is True


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'SM_SYS-6018R'
    ])
def test_supermicro_pizza(fixture):
    dmi = parse(fixture)
    server = SupermicroHost(dmi)
    assert server.get_service_tag() == 'A177950X7709591'
    assert server.get_chassis() == 'SYS-6018R-TDTPR'
    assert server.is_blade() is False


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'QCT_X10E-9N'
    ])
def test_qct_x10(fixture):
    dmi = parse(fixture)
    server = QCTHost(dmi)
    assert server.get_service_tag() == 'QTFCQ57140285'


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'unknown.txt'
    ])
def test_generic_host_service_tag(fixture):
    dmi = parse(fixture)
    server = ServerBase(dmi)
    assert server.get_service_tag() == '42'


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'unknown.txt'
    ])
def test_generic_host_product_name(fixture):
    dmi = parse(fixture)
    server = ServerBase(dmi)
    assert server.get_product_name() == 'SR'


@parametrize_with_fixtures(
    'dmidecode/', only_filenames=[
        'HP_ProLiant_BL460c_Gen10_Graphics_Exp'
    ])
def test_hp_blade_with_gpu_expansion(fixture):
    dmi = parse(fixture)
    server = HPHost(dmi)
    assert server.get_service_tag() == '4242'
    assert server.get_chassis_service_tag() == '4343'
    assert server.is_blade() is True
    assert server.own_expansion_slot() is True
    assert server.get_expansion_service_tag() == '4242 expansion'