/
NDPTransportChunkingDocTest.java
108 lines (93 loc) · 3.65 KB
/
NDPTransportChunkingDocTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
* Copyright (c) 2002-2015 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.neo4j.ndp.docs.v1;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import org.neo4j.ndp.transport.socket.Chunker;
import static java.lang.Integer.parseInt;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.neo4j.kernel.impl.util.Codecs.decodeHexString;
import static org.neo4j.kernel.impl.util.HexPrinter.hex;
import static org.neo4j.ndp.docs.v1.DocSerializationExample.serialization_example;
import static org.neo4j.ndp.docs.v1.DocsRepository.docs;
@RunWith( Parameterized.class )
public class NDPTransportChunkingDocTest
{
@Parameterized.Parameter( 0 )
public DocSerializationExample example;
@Parameterized.Parameters( name = "{0}" )
public static Collection<Object[]> documentedChunkingExamples()
{
Collection<Object[]> mappings = new ArrayList<>();
// Load the documented mappings
for ( DocSerializationExample ex : docs().read(
"dev/transport.asciidoc",
"code[data-lang=\"ndp_chunking_example\"]",
serialization_example ) )
{
mappings.add( new Object[]{ex} );
}
return mappings;
}
@Test
public void serializingLeadsToSpecifiedOutput() throws Throwable
{
int chunkSize = parseInt( example.attribute( "Chunk size" ) );
assertThat( "Serialized data should match documented representation:\n" + example,
normalizedHex( example.serializedData() ),
equalTo( serialize( chunkSize, messages( example ) ) ) );
}
private String serialize( int maxChunkSize, byte[][] messages ) throws IOException
{
return normalizedHex( hex( Chunker.chunk( maxChunkSize, messages ) ) );
}
private byte[][] messages( DocSerializationExample ex )
{
// Not very generic, but gets the job done for now.
if ( ex.attribute( "Message data" ) != null )
{
String hex = ex.attribute( "Message data" ).replace( " ", "" );
return new byte[][]{decodeHexString( hex )};
}
else
{
return new byte[][]{
decodeHexString( ex.attribute( "Message 1 data" ).replace( " ", "" ) ),
decodeHexString( ex.attribute( "Message 2 data" ).replace( " ", "" ) )};
}
}
/** Convert a hex string into a normalized format for string comparison */
private static String normalizedHex( String dirtyHex )
{
StringBuilder str = new StringBuilder( dirtyHex.replace( "\n", "" ).replace( " ", "" ) );
int idx = str.length() - 2;
while ( idx > 0 )
{
str.insert( idx, " " );
idx = idx - 2;
}
return str.toString().toUpperCase();
}
}