Add tests for dispose action