Add api to update engagement states in gitlab project. Add api to get engagements by user (#28)

Author: mcanoy

src/main/java/com/redhat/labs/lodestar/engagements/repository/EngagementRepository.java

@@ -42,6 +42,22 @@ public List<Engagement> getEngagements(PageFilter pageFilter) {
                 .page(pageFilter.getPage(), pageFilter.getPageSize()).list();
     }
 
+    /**
+     *
+     * @param email the email id to match (tech lead, EL or customer contact)
+     * @param engagementIds a list of engagement ids that the user has participated in (from participant service)
+     * @return A list of engagements that include the email or engagement ids of participant
+     */
+    public List<Engagement> getEngagementsByUser(PageFilter pageFilter, String email, Set<String> engagementIds) {
+        List<Bson> ors = new ArrayList<>();
+
+        ors.add(eq("engagementLeadEmail", email));
+        ors.add(eq("technicalLeadEmail", email));
+        ors.add(eq("customerContactEmail", email));
+        ors.add(in("uuid", engagementIds));
+        return mongoCollection().find(or(ors)).sort(pageFilter.getBsonSort()).skip(pageFilter.getStartAt()).limit(pageFilter.getPageSize()).into(new ArrayList<>());
+    }
+
     public List<Engagement> getEngagements(PageFilter pageFilter, Set<String> regions, Set<String> types) {
         String query = "";
         Map<String, Object> params = new HashMap<>();

src/main/java/com/redhat/labs/lodestar/engagements/resource/EngagementResource.java

@@ -53,6 +53,7 @@ public Response getEngagements(@BeanParam PageFilter pagingFilter, @QueryParam("
         }
         return Response.ok(engagements).header(TOTAL_HEADER, total).build(); //no paging yet
     }
+
     @GET
     @Path("inStates")
     public Response getEngagements(@QueryParam("inStates") Set<EngagementState> states) {
@@ -64,7 +65,13 @@ public Response getEngagements(@QueryParam("inStates") Set<EngagementState> stat
         return Response.ok(engagements).header(TOTAL_HEADER, total).build();
     }
 
-    
+    @GET
+    @Path("byUser/{email}")
+    public Response getEngagementsForUser(@PathParam("email") String email, @QueryParam("engagementUuids") Set<String> engagementUuids, @BeanParam PageFilter pagingFilter) {
+        List<Engagement> engagements = engagementService.getEngagementsForUser(pagingFilter, email, engagementUuids);
+        return Response.ok(engagements).header(TOTAL_HEADER, engagements.size()).build();
+    }
+
     @GET
     @Path("category/{category}")
     @Operation(summary = "Gets a list of engagements that have use the category input.")
@@ -170,6 +177,7 @@ public Response updateLastUpdate(@PathParam("uuid") String uuid) {
         engagementService.updateLastUpdate(cleanUuid);
         return Response.ok().build();
     }
+
 
     @PUT
     @Path("refresh")
@@ -184,6 +192,13 @@ public Response refresh(@QueryParam("uuids") Set<String> uuids) {
 
         return Response.ok().header(TOTAL_HEADER, newCount).build();
     }
+
+    @PUT
+    @Path("refresh/state")
+    public Response updateStatesInGitlab() {
+        engagementService.updateAllEngagementStates();
+        return Response.ok().build();
+    }
 
     @GET
     @Path("suggest") 

src/main/java/com/redhat/labs/lodestar/engagements/service/EngagementService.java

@@ -109,6 +109,14 @@ void checkLastUpdate() {
         }
     }
 
+    public void updateAllEngagementStates() {
+        LOGGER.debug("Updating all states in gitlab");
+        engagementRepository.findAll().stream().forEach(e -> {
+                e.setCurrentState(e.getState());
+                bus.publish(UPDATE_STATUS, e);
+        });
+    }
+
     public void create(Engagement engagement) {
         if(engagement.getUuid() != null) { //This will not be true as long as we persist in lodestar-backend
             throw new WebApplicationException("UUID cannot be set before create", Status.BAD_REQUEST);
@@ -270,6 +278,10 @@ public List<Engagement> getEngagements(PageFilter pageFilter, Set<String> region
 
     }
 
+    public List<Engagement> getEngagementsForUser(PageFilter pageFilter, String userEmail, Set<String> engagementUuids) {
+        return engagementRepository.getEngagementsByUser(pageFilter, userEmail, engagementUuids);
+    }
+
     public long countEngagements(String input, String category, Set<String> regions, Set<String> types, Set<EngagementState> states) {
         return engagementRepository.countEngagements(input, category, regions, types, states);
     }

src/test/java/com/redhat/labs/lodestar/engagements/resource/EngagementResourceTest.java

@@ -254,6 +254,23 @@ void testGetEngagementCountByStatus() {
 
     }
 
+    @Test
+    void testGetEngagementByUser() {
+        String email = "Calem@calm.com";
+        given().pathParam("email", email).when().get("byUser/{email}")
+                        .then().body("size()", equalTo(0));
+
+        List<Engagement> engagements = engagementService.getEngagements();
+        engagements.forEach(e-> {
+            e.setEngagementLeadEmail(email);
+            engagementService.update(e, false);
+        });
+
+        given().pathParam("email", email).when().get("byUser/{email}")
+                .then().body("size()", equalTo(2));
+
+    }
+
     @Test
     void testHeadLastUpdate() {
         String uuid = "uuid1";
@@ -306,9 +323,14 @@ void testCreateEngagementNoType() {
 
     @Test
     void testCreateEngagementBadRequest() {
-        Engagement engagement = Engagement.builder().customerName("").region("na").type("Residency").build();
+        Engagement engagement = Engagement.builder().customerName("ab!!!").name("abcd").region("na").type("Residency").build();
         String e = new JsonMarshaller().toJson(engagement);
-        given().contentType(ContentType.JSON).when().body(e).post().then().statusCode(Status.BAD_REQUEST.getStatusCode()).header("Location", nullValue());
+        given().contentType(ContentType.JSON)
+                .when().body(e).post()
+                .then().statusCode(Status.BAD_REQUEST.getStatusCode())
+                    .header("Location", nullValue())
+                    .body("parameter_violations.size()", equalTo(1))
+                    .body("parameter_violations[0].value", equalTo("ab!!!"));
     }
 
     @Test
@@ -516,4 +538,9 @@ void testGetWithCategory() {
         given().pathParam("category", "philanthropy").queryParam("sort", "name|desc,customerName").when().get("category/{category}")
                 .then().statusCode(200).body("size()", equalTo(2)).header("x-total-engagements", equalTo("2"));
     }
+
+    @Test
+    void testUpdateStates() {
+        given().when().put("refresh/state").then().statusCode(200);
+    }
 }