fixes #235

Author: anshooarora

reporters/klov/src/main/java/com/aventstack/extentreports/reporter/ExtentKlovReporter.java

@@ -656,7 +656,7 @@ public synchronized void onLogAdded(Test test, Log log) {
             exceptionCollection.updateOne(new Document("_id", docException.getObjectId("_id")),
                     new Document("$set", doc));
             doc = new Document("exception", exceptionNameObjectIdCollection.get(ex.getName()))
-            		.append("exceptionName", ex.getName());
+                    .append("exceptionName", ex.getName());
             testCollection.updateOne(new Document("_id", test.getInfoMap().get(ID_KEY)), new Document("$set", doc));
             updateTestDesc(test);
         }
@@ -704,10 +704,10 @@ public void onSubscribe(Disposable d) {
             @Override
             public void onNext(MediaEntity value) {
                 try {
-                    if (value.getTest() != null)
-                        onScreenCaptureAdded(value.getTest(), (ScreenCapture) value.getMedia());
                     if (value.getLog() != null)
-                        onScreenCaptureAdded(value.getLog(), (ScreenCapture) value.getMedia());
+                        onScreenCaptureAdded(value.getLog(), value.getTest(), (ScreenCapture) value.getMedia());
+                    else if (value.getTest() != null)
+                        onScreenCaptureAdded(value.getTest(), (ScreenCapture) value.getMedia());
                 } catch (IOException e) {
                     e.printStackTrace();
                 }
@@ -728,8 +728,12 @@ public void onScreenCaptureAdded(Test test, ScreenCapture screenCapture) throws
         saveScreenCapture(test, screenCapture);
     }
 
-    public void onScreenCaptureAdded(Log log, ScreenCapture screenCapture) throws IOException {
+    public void onScreenCaptureAdded(Log log, Test test, ScreenCapture screenCapture) throws IOException {
         screenCapture.getInfoMap().put(LOG_ID_KEY, log.getInfoMap().get(ID_KEY));
+        screenCapture.getInfoMap().put(TEST_ID_KEY, test.getInfoMap().get(ID_KEY));
+        logCollection.updateOne(
+                new Document("_id", log.getInfoMap().get(ID_KEY)),
+                new Document("$set", new Document("mediaCount", 1)));
         saveScreenCapture(log, screenCapture);
     }
 

src/main/java/com/aventstack/extentreports/AbstractProcessor.java

@@ -53,9 +53,9 @@ protected void onMediaAdded(Media m, Test test) {
     }
 
     @Override
-    protected void onMediaAdded(Media m, Log log) {
+    protected void onMediaAdded(Media m, Log log, Test test) {
         tryResolvePath(m);
-        super.onMediaAdded(m, log);
+        super.onMediaAdded(m, log, test);
     }
 
     private void tryResolvePath(Media m) {

src/main/java/com/aventstack/extentreports/ExtentTest.java

@@ -364,11 +364,14 @@ public ExtentTest log(Status status, String details, Throwable t, Media media) {
                 .build();
         ExceptionInfo exceptionInfo = ExceptionInfoService.createExceptionInfo(t);
         log.setException(exceptionInfo);
-        log.addMedia(media);
         if (exceptionInfo != null)
             model.getExceptions().add(exceptionInfo);
         model.addLog(log);
         extent.onLogCreated(log, model);
+        if (media != null) {
+            log.addMedia(media);
+            extent.onMediaAdded(media, log, model);
+        }
         return this;
     }
 

src/main/java/com/aventstack/extentreports/ReactiveSubject.java

@@ -75,8 +75,8 @@ protected void onMediaAdded(Media m, Test test) {
         mediaSubject.onNext(MediaEntity.builder().media(m).test(test).build());
     }
 
-    protected void onMediaAdded(Media m, Log log) {
-        mediaSubject.onNext(MediaEntity.builder().media(m).log(log).build());
+    protected void onMediaAdded(Media m, Log log, Test test) {
+        mediaSubject.onNext(MediaEntity.builder().media(m).test(test).log(log).build());
     }
 
     protected void onFlush() {