Modify test code AND Add test cases to handle multiple add_headers