Simplified tests and exported mocks in a separate file