Add setHeader method in tests