Added tests for rollback handler
Added additional tests.