Add tests for post-save behaviour