As a developer, I want to write comprehensive tests for the agent service, so that I can ensure its logic is correct and compatible with legacy services.

JIRA Issue: BA-1898