Fix tests: use Tests context instead of mocking