add tests for mediator type