From d5ac9ad56e890de6c6e16a8ec154db16bae1cbd6 Mon Sep 17 00:00:00 2001
From: MengYX <admin@mengyx.cn>
Date: Tue, 14 Dec 2021 18:01:59 +0800
Subject: [PATCH] test(QMCv2): coverage standard TEA cipher

(cherry picked from commit c2c89a423ffffc06fb43c86d4714bb32d1936c3e)
---
 src/utils/tea.test.ts | 8 ++++++++
 src/utils/tea.ts      | 3 ---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/utils/tea.test.ts b/src/utils/tea.test.ts
index fc5741c..a3f9273 100644
--- a/src/utils/tea.test.ts
+++ b/src/utils/tea.test.ts
@@ -6,6 +6,7 @@
 
 import {TeaCipher} from "@/utils/tea";
 
+
 test("key size", () => {
     const testKey = new Uint8Array([
         0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
@@ -52,6 +53,12 @@ const teaTests = [
     },
 ]
 
+test("rounds", () => {
+    const tt = teaTests[0];
+    expect(() => new TeaCipher(tt.key, tt.rounds - 1))
+        .toThrow()
+})
+
 
 test("encrypt & decrypt", () => {
     for (const tt of teaTests) {
@@ -67,3 +74,4 @@ test("encrypt & decrypt", () => {
         expect(buf).toStrictEqual(tt.plainText)
     }
 })
+
diff --git a/src/utils/tea.ts b/src/utils/tea.ts
index 22366c3..9b4f57a 100644
--- a/src/utils/tea.ts
+++ b/src/utils/tea.ts
@@ -51,9 +51,6 @@ export class TeaCipher {
 
 
     encrypt(dst: DataView, src: DataView) {
-        if (src.byteLength != 8) {
-            throw Error("src.byteLength != 8")
-        }
 
         let v0 = src.getUint32(0, false)
         let v1 = src.getUint32(4, false)