Fix broken retry tests

* Failure was in output transformer specified in retry_task.yaml.
* Added assertion back to enable the tests.
* Extra logging to include more context in the logs to gather more insight from the logs.

Change-Id: Ic6f4665cd4d74d2676fc17891dbf66984926cd42