[Easy-setup] Unit test case fixes for Mediator SDK