Python get_source_and_piggyback_hosts Examples

Example #1

File: test_piggyback.py Project: surajrb/checkmk

def test_piggyback_default_time_settings():
    time_settings = [(None, "max_cache_age", piggyback_max_cachefile_age)]
    piggybacked_hostname = "test-host"
    piggyback.get_piggyback_raw_data(piggybacked_hostname, time_settings)
    piggyback.get_source_and_piggyback_hosts(time_settings)
    piggyback.has_piggyback_raw_data(piggybacked_hostname, time_settings)
    piggyback.cleanup_piggyback_files(time_settings)

Example #2

def test_get_source_and_piggyback_hosts():
    time_settings: piggyback.PiggybackTimeSettings = [
        (None, "max_cache_age", piggyback_max_cachefile_age)
    ]
    cmk.utils.paths.piggyback_source_dir.mkdir(parents=True, exist_ok=True)

    piggyback.store_piggyback_raw_data(
        HostName("source1"),
        {
            HostName("test-host2"): [
                b"<<<check_mk>>>",
                b"source1",
            ],
            HostName("test-host"): [
                b"<<<check_mk>>>",
                b"source1",
            ],
        },
    )

    # Fake age the test-host piggyback file
    os.utime(
        str(cmk.utils.paths.piggyback_dir / "test-host" / "source1"),
        (time.time() - 10, time.time() - 10),
    )

    piggyback.store_piggyback_raw_data(
        HostName("source1"),
        {HostName("test-host2"): [
             b"<<<check_mk>>>",
             b"source1",
         ]},
    )

    piggyback.store_piggyback_raw_data(
        HostName("source2"),
        {
            HostName("test-host2"): [
                b"<<<check_mk>>>",
                b"source2",
            ],
            HostName("test-host"): [
                b"<<<check_mk>>>",
                b"source2",
            ],
        },
    )

    assert sorted(list(
        piggyback.get_source_and_piggyback_hosts(time_settings))) == sorted([
            (HostName("source1"), HostName("test-host2")),
            (HostName("source2"), HostName("test-host")),
            (HostName("source2"), HostName("test-host2")),
        ])

Example #3

File: test_piggyback.py Project: a-s-995/checkmk-1

def test_get_source_and_piggyback_hosts():
    time_settings = [(None, "max_cache_age", piggyback_max_cachefile_age)
                     ]  # type: piggyback.PiggybackTimeSettings
    cmk.utils.paths.piggyback_source_dir.mkdir(parents=True, exist_ok=True)

    piggyback.store_piggyback_raw_data(
        "source1", {
            "test-host2": [
                b"<<<check_mk>>>",
                b"source1",
            ],
            "test-host": [
                b"<<<check_mk>>>",
                b"source1",
            ]
        })

    # Fake age the test-host piggyback file
    os.utime(str(cmk.utils.paths.piggyback_dir / "test-host" / "source1"),
             (time.time() - 10, time.time() - 10))

    piggyback.store_piggyback_raw_data(
        "source1", {"test-host2": [
            b"<<<check_mk>>>",
            b"source1",
        ]})

    piggyback.store_piggyback_raw_data(
        "source2", {
            "test-host2": [
                b"<<<check_mk>>>",
                b"source2",
            ],
            "test-host": [
                b"<<<check_mk>>>",
                b"source2",
            ]
        })

    assert sorted(list(
        piggyback.get_source_and_piggyback_hosts(time_settings))) == sorted([
            ('source1', 'test-host2'),
            ('source2', 'test-host'),
            ('source2', 'test-host2'),
        ])