Add test

Author: mayya-sharipova

server/src/test/java/org/elasticsearch/search/sort/BucketedSortForIntsTests.java

@@ -0,0 +1,54 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the "Elastic License
+ * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
+ * Public License v 1"; you may not use this file except in compliance with, at
+ * your election, the "Elastic License 2.0", the "GNU Affero General Public
+ * License v3.0 only", or the "Server Side Public License, v 1".
+ */
+
+package org.elasticsearch.search.sort;
+
+import org.apache.lucene.index.LeafReaderContext;
+import org.elasticsearch.search.DocValueFormat;
+
+public class BucketedSortForIntsTests extends BucketedSortTestCase<BucketedSort.ForInts> {
+    @Override
+    public BucketedSort.ForInts build(
+        SortOrder sortOrder,
+        DocValueFormat format,
+        int bucketSize,
+        BucketedSort.ExtraData extra,
+        double[] values
+    ) {
+        return new BucketedSort.ForInts(bigArrays(), sortOrder, format, bucketSize, extra) {
+            @Override
+            public Leaf forLeaf(LeafReaderContext ctx) {
+                return new Leaf(ctx) {
+                    int index = -1;
+
+                    @Override
+                    protected boolean advanceExact(int doc) {
+                        index = doc;
+                        return doc < values.length;
+                    }
+
+                    @Override
+                    protected int docValue() {
+                        return (int) values[index];
+                    }
+                };
+            }
+        };
+    }
+
+    @Override
+    protected SortValue expectedSortValue(double v) {
+        return SortValue.from((int) v);
+    }
+
+    @Override
+    protected double randomValue() {
+        return randomInt();
+    }
+}

server/src/test/java/org/elasticsearch/search/sort/BucketedSortForLongsTests.java

@@ -50,6 +50,6 @@ protected SortValue expectedSortValue(double v) {
     @Override
     protected double randomValue() {
         // 2L^50 fits in the mantisa of a double which the test sort of needs.
-        return randomLongBetween(-2L ^ 50, 2L ^ 50);
+        return randomLongBetween(-(1L << 50), (1L << 50));
     }
 }