Tidied up window mocking