1 /* 2 * Copyright (C) 2023 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.internal.config.sysui; 18 19 import java.util.HashMap; 20 import java.util.Map; 21 22 public class TestableFlagResolver implements SystemUiSystemPropertiesFlags.FlagResolver { 23 private Map<String, Boolean> mOverrides = new HashMap<>(); 24 private Map<String, Integer> mOverridesInt = new HashMap<>(); 25 private Map<String, String> mOverridesString = new HashMap<>(); 26 27 @Override isEnabled(SystemUiSystemPropertiesFlags.Flag flag)28 public boolean isEnabled(SystemUiSystemPropertiesFlags.Flag flag) { 29 return mOverrides.getOrDefault(flag.mSysPropKey, flag.mDefaultValue); 30 } 31 32 @Override getIntValue(SystemUiSystemPropertiesFlags.Flag flag)33 public int getIntValue(SystemUiSystemPropertiesFlags.Flag flag) { 34 return mOverridesInt.getOrDefault(flag.mSysPropKey, flag.mDefaultIntValue); 35 } 36 37 @Override getStringValue(SystemUiSystemPropertiesFlags.Flag flag)38 public String getStringValue(SystemUiSystemPropertiesFlags.Flag flag) { 39 return mOverridesString.getOrDefault(flag.mSysPropKey, flag.mDefaultStringValue); 40 } 41 setFlagOverride(SystemUiSystemPropertiesFlags.Flag flag, boolean isEnabled)42 public TestableFlagResolver setFlagOverride(SystemUiSystemPropertiesFlags.Flag flag, 43 boolean isEnabled) { 44 mOverrides.put(flag.mSysPropKey, isEnabled); 45 return this; 46 } 47 setFlagOverride(SystemUiSystemPropertiesFlags.Flag flag, int value)48 public TestableFlagResolver setFlagOverride(SystemUiSystemPropertiesFlags.Flag flag, 49 int value) { 50 mOverridesInt.put(flag.mSysPropKey, value); 51 return this; 52 } 53 setFlagOverride(SystemUiSystemPropertiesFlags.Flag flag, String value)54 public TestableFlagResolver setFlagOverride(SystemUiSystemPropertiesFlags.Flag flag, 55 String value) { 56 mOverridesString.put(flag.mSysPropKey, value); 57 return this; 58 } 59 } 60