Skip to content

Commit 0af0fc6

Browse files
committed
Simplfied values method and added tests
1 parent f765762 commit 0af0fc6

File tree

2 files changed

+63
-4
lines changed

2 files changed

+63
-4
lines changed

driver/src/main/java/org/neo4j/driver/internal/InternalRecord.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,16 +25,16 @@
2525

2626
import org.neo4j.driver.internal.util.Extract;
2727
import org.neo4j.driver.internal.value.InternalValue;
28-
import org.neo4j.driver.v1.util.Function;
29-
import org.neo4j.driver.v1.util.Pair;
3028
import org.neo4j.driver.v1.Record;
3129
import org.neo4j.driver.v1.Value;
3230
import org.neo4j.driver.v1.Values;
31+
import org.neo4j.driver.v1.util.Function;
32+
import org.neo4j.driver.v1.util.Pair;
3333

3434
import static java.lang.String.format;
3535
import static org.neo4j.driver.internal.util.Format.formatPairs;
36-
import static org.neo4j.driver.v1.Values.ofValue;
3736
import static org.neo4j.driver.v1.Values.ofObject;
37+
import static org.neo4j.driver.v1.Values.ofValue;
3838

3939
public class InternalRecord implements Record
4040
{
@@ -57,7 +57,7 @@ public List<String> keys()
5757
@Override
5858
public List<Value> values()
5959
{
60-
return Values.value( asMap().values()).asList(ofValue());
60+
return Arrays.asList( values );
6161
}
6262

6363
@Override

driver/src/test/java/org/neo4j/driver/internal/InternalRecordTest.java

Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import org.junit.Test;
2222

2323
import java.util.Arrays;
24+
import java.util.Collections;
2425
import java.util.Iterator;
2526
import java.util.List;
2627
import java.util.Map;
@@ -183,6 +184,64 @@ public void testToString()
183184
assertThat( record.toString(), equalTo( "Record<{k1: 0, k2: 1}>" ) );
184185
}
185186

187+
@Test
188+
public void shouldHaveMethodToGetKeys()
189+
{
190+
//GIVEN
191+
List<String> keys = Arrays.asList( "k2", "k1" );
192+
InternalRecord record = new InternalRecord( keys, new Value[]{value( 0 ), value( 1 )} );
193+
194+
//WHEN
195+
List<String> appendedKeys = record.keys();
196+
197+
//THEN
198+
assertThat( appendedKeys, equalTo( keys ) );
199+
}
200+
201+
@Test
202+
public void emptyKeysShouldGiveEmptyList()
203+
{
204+
//GIVEN
205+
List<String> keys = Collections.emptyList();
206+
InternalRecord record = new InternalRecord( keys, new Value[]{} );
207+
208+
//WHEN
209+
List<String> appendedKeys = record.keys();
210+
211+
//THEN
212+
assertThat( appendedKeys, equalTo( keys ) );
213+
}
214+
215+
216+
@Test
217+
public void shouldHaveMethodToGetValues()
218+
{
219+
//GIVEN
220+
List<String> keys = Arrays.asList( "k2", "k1" );
221+
Value[] values = new Value[]{value( 0 ), value( 1 )};
222+
InternalRecord record = new InternalRecord( keys, values );
223+
224+
//WHEN
225+
List<Value> appendedValues = record.values();
226+
227+
//THEN
228+
assertThat( appendedValues, equalTo( Arrays.asList( values ) ) );
229+
}
230+
231+
@Test
232+
public void emptyValuesShouldGiveEmptyList()
233+
{
234+
//GIVEN
235+
List<String> keys = Collections.emptyList();
236+
Value[] values = new Value[]{};
237+
InternalRecord record = new InternalRecord( keys, values );
238+
239+
//WHEN
240+
List<Value> appendedValues = record.values();
241+
242+
//THEN
243+
assertThat( appendedValues, equalTo( Arrays.asList( values ) ) );
244+
}
186245

187246
private InternalRecord createRecord()
188247
{

0 commit comments

Comments
 (0)