update agent tests to be less integrated.