add a away to add a transaction aware to an in-progress transaction context

also add a unit test for TransactionContext