feat(web): announce artifcat generation on TestRunFinishes

Related tickets: serenity-js#1996
viper3400 · Nov 11, 2023 · f9bad89 · f9bad89
1 parent c94c01a
commit f9bad89
Showing 1 changed file with 38 additions and 3 deletions.
diff --git a/packages/web/src/stage/crew/photographer/Photographer.ts b/packages/web/src/stage/crew/photographer/Photographer.ts
@@ -1,6 +1,7 @@
 import { ConfigurationError, LogicError } from '@serenity-js/core';
-import type { DomainEvent } from '@serenity-js/core/lib/events';
-import { ActivityFinished, ActivityStarts } from '@serenity-js/core/lib/events';
+import type { ActivityRelatedArtifactGenerated, DomainEvent} from '@serenity-js/core/lib/events';
+import { ActivityFinished, ActivityStarts, ArtifactGenerated, AsyncOperationAttempted, AsyncOperationCompleted, AsyncOperationFailed, TestRunFinishes} from '@serenity-js/core/lib/events';
+import { CorrelationId, Description,Name } from '@serenity-js/core/lib/model';
 import type { Stage, StageCrewMember } from '@serenity-js/core/lib/stage';
 
 import * as strategies from './strategies';
@@ -153,6 +154,7 @@ import * as strategies from './strategies';
  * @group Stage
  */
 export class Photographer implements StageCrewMember {
+    private artifacts : Array<ActivityRelatedArtifactGenerated>;
 
     /**
      * Instantiates a new {@apilink Photographer} configured to take photos (screenshots)
@@ -222,8 +224,41 @@ export class Photographer implements StageCrewMember {
 
         if (event instanceof ActivityStarts || event instanceof ActivityFinished) {
             this.photoTakingStrategy.considerTakingPhoto(event, this.stage).then( (artifact) => {
-                if (artifact) this.stage.announce(artifact);
+                if (artifact) this.artifacts.push(artifact)
             })
         }
+
+        if (event instanceof TestRunFinishes) {
+            const id = CorrelationId.create();
+
+            this.stage.announce(new AsyncOperationAttempted(
+                new Name(this.constructor.name),
+                new Description(`Generating Serenity images ...`),
+                id,
+                this.stage.currentTime(),
+            ));
+            try {
+                this.artifacts.forEach( (artifact) => this.stage.announce(new ArtifactGenerated(
+                    artifact.sceneId,
+                    artifact.name,
+                    artifact.artifact,
+                    artifact.timestamp
+                ))
+                );
+                this.stage.announce(new AsyncOperationCompleted(
+                    id,
+                    this.stage.currentTime(),
+                ));
+                this.artifacts = [] 
+
+            }
+            catch (error) {
+                this.stage.announce(new AsyncOperationFailed(
+                    error,
+                    id,
+                    this.stage.currentTime(),
+                ));
+            }
+        }
     }
 }