MonitoringServiceFullTest.java

/*
 * #%L
 * Service Activity Monitoring :: Server
 * %%
 * Copyright (C) 2011 - 2012 Talend 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.
 * #L%
 */
package org.talend.esb.sam.soap.service;

import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import javax.activation.DataHandler;
import javax.annotation.Resource;

import junit.framework.Assert;

import org.apache.cxf.endpoint.Client;
import org.apache.cxf.frontend.ClientProxy;
import org.apache.cxf.transport.http.HTTPConduit;
import org.apache.cxf.transports.http.configuration.HTTPClientPolicy;
import org.junit.Test;
import org.junit.After;
import org.junit.runner.RunWith;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.AbstractTransactionalJUnit4SpringContextTests;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.talend.esb.sam._2011._03.common.CustomInfoType;
import org.talend.esb.sam._2011._03.common.EventEnumType;
import org.talend.esb.sam._2011._03.common.EventType;
import org.talend.esb.sam._2011._03.common.MessageInfoType;
import org.talend.esb.sam.common.event.Event;
import org.talend.esb.sam.common.event.EventTypeEnum;
import org.talend.esb.sam.common.event.persistence.EventRepository;
import org.talend.esb.sam.monitoringservice.v1.MonitoringService;
import org.talend.esb.sam.monitoringservice.v1.PutEventsFault;

/**
 * Tests the monitoring service using webservice calls
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"/fulltest-server.xml", "/server-config.xml", "/fulltest-client.xml"})
public class MonitoringServiceFullTest extends AbstractTransactionalJUnit4SpringContextTests {
    @Resource(name = "monitoringServiceV1Client")
    MonitoringService monitoringService;
    
    @Resource
    private EventRepository eventRepository;
    
//    @Before
//    public void setUp() throws Exception {
//        executeSqlScript("create.sql", true);
//    }
    
    @Test
    public void testSendEvents() throws PutEventsFault, MalformedURLException, URISyntaxException {
        Client client = ClientProxy.getClient(monitoringService);
        HTTPConduit conduit = (HTTPConduit)client.getConduit();
        HTTPClientPolicy clientConfig = new HTTPClientPolicy();
        clientConfig.setReceiveTimeout(100000);
        conduit.setClient(clientConfig);

        simpleJdbcTemplate.update("delete from EVENTS");

        List<EventType> events = new ArrayList<EventType>();
        EventType eventType = new EventType();
        eventType.setEventType(EventEnumType.REQ_OUT);
        URL messageContentFile = this.getClass().getResource("/testmessage.xml").toURI().toURL();
        eventType.setContent(new DataHandler(messageContentFile ));
        
        CustomInfoType ciType = new CustomInfoType();
        CustomInfoType.Item prop1 = new CustomInfoType.Item();
        prop1.setKey("mykey1");
        prop1.setValue("myValue1");
        ciType.getItem().add(prop1);
        CustomInfoType.Item prop2 = new CustomInfoType.Item();
        prop2.setKey("mykey2");
        prop2.setValue("myValue2");
        ciType.getItem().add(prop2);
        eventType.setCustomInfo(ciType);
        
        MessageInfoType mit = new MessageInfoType();
        mit.setFlowId("uuid");
        eventType.setMessageInfo(mit);
        
        events.add(eventType);
        String result = monitoringService.putEvents(events);
        Assert.assertEquals("success", result);
        

        long id = simpleJdbcTemplate.queryForLong("select id from EVENTS");
        Event readEvent = eventRepository.readEvent(id);
        Assert.assertEquals(EventTypeEnum.REQ_OUT, readEvent.getEventType());
        Map<String, String> customInfo = readEvent.getCustomInfo();
        Assert.assertEquals("myValue1", customInfo.get("mykey1"));
        Assert.assertEquals("myValue2", customInfo.get("mykey2"));
    }
    
    @After
    public void tearDown() {
        executeSqlScript("drop.sql", true);
    }
}