Function test_analytics

tests/func/test_daemon.py:89–130

(tmp_path, server)

def test_analytics(tmp_path, server):
    addr = server.server_address
    logfile = tmp_path / "logfile"

    env = {
        **os.environ,
        DVC_DAEMON_LOGFILE: str(logfile),
        DVC_ANALYTICS_ENDPOINT: "http://{}:{}".format(*addr),
    }
    env.pop("DVC_TEST", None)
    env.pop("DVC_NO_ANALYTICS", None)
    # The `iterative-telemetry` package calls `gh api` to generate a CI id.
    # This might hang especially on Windows,
    # possibly due to system load from the running tests.
    # Removing the GITHUB_ACTIONS env var avoids calling `gh api`.
    env.pop("GITHUB_ACTIONS", None)

    output = subprocess.check_output(
        [*_get_dvc_args(), "config", "-l", "-vv"],
        env=env,
        text=True,
    )

    match = re.search(r".*Saving analytics report to (.*)", output, flags=re.MULTILINE)
    assert match, "no match for the report file"
    report_file = match.group(1).strip()

    match = re.search(
        r".*Spawned .*analytics.* with pid (.*)", output, flags=re.MULTILINE
    )
    assert match, "no match for the pid"
    pid = int(match.group(1).strip())

    with suppress(psutil.NoSuchProcess):
        psutil.Process(pid).wait(timeout=10)

    log_contents = logfile.read_text(encoding="utf8")
    expected_line = (f"Process {pid} " if os.name != "nt" else "") + "exiting with 0"
    assert expected_line in log_contents

    assert not os.path.exists(report_file)
    assert server.RequestHandlerClass.hits == {"POST": 1}


def test_updater(tmp_dir, dvc, server):