/*

* Copyright 2013 Netflix, 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 feign.sax;

import org.junit.Before;

import org.junit.Rule;

import org.junit.Test;

import org.junit.rules.ExpectedException;

import org.xml.sax.helpers.DefaultHandler;

import java.io.IOException;

import java.text.ParseException;

import java.util.Collection;

import java.util.Collections;

import javax.inject.Inject;

import javax.inject.Provider;

import dagger.ObjectGraph;

import dagger.Provides;

import feign.Response;

import feign.codec.Decoder;

import static feign.Util.UTF_8;

import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertNull;

// unbound wildcards are not currently injectable in dagger.

@SuppressWarnings("rawtypes")

public class SAXDecoderTest {

static String statusFailed = ""//

+ "<soap:Envelope xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\">\n"

//

+ " <soap:Body>\n"//

+ " <ns1:getNeustarNetworkStatusResponse xmlns:ns1=\"http://webservice.api.ultra.neustar.com/v01/\">\n"

//

+ " <NeustarNetworkStatus xmlns:ns2=\"http://schema.ultraservice.neustar.com/v01/\">Failed</NeustarNetworkStatus>\n"

//

+ " </ns1:getNeustarNetworkStatusResponse>\n"//

+ " </soap:Body>\n"//

+ "</soap:Envelope>";

@Rule

public final ExpectedException thrown = ExpectedException.none();

@Inject

Decoder decoder;

@Before

public void inject() {

ObjectGraph.create(new Module()).inject(this);

}

@Test

public void parsesConfiguredTypes() throws ParseException, IOException {

assertEquals(NetworkStatus.FAILED, decoder.decode(statusFailedResponse(), NetworkStatus.class));

assertEquals("Failed", decoder.decode(statusFailedResponse(), String.class));

}

@Test

public void niceErrorOnUnconfiguredType() throws ParseException, IOException {

thrown.expect(IllegalStateException.class);

thrown.expectMessage("type int not in configured handlers");

decoder.decode(statusFailedResponse(), int.class);

}

private Response statusFailedResponse() {

return Response

.create(200, "OK", Collections.<String, Collection<String>>emptyMap(), statusFailed, UTF_8);

}

@Test

public void nullBodyDecodesToNull() throws Exception {

Response

response =

Response

.create(204, "OK", Collections.<String, Collection<String>>emptyMap(), (byte[]) null);

assertNull(decoder.decode(response, String.class));

}

static enum NetworkStatus {

GOOD, FAILED;

}

@dagger.Module(injects = SAXDecoderTest.class)

static class Module {

@Provides

Decoder saxDecoder(Provider<NetworkStatusHandler> networkStatus) {

return SAXDecoder.builder() //

.registerContentHandler(NetworkStatus.class, networkStatus) //

.registerContentHandler(NetworkStatusStringHandler.class) //

.build();

}

}

static class NetworkStatusStringHandler extends DefaultHandler implements

SAXDecoder.ContentHandlerWithResult<String> {

private StringBuilder currentText = new StringBuilder();

private String status;

@Inject

NetworkStatusStringHandler() {

}

@Override

public String result() {

return status;

}

@Override

public void endElement(String uri, String name, String qName) {

if (qName.equals("NeustarNetworkStatus")) {

this.status = currentText.toString().trim();

}

currentText = new StringBuilder();

}

@Override

public void characters(char ch[], int start, int length) {

currentText.append(ch, start, length);

}

}

static class NetworkStatusHandler extends DefaultHandler implements

SAXDecoder.ContentHandlerWithResult<NetworkStatus> {

private StringBuilder currentText = new StringBuilder();

private NetworkStatus status;

@Inject

NetworkStatusHandler() {

}

@Override

public NetworkStatus result() {

return status;

}

@Override

public void endElement(String uri, String name, String qName) {

if (qName.equals("NeustarNetworkStatus")) {

this.status = NetworkStatus.valueOf(currentText.toString().trim().toUpperCase());

}

currentText = new StringBuilder();

}

@Override

public void characters(char ch[], int start, int length) {

currentText.append(ch, start, length);

}

}

}