package com.atlassian.sal.testresources.usersettings;
   
   import com.atlassian.sal.api.user.UserKey;
   import com.atlassian.sal.api.usersettings.UserSettings;
   import com.atlassian.sal.api.usersettings.UserSettingsBuilder;
   import com.atlassian.sal.api.usersettings.UserSettingsService;
   import com.atlassian.sal.core.usersettings.DefaultUserSettings;
   import com.google.common.base.Function;
   
  import java.util.HashMap;
  import java.util.Map;
  
  public class MockUserSettingsService implements UserSettingsService {
      private final Map<String, UserSettings> settingsMap = new HashMap<>();
  
      @Override
      public UserSettings getUserSettings(String userName) {
          return settingsMap.containsKey(userName) ? settingsMap.get(userName) : settingsMap.put(userName, DefaultUserSettings.builder().build());
      }
  
      @Override
      public UserSettings getUserSettings(UserKey user) {
          return getUserSettings(user.getStringValue());
      }
  
      @Override
      public void updateUserSettings(String userName, Function<UserSettingsBuilder, UserSettings> updateFunction) {
          if (settingsMap.containsKey(userName)) {
              settingsMap.put(userName, updateFunction.apply(DefaultUserSettings.builder(settingsMap.get(userName))));
          } else {
              settingsMap.put(userName, updateFunction.apply(DefaultUserSettings.builder()));
          }
      }
  
      @Override
      public void updateUserSettings(UserKey user, Function<UserSettingsBuilder, UserSettings> updateFunction) {
          updateUserSettings(user.getStringValue(), updateFunction);
      }
  }