Migrating tests to fluentmock