com.google.jstestdriver.MockServer.java Source code

Introduction

Here is the source code for com.google.jstestdriver.MockServer.java

Source

/*
 * Copyright 2009 Google Inc.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.google.jstestdriver;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;
import java.util.Queue;

import com.google.common.base.Function;
import com.google.common.collect.Lists;
import com.google.gson.JsonElement;

/**
 * @author jeremiele@google.com (Jeremie Lenfant-Engelmann)
 * @author misko@google.com
 */
public class MockServer implements Server {

    private HashMap<String, Queue<String>> expectations = new HashMap<String, Queue<String>>();

    public void expect(String url, String response) {
        if (!expectations.containsKey(url)) {
            expectations.put(url, new LinkedList<String>());
        }
        expectations.get(url).add(response);
    }

    public String fetch(String url) {
        return get(url);
    }

    public String post(String url, Map<String, String> params) {
        return get(url + "?POST?" + params);
    }

    public String postJson(String url, JsonElement json) {
        return get(url + "?POST?" + json);
    }

    private String get(String request) {
        Queue<String> response = expectations.get(request);
        if (response == null || response.size() == 0) {
            System.out.println("Unexpected request: " + request + "\n in "
                    + Lists.transform(Lists.newArrayList(expectations.keySet()), new Function<String, String>() {
                        public String apply(String arg) {
                            return "\n" + arg;
                        }

                    }));
            throw new IllegalArgumentException("Unexpected request: " + request);
        }
        return response.remove();
    }

    public String startSession(String baseUrl, String id) {
        return "ID";
    }

    public void stopSession(String baseUrl, String id, String sessionId) {
        // noop
    }
}