def METHOD_NAME(self, mock_feature_store_entity_operations: FeatureStoreEntityOperations):
    name = "random_name"
    featureStoreEntity_version = Mock(
        FeaturestoreEntityVersion(properties=Mock(FeaturestoreEntityVersionProperties()))
    )
    featureStoreEntity_version.properties.stage = "Archived"
    version = "1"
    mock_feature_store_entity_operations._operation.get.return_value = featureStoreEntity_version
    mock_feature_store_entity_operations.restore(name=name, version=version)
    mock_feature_store_entity_operations._operation.begin_create_or_update.assert_called_once_with(
        name=name,
        version=version,
        workspace_name=mock_feature_store_entity_operations._workspace_name,
        body=featureStoreEntity_version,
        resource_group_name=mock_feature_store_entity_operations._resource_group_name,
    )