/*
* Copyright 2015 ThoughtWorks, 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.thoughtworks.go.server.web;
import com.thoughtworks.go.server.web.i18n.ResolvableViewableStatus;
import com.thoughtworks.go.util.JsonTester;
import com.thoughtworks.go.util.json.JsonUrl;
import org.jmock.Expectations;
import org.jmock.Mockery;
import org.jmock.lib.legacy.ClassImposteriser;
import org.junit.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import static com.thoughtworks.go.server.web.JsonRenderer.render;
import static com.thoughtworks.go.server.web.JsonView.asMap;
import static com.thoughtworks.go.server.web.i18n.CurrentStatus.WAITING;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
public class JsonViewTest {
private Mockery mockContext = new Mockery() {
{
setImposteriser(ClassImposteriser.INSTANCE);
}
};
private GoRequestContext requestContext = mockContext.mock(GoRequestContext.class);
@Test
public void testShouldReturnOutputWithoutWhitespaceThatIsNotAllowedInHeaders() throws Exception {
JsonView view = new JsonView(requestContext);
MockHttpServletResponse response = new MockHttpServletResponse();
Map<String, Object> map = new LinkedHashMap<>();
map.put("key", "\r\t\n");
view.renderMergedOutputModel(asMap(map), new MockHttpServletRequest(), response);
String json = response.getContentAsString();
new JsonTester(json).is(
"{ 'key' : '\\r\\t\\n' }"
);
}
@Test
public void testShouldRenderEmptyMap() throws Exception {
JsonView view = new JsonView();
String json = view.renderJson(new LinkedHashMap());
new JsonTester(json).is(
"{ }"
);
}
@Test
public void testShouldRenderAllKeyValuePairsFromMap() throws Exception {
Map<String, Object> map = new LinkedHashMap<>();
map.put("key1", "value1");
map.put("key2", "value2");
String json = new JsonView().renderJson(map);
new JsonTester(json).is(
"{ 'key1' : 'value1', 'key2' : 'value2' }"
);
}
@Test
public void testShouldRenderNestedMaps() throws Exception {
Map<String, Object> map = new LinkedHashMap<>();
Map<String, Object> nestedMap = new LinkedHashMap<>();
nestedMap.put("keyA", "valueA");
map.put("key1", nestedMap);
JsonView view = new JsonView();
String json = view.renderJson(map);
new JsonTester(json).is(
"{ 'key1' : {"
+ " 'keyA' : 'valueA'"
+ " }"
+ "}"
);
}
@Test
public void testShouldRenderArray() throws Exception {
List list = new ArrayList();
Map<String, Object> nestedMap = new LinkedHashMap<>();
nestedMap.put("key1", "value1");
list.add(nestedMap);
list.add("value2");
JsonView view = new JsonView();
String json = view.renderJson(list);
new JsonTester(json).is(
"[ { 'key1' : 'value1' }, 'value2' ]"
);
}
@Test
public void testShouldRenderFakeMapsWithoutTheSurroundingMap() throws Exception {
List list = new ArrayList();
Map<String, Object> nestedMap = new LinkedHashMap<>();
nestedMap.put("key1", "value1");
list.add(nestedMap);
list.add("value2");
JsonView view = new JsonView();
String json = view.renderJson(asMap(list));
new JsonTester(json).is(
"[ { 'key1' : 'value1' }, 'value2' ]"
);
}
@Test
public void testShouldRenderI18n() {
Map<String, Object> map = new LinkedHashMap<>();
final String i18nMessage = "waiting(zh_CN)";
final ResolvableViewableStatus resolvableViewableStatus = new ResolvableViewableStatus(WAITING);
map.put("key1", resolvableViewableStatus);
JsonView view = new JsonView(requestContext);
mockContext.checking(new Expectations() {
{
one(requestContext).getMessage(with(equal(resolvableViewableStatus)));
will(returnValue(i18nMessage));
}
});
String json = view.renderJson(map);
mockContext.assertIsSatisfied();
new JsonTester(json).is(
"{ 'key1' : '" + i18nMessage + "' }"
);
}
@Test
public void testShouldRenderUrlsWithContext() {
JsonUrl url = new JsonUrl("/foo/bar/baz");
mockContext.checking(new Expectations() {
{
one(requestContext).getFullRequestPath();
will(returnValue("http://something/context"));
}
});
String json = render(url, requestContext);
assertThat(json, is("\"http://something/context/foo/bar/baz\""));
}
}