From cdd2a8f5b83d5bea4942a61ae5868295d8eadc80 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=B4=E6=B2=BB=E5=9B=BD?= <changye@startdt.com>
Date: Thu, 23 Sep 2021 10:29:15 +0800
Subject: [PATCH] [ISSUE #5095] Add testcase for common base64 (#6952)

* add testcase for common base64

* testcase

* license
---
 .../nacos/common/codec/Base64Test.java        | 82 +++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 common/src/test/java/com/alibaba/nacos/common/codec/Base64Test.java

diff --git a/common/src/test/java/com/alibaba/nacos/common/codec/Base64Test.java b/common/src/test/java/com/alibaba/nacos/common/codec/Base64Test.java
new file mode 100644
index 000000000..e32c1dfcd
--- /dev/null
+++ b/common/src/test/java/com/alibaba/nacos/common/codec/Base64Test.java
@@ -0,0 +1,82 @@
+/*
+ * Copyright 1999-2018 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.alibaba.nacos.common.codec;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.nio.charset.StandardCharsets;
+
+public class Base64Test {
+    
+    @Test
+    public void test() {
+        String origin = "nacos";
+        String encoded = "bmFjb3M=";
+        
+        byte[] encodeBase64 = Base64.encodeBase64(origin.getBytes(StandardCharsets.UTF_8));
+        Assert.assertEquals(encoded, new String(encodeBase64));
+        byte[] decodeBase64 = Base64.decodeBase64(encoded.getBytes(StandardCharsets.UTF_8));
+        Assert.assertEquals(origin, new String(decodeBase64));
+    }
+    
+    @Test
+    public void testEncodeNullOrEmpty() {
+        byte[] b1 = Base64.encodeBase64(null);
+        Assert.assertEquals(null, b1);
+        byte[] b2 = Base64.encodeBase64(new byte[] {});
+        Assert.assertEquals(0, b2.length);
+    }
+    
+    @Test
+    public void testChunk() {
+        String a = "very large characters to test chunk encoding and see if the result is expected or not";
+        byte[] b1 = Base64.encodeBase64(a.getBytes(StandardCharsets.UTF_8), false, false, Integer.MAX_VALUE);
+        byte[] b2 = Base64.encodeBase64(a.getBytes(StandardCharsets.UTF_8), true, false, Integer.MAX_VALUE);
+        String s1 = new String(b1);
+        String s2 = new String(b2);
+        Assert.assertEquals(s1, "dmVyeSBsYXJnZSBjaGFyYWN0ZXJzIHRvIHRlc3QgY2h1bmsgZW5jb2RpbmcgYW5kIHNlZSBpZiB0"
+                + "aGUgcmVzdWx0IGlzIGV4cGVjdGVkIG9yIG5vdA==");
+        Assert.assertEquals(s2, "dmVyeSBsYXJnZSBjaGFyYWN0ZXJzIHRvIHRlc3QgY2h1bmsgZW5jb2RpbmcgYW5kIHNlZSBpZiB0" + "\r\n"
+                + "aGUgcmVzdWx0IGlzIGV4cGVjdGVkIG9yIG5vdA==" + "\r\n");
+    
+        byte[] c1 = Base64.decodeBase64(b1);
+        byte[] c2 = Base64.decodeBase64(b2);
+        String s3 = new String(c1);
+        String s4 = new String(c2);
+        Assert.assertEquals(a, s3);
+        Assert.assertEquals(a, s4);
+    }
+    
+    @Test
+    public void testUrlSafe() {
+        String a = "aa~aa?";
+        byte[] b1 = Base64.encodeBase64(a.getBytes(StandardCharsets.UTF_8), false, false, Integer.MAX_VALUE);
+        byte[] b2 = Base64.encodeBase64(a.getBytes(StandardCharsets.UTF_8), false, true, Integer.MAX_VALUE);
+        String s1 = new String(b1);
+        String s2 = new String(b2);
+        Assert.assertEquals("YWF+YWE/", s1);
+        Assert.assertEquals("YWF-YWE_", s2);
+        
+        byte[] c1 = Base64.decodeBase64(b1);
+        byte[] c2 = Base64.decodeBase64(b2);
+        String s3 = new String(c1);
+        String s4 = new String(c2);
+        Assert.assertEquals("aa~aa?", s3);
+        Assert.assertEquals("aa~aa?", s4);
+    }
+}