Fix flakiness in SnapshotReducerSpec
Added mode to differentiate set/state type features.