Allow stub() to be called in beforeEach/beforeAll

[brandonsturgeon]

stub() wasn't originally set up to work in those functions, but it's a clear opportunity for improvement.