RegistersPollingConsumerTest.java

/*
 *  Copyright 2012 Steven Swor.
 *
 *  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 cameljamod;

import java.net.URI;
import java.util.HashMap;
import java.util.Map;
import net.wimpi.modbus.msg.ReadMultipleRegistersRequest;
import net.wimpi.modbus.msg.ReadMultipleRegistersResponse;
import net.wimpi.modbus.procimg.Register;
import net.wimpi.modbus.procimg.SimpleRegister;
import org.apache.camel.CamelContext;
import org.apache.camel.Processor;
import org.apache.camel.impl.DefaultCamelContext;
import org.junit.AfterClass;
import static org.junit.Assert.*;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 * Unit tests for {@link DiscreteInputsPollingConsumer}.
 * @author Steven Swor
 */
public class RegistersPollingConsumerTest {
    
    /**
     * The test instance.
     */
    private static RegistersPollingConsumer instance = null;
    
    /**
     * Sets up the test instance.
     * @throws Exception if the test instance cannot be set up
     */
    @BeforeClass
    public static void setUpClass() throws Exception {
        CamelContext context = new DefaultCamelContext();
        JamodComponent component = new JamodComponent();
        component.setCamelContext(context);
        URI modbusUri = new URI("tcp://localhost:1024/registers/2");
        Map<String, Object> parameters = new HashMap<String, Object>();
        parameters.put("count", Integer.valueOf(1));
        JamodEndpoint endpoint = new JamodEndpoint(component, modbusUri, parameters);
        endpoint.setCamelContext(context);
        Processor processor = new NoopProcessor();
        instance = new RegistersPollingConsumer(endpoint, processor);
        instance.setCount(1);
        instance.setReferenceAddress(2);
        
    }
    
    /**
     * Destroys the test instance.
     * @throws Exception 
     */
    @AfterClass
    public static void tearDownClass() throws Exception {
        instance = null;
    }

    /**
     * Tests {@link DiscreteInputsPollingConsumer#createRequest()}.
     */
    @Test
    public void testCreateRequest() throws Exception {
        ReadMultipleRegistersRequest request = instance.createRequest();
        assertNotNull(request);
        assertEquals(1, request.getWordCount());
        assertEquals(2, request.getReference());
    }

    /**
     * Tests {@link DiscreteInputsPollingConsumer#getBodyFromResponse(ReadRegistersResponse)}.
     */
    @Test
    public void testGetBodyFromResponse() {
        Register[] registers = new Register[]{new SimpleRegister()};
        ReadMultipleRegistersResponse response = new ReadMultipleRegistersResponse(registers);
        assertArrayEquals(registers, instance.getBodyFromResponse(response));
    }
}