IOS - Mocking UserDefault перед загрузкой контроллера представления

В настоящее время я работаю над тестированием своего приложения, и у меня возникла проблема с издевательством над пользовательскими настройками по умолчанию. Позвольте мне сначала показать вам мою установку:

вот как я издеваюсь над пользовательскими настройками по умолчанию:

class MockUserDefaults: UserDefaults {

    typealias FakeData = Dictionary<String, Any?>
    var data: FakeData

    convenience init() {
        self.init(suiteName: "mocking")!
    }

    override init?(suiteName suitename: String?) {
        data = FakeDefaults()
        UserDefaults().removePersistentDomain(forName: suitename!)
        super.init(suiteName: suitename)
    }

    override func object(forKey defaultName: String) -> Any? {
        if let data = data[defaultName] {
            return data
        }
        return nil
    }

    override func set(_ value: Any?, forKey defaultName: String) {
        if defaultName == "favs"{
            data[defaultName] = value
        }

    }

}

У меня есть переменная в моем контроллере представления, называемая: userDefaults, и я установил ее так:

var userDefaults : UserDefaults {
        if (NSClassFromString("XCTest") != nil) {
            return MockUserDefaults()
        }
        return UserDefaults.standard
    }

эта переменная на самом деле является расширением протокола, которому настроенный uiviewcontroller соответствует ему, чтобы убедиться, что все мои контроллеры представления имеют эту переменную.

У меня также есть переменная в myViewcontroller под названием favouriteMovie, которую я установил следующим образом:

private var favoriteMovie: Favorite? {
    if let favoriteString = userDefaults.value(forKey: "favs") as? String {
        return favorites.first(where: {$0.name == favoriteString})
    }
    return nil   
}

Теперь вот где проблема, когда я иду и пытаюсь протестировать этот контроллер представления, мне нужно установить userDefault с объектом, например:

    myviewController.userDefaults.set("avengers", forKey: "favs")

перед запуском теста, но проблема в том, что переменная favouriteMovie всегда возвращает nil, и мне нужно, чтобы она возвращала объект до запуска теста. Любая помощь. Заранее спасибо.

ОБНОВИТЬ :

это протокол:

protocol Mockable: class {
    var userDefaults: UserDefaults { get }
}

это расширение:

extension UIViewController: Mockable {}
extension Mockable {
        var userDefaults : UserDefaults {
            if (NSClassFromString("XCTest") != nil) {
                return MockUserDefaults()
            }
            return UserDefaults.standard
        }
}