add sampling and make setup mocks to make sure it gets executed as many times as expected