Add tests for prison_client container