Fix WebSocket to set first party for cookies

third_party/WebKit/LayoutTests/http/tests/websocket/set-cookie_wsh.py

 # Copyright 2013, Google Inc.  All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions are
 # met:
 #
 #     * Redistributions of source code must retain the above copyright
 # notice, this list of conditions and the following disclaimer.
 #     * Redistributions in binary form must reproduce the above
 # copyright notice, this list of conditions and the following disclaimer
@@ skipping 17 matching lines @@
 
 
 import urlparse
 
 
 def web_socket_do_extra_handshake(request):
     query = urlparse.urlparse(request.uri)[4]
     max_age = ''
     if 'clear' in urlparse.parse_qs(query):
         max_age = '; Max-Age=0'
-    request.extra_headers.append(
-        ('Set-Cookie', 'WK-WebSocket-test-domain-pass=1; Domain=127.0.0.1' + max_age))
-    request.extra_headers.append(
-        ('Set-Cookie', 'WK-WebSocket-test-domain-fail=1; Domain=example.com' + max_age))
-    request.extra_headers.append(
-        ('Set-Cookie', 'WK-WebSocket-test-path-pass=1; Path=/' + max_age))
-    request.extra_headers.append(
-        ('Set-Cookie', 'WK-WebSocket-test-path-fail=1; Path=/foo/bar' + max_age))
-    request.extra_headers.append(('Set-Cookie',
-                                  'WK-WebSocket-test=1' + max_age))
+    cookies = [
+        'WK-WebSocket-test-domain-pass=1; Domain=127.0.0.1' + max_age,
+        'WK-WebSocket-test-domain-fail=1; Domain=example.com' + max_age,
+        'WK-WebSocket-test-path-pass=1; Path=/' + max_age,
+        'WK-WebSocket-test-path-fail=1; Path=/foo/bar' + max_age,
+        'WK-WebSocket-test=1' + max_age,
+        'same-site-strict=1; SameSite=Strict' + max_age,
+        'same-site-lax=1; SameSite=Lax' + max_age
+    ]
+    for value in cookies:
+        request.extra_headers.append(('Set-Cookie', value))
 
 
 def web_socket_transfer_data(request):
     request.ws_stream.send_message('url = ' + request.uri)
 
 
 # vi:sts=4 sw=4 et