fix test cases for CertificateRequest

aliyun · Jun 8, 2023 · 677124a · 677124a
1 parent f678fcf
commit 677124a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/main/java/com/aliyun/gmsse/handshake/CertificateRequest.java b/src/main/java/com/aliyun/gmsse/handshake/CertificateRequest.java
@@ -23,7 +23,7 @@ public CertificateRequest(short[] types, Vector<byte[]> certificateAuthorities)
     public byte[] getBytes() throws IOException {
         ByteArrayOutputStream ba = new ByteArrayOutputStream();
         // write type length
-        int length = this.certificateAuthorities.size();
+        int length = this.certificateTypes.length;
         ba.write(length);
         // write types
         for (int i = 0; i < length; i++) {

diff --git a/src/test/java/com/aliyun/gmsse/handshake/CertificateRequestTest.java b/src/test/java/com/aliyun/gmsse/handshake/CertificateRequestTest.java
@@ -8,9 +8,10 @@
 public class CertificateRequestTest {
 
     @Test
-    public void nullTest() throws Exception{
+    public void getBytesTest() throws Exception{
         short[] types = new short[] {1, 2, 3};
         Vector<byte[]> v = new Vector<>();
-        Assert.assertNull(new CertificateRequest(types, v).getBytes());
+        byte[] expected = new byte[] {0x03, 0x01, 0x02, 0x03, 0x00, 0x00};
+        Assert.assertArrayEquals(expected, new CertificateRequest(types, v).getBytes());
     }
 }