From 29c433518fb1054afe6f6dfdbf97ae68be44ab90 Mon Sep 17 00:00:00 2001
From: Anton Persson <anton.persson@neotechnology.com>
Date: Wed, 26 Sep 2018 14:39:19 +0200
Subject: [PATCH] Use RandomValues in NumberFullScanNonUniqueIndexSamplerTest

---
 ...mberFullScanNonUniqueIndexSamplerTest.java | 29 ++++++++++---------
 1 file changed, 15 insertions(+), 14 deletions(-)

diff --git a/community/kernel/src/test/java/org/neo4j/kernel/impl/index/schema/NumberFullScanNonUniqueIndexSamplerTest.java b/community/kernel/src/test/java/org/neo4j/kernel/impl/index/schema/NumberFullScanNonUniqueIndexSamplerTest.java
index 04d50d2494b1a..4cd49381738ef 100644
--- a/community/kernel/src/test/java/org/neo4j/kernel/impl/index/schema/NumberFullScanNonUniqueIndexSamplerTest.java
+++ b/community/kernel/src/test/java/org/neo4j/kernel/impl/index/schema/NumberFullScanNonUniqueIndexSamplerTest.java
@@ -26,15 +26,15 @@
 import org.neo4j.index.internal.gbptree.GBPTree;
 import org.neo4j.index.internal.gbptree.Writer;
 import org.neo4j.io.pagecache.IOLimiter;
-import org.neo4j.kernel.api.index.IndexEntryUpdate;
-import org.neo4j.storageengine.api.schema.IndexDescriptor;
 import org.neo4j.storageengine.api.schema.IndexSample;
-import org.neo4j.values.storable.Values;
+import org.neo4j.values.storable.NumberValue;
+import org.neo4j.values.storable.RandomValues;
+import org.neo4j.values.storable.Value;
+import org.neo4j.values.storable.ValueGroup;
 
 import static org.junit.Assert.assertEquals;
 import static org.neo4j.kernel.impl.index.schema.LayoutTestUtil.countUniqueValues;
 import static org.neo4j.kernel.impl.index.schema.NativeIndexKey.Inclusion.NEUTRAL;
-import static org.neo4j.values.storable.Values.values;
 
 public class NumberFullScanNonUniqueIndexSamplerTest extends NativeIndexTestUtil<NumberIndexKey,NativeIndexValue>
 {
@@ -42,7 +42,7 @@ public class NumberFullScanNonUniqueIndexSamplerTest extends NativeIndexTestUtil
     public void shouldIncludeAllValuesInTree() throws Exception
     {
         // GIVEN
-        Number[] values = generateNumberValues();
+        Value[] values = generateNumberValues();
         buildTree( values );
 
         // WHEN
@@ -60,18 +60,19 @@ public void shouldIncludeAllValuesInTree() throws Exception
         assertEquals( values.length, sample.indexSize() );
     }
 
-    private Number[] generateNumberValues()
+    private Value[] generateNumberValues()
     {
-        IndexEntryUpdate<IndexDescriptor>[] updates = layoutUtil.someUpdates();
-        Number[] result = new Number[updates.length];
-        for ( int i = 0; i < updates.length; i++ )
+        RandomValues.Type[] numberTypes = RandomValues.including( t -> t.valueGroup == ValueGroup.NUMBER );
+        int size = 20;
+        Value[] result = new NumberValue[size];
+        for ( int i = 0; i < size; i++ )
         {
-            result[i] = (Number) updates[i].values()[0].asObject();
+            result[i] = randomValues.nextValueOfTypes( numberTypes );
         }
         return result;
     }
 
-    private void buildTree( Number[] values ) throws IOException
+    private void buildTree( Value[] values ) throws IOException
     {
         try ( GBPTree<NumberIndexKey,NativeIndexValue> gbpTree = getTree() )
         {
@@ -80,11 +81,11 @@ private void buildTree( Number[] values ) throws IOException
                 NumberIndexKey key = layout.newKey();
                 NativeIndexValue value = layout.newValue();
                 long nodeId = 0;
-                for ( Number number : values )
+                for ( Value number : values )
                 {
                     key.initialize( nodeId );
-                    key.initFromValue( 0, Values.of( number ), NEUTRAL );
-                    value.from( values( number ) );
+                    key.initFromValue( 0, number, NEUTRAL );
+                    value.from( number );
                     writer.put( key, value );
                     nodeId++;
                 }