merge TestGenerateReverseTCP_Fake_Compare to TestGenerateReverseTCP_Fake.

Author: For-ACGN

generator_test.go

@@ -229,89 +229,91 @@ func TestGenerateReverseHTTPS(t *testing.T) {
 }
 
 func TestGenerateReverseTCP_Fake(t *testing.T) {
-	const (
-		fileNameFlag = "ReverseTCP.java"
-		hostFlag     = "${host}"
-		portFlag     = "${port}"
-		tokenFlag    = "${token}"
-		className    = "ReverseTCP\x0C"
-	)
-
-	buf := bytes.NewBuffer(make([]byte, 0, 128))
-	buf.Write([]byte{0xCA, 0xFE})
-	buf.Write([]byte{0x00, 0x00})
-
-	size := make([]byte, 2)
-
-	binary.BigEndian.PutUint16(size, uint16(len(fileNameFlag)))
-	buf.Write(size)
-	buf.WriteString(fileNameFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(hostFlag)))
-	buf.Write(size)
-	buf.WriteString(hostFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(portFlag)))
-	buf.Write(size)
-	buf.WriteString(portFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(tokenFlag)))
-	buf.Write(size)
-	buf.WriteString(tokenFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(className)))
-	buf.Write(size)
-	buf.WriteString(className)
-	buf.Write([]byte{0x00, 0x00})
-
-	err := os.WriteFile("testdata/template/ReverseTCP.class", buf.Bytes(), 0600)
-	require.NoError(t, err)
-}
+	t.Run("template", func(t *testing.T) {
+		const (
+			fileNameFlag = "ReverseTCP.java"
+			hostFlag     = "${host}"
+			portFlag     = "${port}"
+			tokenFlag    = "${token}"
+			className    = "ReverseTCP\x0C"
+		)
+
+		buf := bytes.NewBuffer(make([]byte, 0, 128))
+		buf.Write([]byte{0xCA, 0xFE})
+		buf.Write([]byte{0x00, 0x00})
+
+		size := make([]byte, 2)
+
+		binary.BigEndian.PutUint16(size, uint16(len(fileNameFlag)))
+		buf.Write(size)
+		buf.WriteString(fileNameFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(hostFlag)))
+		buf.Write(size)
+		buf.WriteString(hostFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(portFlag)))
+		buf.Write(size)
+		buf.WriteString(portFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(tokenFlag)))
+		buf.Write(size)
+		buf.WriteString(tokenFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(className)))
+		buf.Write(size)
+		buf.WriteString(className)
+		buf.Write([]byte{0x00, 0x00})
+
+		err := os.WriteFile("testdata/template/ReverseTCP.class", buf.Bytes(), 0600)
+		require.NoError(t, err)
+	})
 
-func TestGenerateReverseTCP_Fake_Compare(t *testing.T) {
-	const (
-		fileNameFlag = "ReTCP.java"
-		hostFlag     = "127.0.0.1"
-		portFlag     = "9979"
-		tokenFlag    = "test"
-		className    = "ReTCP\x0C"
-	)
-
-	buf := bytes.NewBuffer(make([]byte, 0, 128))
-	buf.Write([]byte{0xCA, 0xFE})
-	buf.Write([]byte{0x00, 0x00})
-
-	size := make([]byte, 2)
-
-	binary.BigEndian.PutUint16(size, uint16(len(fileNameFlag)))
-	buf.Write(size)
-	buf.WriteString(fileNameFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(hostFlag)))
-	buf.Write(size)
-	buf.WriteString(hostFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(portFlag)))
-	buf.Write(size)
-	buf.WriteString(portFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(tokenFlag)))
-	buf.Write(size)
-	buf.WriteString(tokenFlag)
-	buf.Write([]byte{0x00, 0x00})
-
-	binary.BigEndian.PutUint16(size, uint16(len(className)-1))
-	buf.Write(size)
-	buf.WriteString(className)
-	buf.Write([]byte{0x00, 0x00})
-
-	err := os.WriteFile("testdata/template/compare/ReTCP.class", buf.Bytes(), 0600)
-	require.NoError(t, err)
+	t.Run("compare", func(t *testing.T) {
+		const (
+			fileNameFlag = "ReTCP.java"
+			hostFlag     = "127.0.0.1"
+			portFlag     = "9979"
+			tokenFlag    = "test"
+			className    = "ReTCP\x0C"
+		)
+
+		buf := bytes.NewBuffer(make([]byte, 0, 128))
+		buf.Write([]byte{0xCA, 0xFE})
+		buf.Write([]byte{0x00, 0x00})
+
+		size := make([]byte, 2)
+
+		binary.BigEndian.PutUint16(size, uint16(len(fileNameFlag)))
+		buf.Write(size)
+		buf.WriteString(fileNameFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(hostFlag)))
+		buf.Write(size)
+		buf.WriteString(hostFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(portFlag)))
+		buf.Write(size)
+		buf.WriteString(portFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(tokenFlag)))
+		buf.Write(size)
+		buf.WriteString(tokenFlag)
+		buf.Write([]byte{0x00, 0x00})
+
+		binary.BigEndian.PutUint16(size, uint16(len(className)-1))
+		buf.Write(size)
+		buf.WriteString(className)
+		buf.Write([]byte{0x00, 0x00})
+
+		err := os.WriteFile("testdata/template/compare/ReTCP.class", buf.Bytes(), 0600)
+		require.NoError(t, err)
+	})
 }