Isolate functions that are only meant to run in production environment. Integrate them back while testing.