changing useragent

Signed-off-by: h4l0gen <ks3913688@gmail.com>
This commit is contained in:
h4l0gen 2024-05-20 21:53:54 +05:30
parent 1b0c9f7d44
commit d2f9f6dee6
No known key found for this signature in database
2 changed files with 3 additions and 3 deletions

View file

@ -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(
@ -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__":

View file

@ -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