001/*
002 * SonarQube, open source software quality management tool.
003 * Copyright (C) 2008-2014 SonarSource
004 * mailto:contact AT sonarsource DOT com
005 *
006 * SonarQube is free software; you can redistribute it and/or
007 * modify it under the terms of the GNU Lesser General Public
008 * License as published by the Free Software Foundation; either
009 * version 3 of the License, or (at your option) any later version.
010 *
011 * SonarQube is distributed in the hope that it will be useful,
012 * but WITHOUT ANY WARRANTY; without even the implied warranty of
013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
014 * Lesser General Public License for more details.
015 *
016 * You should have received a copy of the GNU Lesser General Public License
017 * along with this program; if not, write to the Free Software Foundation,
018 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
019 */
020package org.sonar.api.server.ws.internal;
021
022import com.google.common.collect.Maps;
023import java.io.InputStream;
024import java.util.Map;
025import javax.annotation.Nullable;
026import org.apache.commons.io.IOUtils;
027import org.sonar.api.server.ws.Request;
028
029import static com.google.common.base.Preconditions.checkNotNull;
030
031/**
032 * Fake implementation of {@link org.sonar.api.server.ws.Request} used
033 * for testing. Call the method {@link #setParam(String, String)} to
034 * emulate some parameter values.
035 */
036public class SimpleGetRequest extends Request {
037
038  private final Map<String, String> params = Maps.newHashMap();
039  private String mediaType = "application/json";
040
041  @Override
042  public String method() {
043    return "GET";
044  }
045
046  @Override
047  public String getMediaType() {
048    return mediaType;
049  }
050
051  public SimpleGetRequest setMediaType(String mediaType) {
052    checkNotNull(mediaType);
053    this.mediaType = mediaType;
054    return this;
055  }
056
057  @Override
058  public boolean hasParam(String key) {
059    return params.keySet().contains(key);
060  }
061
062  @Override
063  public String param(String key) {
064    return params.get(key);
065  }
066
067  @Override
068  public InputStream paramAsInputStream(String key) {
069    return IOUtils.toInputStream(param(key));
070  }
071
072  public SimpleGetRequest setParam(String key, @Nullable String value) {
073    if (value != null) {
074      params.put(key, value);
075    }
076    return this;
077  }
078
079}