| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 org.apache.camel.component.atomix.client.map; |
| import java.util.Collections; |
| import java.util.Map; |
| import java.util.UUID; |
| |
| import io.atomix.collections.DistributedMap; |
| import org.apache.camel.Component; |
| import org.apache.camel.EndpointInject; |
| import org.apache.camel.FluentProducerTemplate; |
| import org.apache.camel.Message; |
| import org.apache.camel.RoutesBuilder; |
| import org.apache.camel.builder.RouteBuilder; |
| import org.apache.camel.component.atomix.client.AtomixClientConstants; |
| import org.apache.camel.component.atomix.client.AtomixClientTestSupport; |
| import org.junit.After; |
| import org.junit.Test; |
| |
| public class AtomixMapNodesProducerTest extends AtomixClientTestSupport { |
| |
| private static final String MAP_NAME = UUID.randomUUID().toString(); |
| private DistributedMap<Object, Object> map; |
| |
| @EndpointInject("direct:start") |
| private FluentProducerTemplate fluent; |
| |
| // ************************************ |
| // Setup |
| // ************************************ |
| |
| @Override |
| protected Map<String, Component> createComponents() { |
| return Collections.emptyMap(); |
| } |
| |
| @Override |
| protected void doPreSetup() throws Exception { |
| super.doPreSetup(); |
| |
| map = getClient().getMap(MAP_NAME).join(); |
| } |
| |
| @Override |
| @After |
| public void tearDown() throws Exception { |
| map.close(); |
| |
| super.tearDown(); |
| } |
| |
| // ************************************ |
| // Test |
| // ************************************ |
| |
| @Test |
| public void testPut() throws Exception { |
| final String key = context().getUuidGenerator().generateUuid(); |
| final String val = context().getUuidGenerator().generateUuid(); |
| |
| Message result; |
| |
| result = fluent.clearAll() |
| .withHeader(AtomixClientConstants.RESOURCE_ACTION, AtomixMap.Action.PUT) |
| .withHeader(AtomixClientConstants.RESOURCE_KEY, key) |
| .withBody(val) |
| .request(Message.class); |
| |
| assertFalse(result.getHeader(AtomixClientConstants.RESOURCE_ACTION_HAS_RESULT, Boolean.class)); |
| assertEquals(val, result.getBody()); |
| assertEquals(val, map.get(key).join()); |
| } |
| |
| // ************************************ |
| // Routes |
| // ************************************ |
| |
| @Override |
| protected RoutesBuilder createRouteBuilder() throws Exception { |
| return new RouteBuilder() { |
| public void configure() { |
| from("direct:start") |
| .toF("atomix-map:%s?nodes=%s:%d", MAP_NAME, replicaAddress.host(), replicaAddress.port()); |
| } |
| }; |
| } |
| } |