From a8fd2ef9549e52ad75d15fbcf5a26f043fe5aa20 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Tue, 7 Jun 2022 06:38:35 +0300
Subject: [PATCH] [#227] netmap: Benchmark JSON tests

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
---
 netmap/json_test.go | 56 ++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 55 insertions(+), 1 deletion(-)

diff --git a/netmap/json_test.go b/netmap/json_test.go
index 1d0236bc..e0fd0e57 100644
--- a/netmap/json_test.go
+++ b/netmap/json_test.go
@@ -26,7 +26,7 @@ type TestCase struct {
 	}
 }
 
-func compareNodes(t *testing.T, expected [][]int, nodes Nodes, actual []Nodes) {
+func compareNodes(t testing.TB, expected [][]int, nodes Nodes, actual []Nodes) {
 	require.Equal(t, len(expected), len(actual))
 	for i := range expected {
 		require.Equal(t, len(expected[i]), len(actual[i]))
@@ -80,6 +80,60 @@ func TestPlacementPolicy_Interopability(t *testing.T) {
 		})
 	}
 }
+
+func BenchmarkPlacementPolicyInteropability(b *testing.B) {
+	const testsDir = "./json_tests"
+
+	f, err := os.Open(testsDir)
+	require.NoError(b, err)
+
+	ds, err := f.ReadDir(0)
+	require.NoError(b, err)
+
+	for i := range ds {
+		bs, err := ioutil.ReadFile(filepath.Join(testsDir, ds[i].Name()))
+		require.NoError(b, err)
+
+		var tc TestCase
+		require.NoError(b, json.Unmarshal(bs, &tc), "cannot unmarshal %s", ds[i].Name())
+
+		b.Run(tc.Name, func(b *testing.B) {
+			nodes := NodesFromInfo(tc.Nodes)
+			nm, err := NewNetmap(nodes)
+			require.NoError(b, err)
+
+			for name, tt := range tc.Tests {
+				b.Run(name, func(b *testing.B) {
+					b.ReportAllocs()
+					b.ResetTimer()
+					for i := 0; i < b.N; i++ {
+						b.StartTimer()
+						v, err := nm.GetContainerNodes(&tt.Policy, tt.Pivot)
+						b.StopTimer()
+						if tt.Result == nil {
+							require.Error(b, err)
+							require.Contains(b, err.Error(), tt.Error)
+						} else {
+							require.NoError(b, err)
+
+							res := v.Replicas()
+							compareNodes(b, tt.Result, nodes, res)
+
+							if tt.Placement.Result != nil {
+								b.StartTimer()
+								res, err := nm.GetPlacementVectors(v, tt.Placement.Pivot)
+								b.StopTimer()
+								require.NoError(b, err)
+								compareNodes(b, tt.Placement.Result, nodes, res)
+							}
+						}
+					}
+				})
+			}
+		})
+	}
+}
+
 func BenchmarkManySelects(b *testing.B) {
 	testsFile := filepath.Join("json_tests", "many_selects.json")
 	bs, err := ioutil.ReadFile(testsFile)