Add a SetDoNotFragment() method to DatagramSocket, and call this for

net/udp/udp_socket_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/udp/udp_socket.h"
 
 #include "net/udp/udp_client_socket.h"
 #include "net/udp/udp_server_socket.h"
 
 #include "base/bind.h"
@@ skipping 559 matching lines @@
   CreateUDPAddress("127.0.0.1", 0, &bind_address);
   UDPServerSocket server(NULL, NetLog::Source());
   int rv = server.Listen(bind_address);
   EXPECT_THAT(rv, IsOk());
 
   IPEndPoint peer_address;
   rv = server.GetPeerAddress(&peer_address);
   EXPECT_EQ(rv, ERR_SOCKET_NOT_CONNECTED);
 }
 
+TEST_F(UDPSocketTest, ClientSetDoNotFragment) {
+  for (std::string ip : {"127.0.0.1", "::1"}) {
+    LOG(INFO) << "ip: " << ip;
+    UDPClientSocket client(DatagramSocket::DEFAULT_BIND, RandIntCallback(),
+                           nullptr, NetLog::Source());
+    IPAddress ip_address;
+    EXPECT_TRUE(ip_address.AssignFromIPLiteral(ip));
+    IPEndPoint remote_address(ip_address, 80);
+    int rv = client.Connect(remote_address);
+    // May fail on IPv6 is IPv6 is not configured.
+    if (ip_address.IsIPv6() && rv == ERR_ADDRESS_UNREACHABLE)
+      return;
+    EXPECT_THAT(rv, IsOk());
+
+#if defined(OS_MACOSX)
+    EXPECT_EQ(ERR_NOT_IMPLEMENTED, client.SetDoNotFragment());
+#else
+    rv = client.SetDoNotFragment();
+    EXPECT_THAT(rv, IsOk());
+#endif
+  }
+}
+
+TEST_F(UDPSocketTest, ServerSetDoNotFragment) {
+  for (std::string ip : {"127.0.0.1", "::1"}) {
+    LOG(INFO) << "ip: " << ip;
+    IPEndPoint bind_address;
+    CreateUDPAddress(ip, 0, &bind_address);
+    UDPServerSocket server(nullptr, NetLog::Source());
+    int rv = server.Listen(bind_address);
+    // May fail on IPv6 is IPv6 is not configure
+    if (bind_address.address().IsIPv6() && rv == ERR_ADDRESS_INVALID)
+      return;
+    EXPECT_THAT(rv, IsOk());
+
+#if defined(OS_MACOSX)
+    EXPECT_EQ(ERR_NOT_IMPLEMENTED, server.SetDoNotFragment());
+#else
+    rv = server.SetDoNotFragment();
+    EXPECT_THAT(rv, IsOk());
+#endif
+  }
+}
+
 // Close the socket while read is pending.
 TEST_F(UDPSocketTest, CloseWithPendingRead) {
   IPEndPoint bind_address;
   CreateUDPAddress("127.0.0.1", 0, &bind_address);
   UDPServerSocket server(NULL, NetLog::Source());
   int rv = server.Listen(bind_address);
   EXPECT_THAT(rv, IsOk());
 
   TestCompletionCallback callback;
   IPEndPoint from;
@@ skipping 265 matching lines @@
   g_expected_traffic_type = QOSTrafficTypeExcellentEffort;
   EXPECT_THAT(client.SetDiffServCodePoint(DSCP_NO_CHANGE), IsOk());
   g_expected_dscp = DSCP_DEFAULT;
   g_expected_traffic_type = QOSTrafficTypeBestEffort;
   EXPECT_THAT(client.SetDiffServCodePoint(DSCP_DEFAULT), IsOk());
   client.Close();
 }
 #endif
 
 }  // namespace net