updated benchmark test

rudderlabs · Sep 9, 2024 · e5be841 · e5be841
1 parent 455b1ad
commit e5be841
Showing 1 changed file with 17 additions and 10 deletions.
diff --git a/encrypt/benchmark_test.go b/encrypt/benchmark_test.go
@@ -22,16 +22,23 @@ BenchmarkEncryptDecrypt/AESGCM_AES256-12         	  687800	      1720 ns/op
 */
 func BenchmarkEncryptDecrypt(b *testing.B) {
 	tests := []struct {
-		name  string
-		algo  EncryptionAlgorithm
-		level EncryptionLevel
+		payload []byte
+		name    string
+		algo    EncryptionAlgorithm
+		level   EncryptionLevel
 	}{
-		{"AESCFB_AES128", EncryptionAlgoAESCFB, EncryptionLevelAES128},
-		{"AESCFB_AES192", EncryptionAlgoAESCFB, EncryptionLevelAES192},
-		{"AESCFB_AES256", EncryptionAlgoAESCFB, EncryptionLevelAES256},
-		{"AESGCM_AES128", EncryptionAlgoAESGCM, EncryptionLevelAES128},
-		{"AESGCM_AES192", EncryptionAlgoAESGCM, EncryptionLevelAES192},
-		{"AESGCM_AES256", EncryptionAlgoAESGCM, EncryptionLevelAES256},
+		{[]byte("small payload"), "SMALL_AESCFB_AES128", EncryptionAlgoAESCFB, EncryptionLevelAES128},
+		{[]byte("small payload"), "SMALL_AESCFB_AES192", EncryptionAlgoAESCFB, EncryptionLevelAES192},
+		{[]byte("small payload"), "SMALL_AESCFB_AES256", EncryptionAlgoAESCFB, EncryptionLevelAES256},
+		{[]byte("small payload"), "SMALL_AESGCM_AES128", EncryptionAlgoAESGCM, EncryptionLevelAES128},
+		{[]byte("small payload"), "SMALL_AESGCM_AES192", EncryptionAlgoAESGCM, EncryptionLevelAES192},
+		{[]byte("small payload"), "SMALL_AESGCM_AES256", EncryptionAlgoAESGCM, EncryptionLevelAES256},
+		{make([]byte, 2*1024*1024), "LARGE_AESCFB_AES128", EncryptionAlgoAESCFB, EncryptionLevelAES128},
+		{make([]byte, 2*1024*1024), "LARGE_AESCFB_AES192", EncryptionAlgoAESCFB, EncryptionLevelAES192},
+		{make([]byte, 2*1024*1024), "LARGE_AESCFB_AES256", EncryptionAlgoAESCFB, EncryptionLevelAES256},
+		{make([]byte, 2*1024*1024), "LARGE_AESGCM_AES128", EncryptionAlgoAESGCM, EncryptionLevelAES128},
+		{make([]byte, 2*1024*1024), "LARGE_AESGCM_AES192", EncryptionAlgoAESGCM, EncryptionLevelAES192},
+		{make([]byte, 2*1024*1024), "LARGE_AESGCM_AES256", EncryptionAlgoAESGCM, EncryptionLevelAES256},
 	}
 
 	for _, tt := range tests {
@@ -43,7 +50,7 @@ func BenchmarkEncryptDecrypt(b *testing.B) {
 			key, err := generateRandomString(int(tt.level / 8))
 			require.NoError(b, err)
 
-			plaintext := loremIpsumDolor
+			plaintext := tt.payload
 
 			b.ResetTimer()
 			for i := 0; i < b.N; i++ {