File: test_user_agent.py

package info (click to toggle)
cwltool 3.1.20250110105449-3
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 10,516 kB
  • sloc: python: 20,992; javascript: 20,796; sh: 233; makefile: 152
file content (24 lines) | stat: -rw-r--r-- 746 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import requests

from cwltool.main import append_word_to_default_user_agent, main


def get_user_agent() -> str:
    return requests.utils.default_headers()["User-Agent"]


def test_cwltool_in_user_agent() -> None:
    """python-requests HTTP User-Agent should include the string 'cwltool'."""
    try:
        assert main(["--version"]) == 0
    except SystemExit as err:
        assert err.code == 0
    assert "cwltool" in get_user_agent()


def test_append_word_to_default_user_agent() -> None:
    """Confirm that append_word_to_default_user_agent works."""
    word_to_append = "foobar123"
    assert word_to_append not in get_user_agent()
    append_word_to_default_user_agent(word_to_append)
    assert word_to_append in get_user_agent()