Improve mocking of window and document

In several places the globals 'window' and 'document' are used directly. When we switch to Angular 4 this gives typing issues. We should use $window and $document instead so these globals can be mocked properly.