Converted tests to MockIO