DynamicLoadbalanceEndpoint loadbalanceEndpoint = new DynamicLoadbalanceEndpoint();
DynamicLoadbalanceEndpoint loadbalanceEndpoint = new DynamicLoadbalanceEndpoint();
/** * Test initialization of DynamicLoadBalanceEndpoint * * @throws AxisFault on an issue initializing DynamicLoadBalanceEndpoint */ @Test public void testInit() throws AxisFault { DynamicLoadbalanceEndpoint dynamicLoadbalanceEndpoint = new DynamicLoadbalanceEndpoint(); dynamicLoadbalanceEndpoint.init(getMockedSynapseEnvironment()); }
/** * Test sending a message to a load balance EP * * @throws AxisFault on an issue sending message to DynamicLoadBalanceEndpoint */ @Test public void testSend() throws AxisFault { DynamicLoadbalanceEndpoint dynamicLoadbalanceEndpoint = new DynamicLoadbalanceEndpoint(); SynapseEnvironment synapseEnvironment = getMockedSynapseEnvironment(); dynamicLoadbalanceEndpoint.init(synapseEnvironment); MessageContext messageContext = createMessageContext(); //Mock a LoadBalanceMembershipHandler and set LoadBalanceMembershipHandler loadBalanceMembershipHandler = PowerMockito. mock(LoadBalanceMembershipHandler.class); Member member1 = new Member("localhost", 9000); Mockito.when(loadBalanceMembershipHandler. getNextApplicationMember(any(AlgorithmContext.class))).thenReturn(member1); dynamicLoadbalanceEndpoint.setLoadBalanceMembershipHandler(loadBalanceMembershipHandler); //set mocked SynapseEnvironment to message context ((Axis2MessageContext)messageContext).getAxis2MessageContext(). getConfigurationContext().getAxisConfiguration(). addParameter(SynapseConstants.SYNAPSE_ENV, synapseEnvironment); messageContext.setEnvironment(synapseEnvironment); //send message dynamicLoadbalanceEndpoint.send(messageContext); Assert.assertNotNull("SynapseConstants.LAST_ENDPOINT should return a not null value", messageContext.getProperty(SynapseConstants.LAST_ENDPOINT)); }