test: update tests to new on_exit stuff