@@ -40,31 +40,31 @@ trait ResourceTestKit[Resource <: GenericResource]
4040
4141 def createResource (entityId : String , bytes : Array [Byte ], uri : Option [String ])(implicit
4242 system : ActorSystem [_]
43- ): ResourceCreatedEvent = {
43+ ): ResourceEvent = {
4444 val probe = createTestProbe[ResourceEvent ]()
4545 subscribeProbe(probe)
4646 resourceDao.createResource(entityId, bytes, uri) await {
47- case ResourceCreated => probe.expectMessageType[ ResourceCreatedEvent ]
47+ case ResourceCreated => probe.receiveMessage()
4848 case _ => fail(s " Resource $entityId not created " )
4949 }
5050 }
5151
5252 def updateResource (entityId : String , bytes : Array [Byte ], uri : Option [String ])(implicit
5353 system : ActorSystem [_]
54- ): ResourceUpdatedEvent = {
54+ ): ResourceEvent = {
5555 val probe = createTestProbe[ResourceEvent ]()
5656 subscribeProbe(probe)
5757 resourceDao.updateResource(entityId, bytes, uri) await {
58- case ResourceUpdated => probe.expectMessageType[ ResourceUpdatedEvent ]
58+ case ResourceUpdated => probe.receiveMessage()
5959 case _ => fail(s " Resource $entityId not updated " )
6060 }
6161 }
6262
63- def deleteResource (entityId : String )(implicit system : ActorSystem [_]): ResourceDeletedEvent = {
63+ def deleteResource (entityId : String )(implicit system : ActorSystem [_]): ResourceEvent = {
6464 val probe = createTestProbe[ResourceEvent ]()
6565 subscribeProbe(probe)
6666 resourceDao.deleteResource(entityId) await {
67- case ResourceDeleted => probe.expectMessageType[ ResourceDeletedEvent ]
67+ case ResourceDeleted => probe.receiveMessage()
6868 case _ => fail(s " Resource $entityId not deleted " )
6969 }
7070 }
0 commit comments