@@ -36,13 +36,15 @@ public class StaffImageApiTest {
3636
3737 private RequestSpecification requestSpec ;
3838 private ResponseSpecification responseSpec ;
39+ private ResponseSpecification responseSpec404 ;
3940
4041 @ BeforeEach
4142 public void setup () {
4243 Utils .initializeRESTAssured ();
4344 this .requestSpec = new RequestSpecBuilder ().setContentType (ContentType .JSON ).build ();
4445 this .requestSpec .header ("Authorization" , "Basic " + Utils .loginIntoServerAndGetBase64EncodedAuthenticationKey ());
4546 this .responseSpec = new ResponseSpecBuilder ().expectStatusCode (200 ).build ();
47+ this .responseSpec404 = new ResponseSpecBuilder ().expectStatusCode (404 ).build ();
4648 }
4749
4850 @ Test
@@ -57,6 +59,7 @@ public void getStaffImage() {
5759 Integer staffId = StaffHelper .createStaff (this .requestSpec , this .responseSpec );
5860 Integer imageId = ImageHelper .createImageForStaff (this .requestSpec , this .responseSpec , staffId );
5961 assertNotNull (imageId , "Image id should not be null" );
62+
6063 String imageAsText = ImageHelper .getStaffImageAsText (this .requestSpec , this .responseSpec , staffId );
6164 assertNotNull ("Image id should not be null" , imageAsText );
6265 assertEquals (ImageHelper .generateImageAsText (), imageAsText );
@@ -67,6 +70,7 @@ public void getStaffImageAsBinary() {
6770 Integer staffId = StaffHelper .createStaff (this .requestSpec , this .responseSpec );
6871 Integer imageId = ImageHelper .createImageForStaff (this .requestSpec , this .responseSpec , staffId );
6972 assertNotNull (imageId , "Image id should not be null" );
73+
7074 byte [] imageAsBytes = ImageHelper .getStaffImageAsBinary (this .requestSpec , this .responseSpec , staffId );
7175 assertNotNull (imageAsBytes , "Image content should not be null" );
7276 assertEquals (251 , imageAsBytes .length );
@@ -77,6 +81,7 @@ public void updateImage() {
7781 Integer staffId = StaffHelper .createStaff (this .requestSpec , this .responseSpec );
7882 Integer imageId = ImageHelper .createImageForStaff (this .requestSpec , this .responseSpec , staffId );
7983 assertNotNull (imageId , "Image id should not be null" );
84+
8085 imageId = ImageHelper .updateImageForStaff (this .requestSpec , this .responseSpec , staffId );
8186 assertNotNull (imageId , "Image id should not be null" );
8287 }
@@ -86,7 +91,11 @@ public void deleteStaffImage() {
8691 Integer staffId = StaffHelper .createStaff (this .requestSpec , this .responseSpec );
8792 Integer imageId = ImageHelper .createImageForStaff (this .requestSpec , this .responseSpec , staffId );
8893 assertNotNull (imageId , "Image id should not be null" );
94+
8995 imageId = ImageHelper .deleteStaffImage (this .requestSpec , this .responseSpec , staffId );
9096 assertNotNull (imageId , "Image id should not be null" );
97+
98+ ImageHelper .getStaffImageAsBinary (this .requestSpec , this .responseSpec404 , staffId );
99+ // No additional Assertion required, the point is the responseSpec404
91100 }
92101}
0 commit comments