5 files changed, 7 insertions, 5 deletions
diff --git a/net/udp/datagram_client_socket.h b/net/udp/datagram_client_socket.h
index da0f48c..9eb1d55 100644
--- a/net/udp/datagram_client_socket.h
+++ b/net/udp/datagram_client_socket.h
@@ -13,7 +13,7 @@ namespace net {
 
 class IPEndPoint;
 
-class DatagramClientSocket : public DatagramSocket, public Socket {
+class NET_TEST DatagramClientSocket : public DatagramSocket, public Socket {
  public:
   virtual ~DatagramClientSocket() {}
 
diff --git a/net/udp/datagram_server_socket.h b/net/udp/datagram_server_socket.h
index 9841d53..ed34580 100644
--- a/net/udp/datagram_server_socket.h
+++ b/net/udp/datagram_server_socket.h
@@ -15,7 +15,7 @@ class IPEndPoint;
 class IOBuffer;
 
 // A UDP Socket.
-class DatagramServerSocket : public DatagramSocket {
+class NET_API DatagramServerSocket : public DatagramSocket {
  public:
   virtual ~DatagramServerSocket() {}
 
diff --git a/net/udp/datagram_socket.h b/net/udp/datagram_socket.h
index 88a6bc1..b60e079 100644
--- a/net/udp/datagram_socket.h
+++ b/net/udp/datagram_socket.h
@@ -6,13 +6,15 @@
 #define NET_UDP_DATAGRAM_SOCKET_H_
 #pragma once
 
+#include "net/base/net_api.h"
+
 namespace net {
 
 class IPEndPoint;
 
 // A datagram socket is an interface to a protocol which exchanges
 // datagrams, like UDP.
-class DatagramSocket {
+class NET_TEST DatagramSocket {
  public:
   virtual ~DatagramSocket() {}
 
diff --git a/net/udp/udp_client_socket.h b/net/udp/udp_client_socket.h
index e21478f..3dc8778 100644
--- a/net/udp/udp_client_socket.h
+++ b/net/udp/udp_client_socket.h
@@ -15,7 +15,7 @@ namespace net {
 class BoundNetLog;
 
 // A client socket that uses UDP as the transport layer.
-class UDPClientSocket : public DatagramClientSocket {
+class NET_TEST UDPClientSocket : public DatagramClientSocket {
  public:
   UDPClientSocket(net::NetLog* net_log,
                   const net::NetLog::Source& source);
diff --git a/net/udp/udp_server_socket.h b/net/udp/udp_server_socket.h
index 4eae911..300899e 100644
--- a/net/udp/udp_server_socket.h
+++ b/net/udp/udp_server_socket.h
@@ -16,7 +16,7 @@ class IPEndPoint;
 class BoundNetLog;
 
 // A client socket that uses UDP as the transport layer.
-class UDPServerSocket : public DatagramServerSocket {
+class NET_API UDPServerSocket : public DatagramServerSocket {
  public:
   UDPServerSocket(net::NetLog* net_log,
                   const net::NetLog::Source& source);