Update/tests

@ianstormtaylor
i think we'll have a `test.restore()` that we can call