package crypto

import (
	"testing"
)

func TestWestDexEncryptDecrypt(t *testing.T) {
	testCases := []struct {
		name      string
		data      string
		secretKey string
	}{
		{
			name:      "简单文本",
			data:      "hello world",
			secretKey: "mySecretKey123",
		},
		{
			name:      "中文文本",
			data:      "你好世界",
			secretKey: "中文密钥",
		},
		{
			name:      "JSON数据",
			data:      `{"name":"张三","age":30,"city":"北京"}`,
			secretKey: "jsonSecretKey",
		},
		{
			name:      "长文本",
			data:      "这是一个很长的文本，用来测试加密解密功能是否正常工作。包含各种字符：123456789!@#$%^&*()_+-=[]{}|;':\",./<>?",
			secretKey: "longTextKey",
		},
		{
			name:      "空字符串",
			data:      "",
			secretKey: "emptyDataKey",
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			// 加密
			encrypted, err := WestDexEncrypt(tc.data, tc.secretKey)
			if err != nil {
				t.Fatalf("加密失败: %v", err)
			}

			t.Logf("原始数据: %s", tc.data)
			t.Logf("密钥: %s", tc.secretKey)
			t.Logf("加密结果: %s", encrypted)

			// 解密
			decrypted, err := WestDexDecrypt(encrypted, tc.secretKey)
			if err != nil {
				t.Fatalf("解密失败: %v", err)
			}

			decryptedStr := string(decrypted)
			t.Logf("解密结果: %s", decryptedStr)

			// 验证解密结果是否与原始数据一致
			if decryptedStr != tc.data {
				t.Errorf("解密结果不匹配: 期望 %s, 实际 %s", tc.data, decryptedStr)
			}
		})
	}
}

func TestWestDexDecryptOutput(t *testing.T) {
	// 专门用来查看解密结果的测试
	testData := []struct {
		name        string
		data        string
		secretKey   string
		encryptedData string // 预设的加密数据
	}{
		{
			name:      "测试数据1",
			data:      "DLrbtEki5o/5yTvQWR+dWWUZYEo5s58D8LTnhhlAl99SwZbECa34KpStmR+Qr0gbbKzh3y4t5+/vbFFZgv03DtnYlLQcQt+rSgtxkCN/PCBPaFE0QZRTufd7djJfUww0Eh6DMHD7NS9pcuCa0PHGVoE+Vwo2YSwOnh2gtx3Bt0Qhs+w76tfCwIeufZ8tcpFs/nb84HIZxk+0cH1bTfNE6VsXI6vMpKvnS02O3oE2642ozeHgglCNuiOFMcCL8Erw4FKPnfRCUYdeKc2dZ7OF2IZqt0t4WiJBxjB/6k4tgAj/HepE2gaulWU8RVvAF+vPF5i3ekHHq8T7226rNlVfuagodaRXiOqO5E1h6Mx9ygcDL0HXvQKsxxJdl/bUP+t/+rOjA+k/IR/vF1UJGrGrkSJVfkcWXPP85cgws18gE9rIs2Ji1HGjvOmnez370L0+",
			secretKey: "121a1e41fc1690dd6b90afbcacd80cf4",
		},
		{
			name:      "中文数据",
			data:      "用户数据",
			secretKey: "密钥123",
		},
		{
			name:      "API数据",
			data:      "api_call_data",
			secretKey: "production_key",
		},
		{
			name:      "JSON格式",
			data:      `{"user_id":12345,"name":"张三","status":"active"}`,
			secretKey: "json_key",
		},
	}

	for i, td := range testData {
		decrypted, err := WestDexDecrypt(td.data, td.secretKey)
		if err != nil {
			t.Fatalf("解密失败: %v", err)
		}

		t.Logf("测试 %d - %s:", i+1, td.name)
		t.Logf("  原始数据: %s", td.data)
		t.Logf("  使用密钥: %s", td.secretKey)
		t.Logf("  解密结果: %s", string(decrypted))
		t.Logf("  解密正确: %v", string(decrypted) == td.data)
		t.Log("---")
	}
}

func TestSpecificDecrypt(t *testing.T) {
	// 如果你有特定的加密数据想要解密，可以在这里测试
	specificTests := []struct {
		name          string
		encryptedData string
		secretKey     string
		expectedData  string // 如果知道预期结果的话
	}{
		// 示例：如果你有具体的加密数据想要解密，可以添加到这里
		// {
		//     name:          "特定数据解密",
		//     encryptedData: "你的加密数据",
		//     secretKey:     "你的密钥",
		//     expectedData:  "预期的解密结果",
		// },
	}

	t.Log("=== 特定数据解密测试 ===")
	for _, test := range specificTests {
		decrypted, err := WestDexDecrypt(test.encryptedData, test.secretKey)
		if err != nil {
			t.Logf("%s - 解密失败: %v", test.name, err)
			continue
		}

		result := string(decrypted)
		t.Logf("%s:", test.name)
		t.Logf("  加密数据: %s", test.encryptedData)
		t.Logf("  使用密钥: %s", test.secretKey)
		t.Logf("  解密结果: %s", result)
		
		if test.expectedData != "" {
			t.Logf("  预期结果: %s", test.expectedData)
			t.Logf("  解密正确: %v", result == test.expectedData)
		}
		t.Log("---")
	}
}

func TestWestDexDecryptWithWrongKey(t *testing.T) {
	// 测试用错误密钥解密
	data := "sensitive data"
	correctKey := "correct_key"
	wrongKey := "wrong_key"

	// 用正确密钥加密
	encrypted, err := WestDexEncrypt(data, correctKey)
	if err != nil {
		t.Fatalf("加密失败: %v", err)
	}

	// 用错误密钥解密
	decrypted, err := WestDexDecrypt(encrypted, wrongKey)
	if err != nil {
		t.Logf("用错误密钥解密失败（这是预期的）: %v", err)
		return
	}

	decryptedStr := string(decrypted)
	t.Logf("原始数据: %s", data)
	t.Logf("用错误密钥解密结果: %s", decryptedStr)
	
	// 验证解密结果应该与原始数据不同
	if decryptedStr == data {
		t.Error("用错误密钥解密不应该得到正确结果")
	}
}

// 基准测试
func BenchmarkWestDexEncrypt(b *testing.B) {
	data := "这是一个用于基准测试的数据字符串"
	secretKey := "benchmarkKey"
	
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		_, err := WestDexEncrypt(data, secretKey)
		if err != nil {
			b.Fatalf("加密失败: %v", err)
		}
	}
}

func BenchmarkWestDexDecrypt(b *testing.B) {
	data := "这是一个用于基准测试的数据字符串"
	secretKey := "benchmarkKey"
	
	// 先加密一次获得密文
	encrypted, err := WestDexEncrypt(data, secretKey)
	if err != nil {
		b.Fatalf("预加密失败: %v", err)
	}
	
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		_, err := WestDexDecrypt(encrypted, secretKey)
		if err != nil {
			b.Fatalf("解密失败: %v", err)
		}
	}
}