Move state restoration from setup/teardown into affected test cases.