Modify user agent test to use mock subscription client