404, 401 error handled

Author: pashidlos

lombok.config

@@ -1,3 +1,2 @@
 # This file is generated by the 'io.freefair.lombok' Gradle plugin
 config.stopBubbling = true
-lombok.addLombokGeneratedAnnotation = true

src/main/java/io/visual_regression_tracker/sdk_java/VisualRegressionTracker.java

@@ -5,7 +5,12 @@
 import io.visual_regression_tracker.sdk_java.request.TestRunRequest;
 import io.visual_regression_tracker.sdk_java.response.BuildResponse;
 import io.visual_regression_tracker.sdk_java.response.TestRunResponse;
-import okhttp3.*;
+import okhttp3.MediaType;
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.RequestBody;
+import okhttp3.Response;
+import okhttp3.ResponseBody;
 
 import java.io.IOException;
 
@@ -39,13 +44,17 @@ void startBuild() throws IOException {
                     .post(body)
                     .build();
 
-            try (ResponseBody responseBody = client.newCall(request).execute().body()) {
+            try (Response response = client.newCall(request).execute()) {
+                if (response.code() == 404) {
+                    throw new TestRunException("Project not found");
+                }
+                if (response.code() == 401) {
+                    throw new TestRunException("Unauthorized");
+                }
+                ResponseBody responseBody = response.body();
                 BuildResponse buildDTO = new Gson().fromJson(responseBody.string(), BuildResponse.class);
                 this.buildId = buildDTO.getId();
                 this.projectId = buildDTO.getProjectId();
-            } catch(Exception ex){
-                System.out.println(ex.getMessage());
-                throw ex;
             }
         }
     }

src/test/java/io/visual_regression_tracker/sdk_java/VisualRegressionTrackerTest.java

@@ -24,9 +24,9 @@ public class VisualRegressionTrackerTest {
     MockWebServer server;
     VisualRegressionTracker vrt;
     VisualRegressionTrackerConfig config = new VisualRegressionTrackerConfig(
-            "http://localhost",
+            "http://localhost:4200",
             "733c148e-ef70-4e6d-9ae5-ab22263697cc",
-            "BAZ0EG0PRH4CRQPH19ZKAVADBP9E1",
+            "XHGDZDFD3GMJDNM87JKEMP0JS1G5",
             "develop"
     );
 
@@ -71,18 +71,33 @@ void shouldStartBuild() throws IOException, InterruptedException {
     }
 
     @Test
-    void shouldThrowExceptionIfProjectNotFound() throws IOException, InterruptedException {
-        String projectId = "non-existing";
-        BuildRequest buildRequest = BuildRequest.builder()
-                .branchName(this.config.branchName)
-                .project(this.config.project)
-                .build();
+    void shouldThrowExceptionIfProjectNotFound() throws IOException {
         server.enqueue(new MockResponse()
-                .setHttp2ErrorCode(404)
+                .setResponseCode(404)
                 .setBody("{\r\n  \"statusCode\": 404,\r\n  \"message\": \"Project not found\"\r\n}"));
 
-        vrt.startBuild();
+        String exceptionMessage = "";
+        try {
+            vrt.startBuild();
+        } catch (TestRunException ex) {
+            exceptionMessage = ex.getMessage();
+        }
+        MatcherAssert.assertThat(exceptionMessage, CoreMatchers.is("Project not found"));
+    }
 
+    @Test
+    void shouldThrowExceptionIfUnauthorized() throws IOException {
+        server.enqueue(new MockResponse()
+                .setResponseCode(401)
+                .setBody("{\r\n  \"statusCode\": 401,\r\n  \"message\": \"Unauthorized\"\r\n}"));
+
+        String exceptionMessage = "";
+        try {
+            vrt.startBuild();
+        } catch (TestRunException ex) {
+            exceptionMessage = ex.getMessage();
+        }
+        MatcherAssert.assertThat(exceptionMessage, CoreMatchers.is("Unauthorized"));
     }
 
     @Test
@@ -144,7 +159,6 @@ public Object[][] shouldTrackThrowExceptionCases() {
         };
     }
 
-
     @Test(dataProvider = "shouldTrackThrowExceptionCases")
     public void shouldTrackThrowException(TestRunResponse testRunResponse, String expectedExceptionMessage) throws IOException {
         VisualRegressionTracker vrtMocked = Mockito.mock(VisualRegressionTracker.class);