88import org .mockito .Mockito ;
99
1010import uk .gov .gchq .palisade .Context ;
11+ import uk .gov .gchq .palisade .RequestId ;
1112import uk .gov .gchq .palisade .UserId ;
1213import uk .gov .gchq .palisade .data .serialise .Serialiser ;
1314import uk .gov .gchq .palisade .data .serialise .SimpleStringSerialiser ;
@@ -82,19 +83,17 @@ public void shouldAddMultipleWithComma() {
8283 assertEquals (json , c .get (PalisadeInputFormat .REGISTER_REQUESTS_KEY ));
8384 }
8485
85- @ Test
86- public void shouldAddEmptyRequest () {
86+ @ Test ( expected = RuntimeException . class )
87+ public void shouldErrorWhenAddingEmptyRequest () {
8788 //Given
8889 Configuration c = new Configuration ();
8990 JobContext mockJob = Mockito .mock (JobContext .class );
9091 when (mockJob .getConfiguration ()).thenReturn (c );
9192 RegisterDataRequest rdr = new RegisterDataRequest ();
92- RegisterDataRequest [] rdrArray = {rdr };
93- String json = new String (JSONSerialiser .serialise (rdrArray ), PalisadeInputFormat .UTF8 );
9493 //When
9594 PalisadeInputFormat .addDataRequest (mockJob , rdr );
9695 //Then
97- assertEquals ( json , c . get ( PalisadeInputFormat . REGISTER_REQUESTS_KEY ) );
96+ fail ( );
9897 }
9998
10099 @ Test
@@ -118,10 +117,9 @@ public void addAndGetRequests() {
118117 when (mockJob .getConfiguration ()).thenReturn (c );
119118 RegisterDataRequest rdr = new RegisterDataRequest ().resourceId ("testResource" ).userId (new UserId ().id ("user" )).context (new Context ().justification ("justification" ));
120119 RegisterDataRequest rdr2 = new RegisterDataRequest ().resourceId ("testResource2" ).userId (new UserId ().id ("user2" )).context (new Context ().justification ("justification2" ));
121- RegisterDataRequest rdr3 = new RegisterDataRequest ();
122120 //When
123- PalisadeInputFormat .addDataRequests (mockJob , rdr , rdr2 , rdr3 );
124- List <RegisterDataRequest > expected = Stream .of (rdr , rdr2 , rdr3 ).collect (Collectors .toList ());
121+ PalisadeInputFormat .addDataRequests (mockJob , rdr , rdr2 );
122+ List <RegisterDataRequest > expected = Stream .of (rdr , rdr2 ).collect (Collectors .toList ());
125123 //Then
126124 assertEquals (expected , PalisadeInputFormat .getDataRequests (mockJob ));
127125 }
@@ -149,17 +147,17 @@ public void shouldThrowOnNoRequests() throws IOException {
149147 @ BeforeClass
150148 public static void setup () {
151149 request1 = new RegisterDataRequest ().resourceId ("res1" ).userId (new UserId ().id ("user1" )).context (new Context ().justification ("just1" ));
152- req1Response = new DataRequestResponse ();
153- req1Response . getResources (). put ( new StubResource ("type1" , "id1" , "format1" ), new StubConnectionDetail ("con1" ));
154- req1Response . getResources (). put ( new StubResource ("type2" , "id2" , "format2" ), new StubConnectionDetail ("con2" ));
155- req1Response . getResources (). put ( new StubResource ("type3" , "id3" , "format3" ), new StubConnectionDetail ("con3" ));
156- req1Response . getResources (). put ( new StubResource ("type4" , "id4" , "format4" ), new StubConnectionDetail ("con4" ));
157- req1Response . getResources (). put (new StubResource ("type5" , "id5" , "format5" ), new StubConnectionDetail ("con5" ));
150+ req1Response = new DataRequestResponse (). requestId ( new RequestId (). id ( "request1" ))
151+ . resource ( new StubResource ("type1" , "id1" , "format1" ), new StubConnectionDetail ("con1" ))
152+ . resource ( new StubResource ("type2" , "id2" , "format2" ), new StubConnectionDetail ("con2" ))
153+ . resource ( new StubResource ("type3" , "id3" , "format3" ), new StubConnectionDetail ("con3" ))
154+ . resource ( new StubResource ("type4" , "id4" , "format4" ), new StubConnectionDetail ("con4" ))
155+ . resource (new StubResource ("type5" , "id5" , "format5" ), new StubConnectionDetail ("con5" ));
158156
159157 request2 = new RegisterDataRequest ().resourceId ("res2" ).userId (new UserId ().id ("user2" )).context (new Context ().justification ("just2" ));
160- req2Response = new DataRequestResponse ();
161- req2Response . getResources (). put ( new StubResource ("type_a" , "id6" , "format6" ), new StubConnectionDetail ("con6" ));
162- req2Response . getResources (). put (new StubResource ("type_b" , "id7" , "format7" ), new StubConnectionDetail ("con7" ));
158+ req2Response = new DataRequestResponse (). requestId ( new RequestId (). id ( "request2" ))
159+ . resource ( new StubResource ("type_a" , "id6" , "format6" ), new StubConnectionDetail ("con6" ))
160+ . resource (new StubResource ("type_b" , "id7" , "format7" ), new StubConnectionDetail ("con7" ));
163161 }
164162
165163 /**
0 commit comments