add test coverage

IN PROGRESS

Increase agent test coverage.