Skip to content

Commit

Permalink
changing useragent
Browse files Browse the repository at this point in the history
Signed-off-by: h4l0gen <[email protected]>
  • Loading branch information
h4l0gen committed May 20, 2024
1 parent 1b0c9f7 commit d2f9f6d
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions tests/test_updater_ng.py
Original file line number Diff line number Diff line change
Expand Up @@ -325,7 +325,7 @@ def test_user_agent(self) -> None:
self.updater.refresh()
session = next(iter(self.updater._fetcher._sessions.values()))
ua = session.headers["User-Agent"]
self.assertEqual(ua[:4], "tuf/")
self.assertEqual(ua[:11], "python-tuf/")

# test custom UA
updater = Updater(
Expand All @@ -339,7 +339,7 @@ def test_user_agent(self) -> None:
session = next(iter(updater._fetcher._sessions.values()))
ua = session.headers["User-Agent"]

self.assertEqual(ua[:16], "MyApp/1.2.3 tuf/")
self.assertEqual(ua[:23], "MyApp/1.2.3 python-tuf/")


if __name__ == "__main__":
Expand Down
2 changes: 1 addition & 1 deletion tuf/ngclient/_internal/requests_fetcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ def _get_session(self, url: str) -> requests.Session:
session = requests.Session()
self._sessions[session_index] = session

ua = f"tuf/{tuf.__version__} {session.headers['User-Agent']}"
ua = f"python-tuf/{tuf.__version__} {session.headers['User-Agent']}"
if self.app_user_agent is not None:
ua = f"{self.app_user_agent} {ua}"
session.headers["User-Agent"] = ua
Expand Down

0 comments on commit d2f9f6d

Please sign in to comment.