From 3ad7757cf286950855f484d2779f3b8a6c7a3d7e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AD=90=E9=93=AD?= <zi__mm@163.com>
Date: Thu, 13 Mar 2025 17:54:04 +0800
Subject: [PATCH] =?UTF-8?q?add=20=E6=9F=A5=E8=AF=A2=E5=9B=9E=E8=B0=83?=
 =?UTF-8?q?=E5=9C=B0=E5=9D=80=E6=8E=A5=E5=8F=A3,=E8=AE=BE=E7=BD=AE?=
 =?UTF-8?q?=E5=9B=9E=E8=B0=83=E9=80=9A=E7=9F=A5i=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/data/mixrepoimpl/cmb.go |  5 +++++
 internal/pkg/cmb/sm2_test.go     | 10 +++++++---
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/internal/data/mixrepoimpl/cmb.go b/internal/data/mixrepoimpl/cmb.go
index f4143fd..831039f 100644
--- a/internal/data/mixrepoimpl/cmb.go
+++ b/internal/data/mixrepoimpl/cmb.go
@@ -247,5 +247,10 @@ func (s *CmbMixRepoImpl) Request(ctx context.Context, req *v1.CmbRequest, uri st
 }
 
 func (s *CmbMixRepoImpl) Decrypt(_ context.Context, encryptBody string) (string, error) {
+
+	if len(s.bc.Cmb.CmbSm2Pik) == 0 {
+		return "", errors.New("mock CmbSm2Pik is empty")
+	}
+
 	return cmb.Decrypt(s.bc.Cmb.CmbSm2Pik, encryptBody)
 }
diff --git a/internal/pkg/cmb/sm2_test.go b/internal/pkg/cmb/sm2_test.go
index 9b22ff7..7748b73 100644
--- a/internal/pkg/cmb/sm2_test.go
+++ b/internal/pkg/cmb/sm2_test.go
@@ -24,10 +24,14 @@ func TestEncrypt(t *testing.T) {
 }
 
 func TestDecrypt(t *testing.T) {
-	priKey := "9450c673cf801164435b9c164ac1404e87997245bc6f323fde22015875a03f6e"
-	content := "BHeko/ZYFzQOJn6Q3y46X1AjNz8Nh5fq1FfMuWebh+TangLnlK5iFqePCst4rjG/FKJInijiKO2Qq18sJULlMEEgri05s+bHHDKM+Y+73crAbCnIhHbZxUjt8A0cq2rKjzkl8bxW33dU18uuiTEAmsAvKvmZgE6zJ1eDyjFWefHEIFJaKCNY2cTQOInt|8UYFZFTkx0DovPhaWCbdBkAqbgGmegT14F5gwXLJ6G1uWdYNvX+i5QWAYUGtd8u9"
+	priKey := "8d39ff3d2559258c163f4510f082727f51531e1953ab203d5ab1ea4a6d94fd73"
+	content := "BHhdJbBP8hybQ57niOICuPiaNTGYKgLF3Ntf1ShdpHDjS6s3UJOCpSxEj0nWCVGlJMcTBJIBZ0pvPD4/oWKxgmp69MhpR3VWJeDcHhJHALdvzl2CR2GTXB7lgy1eUEBH/HlYxCK2P3kcV2ENKkazQRpT2Gk5WJ0rCV8iSaEqovmjRsTV3V4eLLrcuzK+|NZQ2Ae4McLlpSjBhYWQFmc9g0h/DQHcykGwIproe0AA0HayMun04aalcZmzh4Qp5p1SBZNBfwqtETyEjHYlWTj7/ht+kWsWRkuVx1uipZVIYPkOw5thsjoCdK+q1S76EssDCXOTVf/XAoSgqV+6/a1dGwkQgzXVGkGdAUoOdN0Rry8g8JnTBQZyiQY/fjDTiieJiS9MUmm6JiE/wZGG+LliUKHVsF8ZdrIEyrkSgAWw="
 	xx, err := Decrypt(priKey, content)
-	t.Log(xx, err)
+	if err != nil {
+		t.Log(err)
+		return
+	}
+	t.Logf("===%v==%s", xx, xx)
 }
 
 func TestEncryptDecrypt(t *testing.T) {