mock

code/tp3-ssl/src/test/java/fr/istic/vv/TLSSocketFactoryTestMocks.java

@@ -1,6 +1,9 @@
 package fr.istic.vv;
 
 import org.junit.jupiter.api.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
 
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -8,8 +11,51 @@ import java.util.Collections;
 import java.util.List;
 
 import static org.junit.jupiter.api.Assertions.*;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
 
 public class TLSSocketFactoryTestMocks {
 
+    @Test
+    public void preparedSocket_NullProtocols(){
+        TLSSocketFactory f = new TLSSocketFactory();
+
+        SSLSocket socket = Mockito.mock(SSLSocket.class);
+
+
+        when(socket.getSupportedProtocols()).thenReturn(null);
+        when(socket.getEnabledProtocols()).thenReturn(null);  
+
+        f.prepareSocket(socket);
+
+        verify(socket, never()).setEnabledProtocols(any());
+    }
+
+    @Test
+    public void typical()  {
+        TLSSocketFactory f = new TLSSocketFactory();
+
+        SSLSocket socket = Mockito.mock(SSLSocket.class);
+
+        when(socket.getSupportedProtocols()).thenReturn(shuffle(new String[]{"SSLv2Hello", "SSLv3", "TLSv1", "TLSv1.1", "TLSv1.2"}));
+        when(socket.getEnabledProtocols()).thenReturn(new String[]{"SSLv3", "TLSv1"});
+        
+        f.prepareSocket(socket);
+
+        verify(socket).setEnabledProtocols(
+            eq(new String[] {"TLSv1.2", "TLSv1.1", "TLSv1", "SSLv3"})
+        );
+    }
+
+    private String[] shuffle(String[] in) {
+        List<String> list = new ArrayList<String>(Arrays.asList(in));
+        Collections.shuffle(list);
+        return list.toArray(new String[0]);
+    }
+
 
 }