[CAMEL-17801] Add tests in camel-mongodb-starter

components-starter/camel-mongodb-starter/pom.xml

@@ -39,11 +39,87 @@
       <artifactId>camel-mongodb</artifactId>
       <version>${camel-version}</version>
     </dependency>
+    <!-- test infra -->
+    <dependency>
+      <groupId>org.apache.camel</groupId>
+      <artifactId>camel-test-infra-mongodb</artifactId>
+      <version>${camel-version}</version>
+      <type>test-jar</type>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.apache.commons</groupId>
+      <artifactId>commons-lang3</artifactId>
+      <version>${commons-lang3-version}</version>
+      <scope>test</scope>
+    </dependency>
     <!--START OF GENERATED CODE-->
     <dependency>
       <groupId>org.apache.camel.springboot</groupId>
       <artifactId>camel-core-starter</artifactId>
     </dependency>
     <!--END OF GENERATED CODE-->
   </dependencies>
+  <profiles>
+    <!-- activate integration test if the docker socket file is accessible -->
+    <profile>
+      <id>mongodb-integration-tests-docker-file</id>
+      <activation>
+        <file>
+          <exists>/var/run/docker.sock</exists>
+        </file>
+      </activation>
+      <build>
+        <plugins>
+          <plugin>
+            <groupId>org.apache.maven.plugins</groupId>
+            <artifactId>maven-failsafe-plugin</artifactId>
+            <configuration>
+              <systemPropertyVariables>
+                <visibleassertions.silence>true</visibleassertions.silence>
+              </systemPropertyVariables>
+            </configuration>
+            <executions>
+              <execution>
+                <goals>
+                  <goal>integration-test</goal>
+                  <goal>verify</goal>
+                </goals>
+              </execution>
+            </executions>
+          </plugin>
+        </plugins>
+      </build>
+    </profile>
+    <!-- activate integration test if the DOCKER_HOST env var is set -->
+    <profile>
+      <id>mongodb-integration-tests-docker-env</id>
+      <activation>
+        <property>
+          <name>env.DOCKER_HOST</name>
+        </property>
+      </activation>
+      <build>
+        <plugins>
+          <plugin>
+            <groupId>org.apache.maven.plugins</groupId>
+            <artifactId>maven-failsafe-plugin</artifactId>
+            <configuration>
+              <systemPropertyVariables>
+                <visibleassertions.silence>true</visibleassertions.silence>
+              </systemPropertyVariables>
+            </configuration>
+            <executions>
+              <execution>
+                <goals>
+                  <goal>integration-test</goal>
+                  <goal>verify</goal>
+                </goals>
+              </execution>
+            </executions>
+          </plugin>
+        </plugins>
+      </build>
+    </profile>
+  </profiles>
 </project>

components-starter/camel-mongodb-starter/src/test/java/org/apache/camel/component/mongodb/integration/AbstractMongoDbITSupport.java

@@ -0,0 +1,179 @@
+/*
+ * 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.mongodb.integration;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import org.apache.camel.CamelContext;
+import org.apache.camel.CamelExecutionException;
+import org.apache.camel.ProducerTemplate;
+import org.apache.camel.component.mock.MockEndpoint;
+import org.apache.camel.component.mongodb.CamelMongoDbException;
+import org.apache.camel.component.mongodb.MongoDbComponent;
+import org.apache.camel.test.infra.mongodb.services.MongoDBService;
+import org.apache.camel.test.infra.mongodb.services.MongoDBServiceFactory;
+import org.apache.camel.util.IOHelper;
+import org.apache.camel.util.ObjectHelper;
+
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.extension.RegisterExtension;
+
+import org.bson.Document;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import com.mongodb.BasicDBObject;
+import com.mongodb.client.MongoClient;
+import com.mongodb.client.MongoClients;
+import com.mongodb.client.MongoCollection;
+import com.mongodb.client.MongoDatabase;
+
+import java.util.Formatter;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+public abstract class AbstractMongoDbITSupport {
+
+	protected static final String SCHEME = "mongodb";
+	protected static final String USER = "test-user";
+	protected static final String PASSWORD = "test-pwd";
+	@RegisterExtension
+	public static MongoDBService service = MongoDBServiceFactory.createService();
+	protected static MongoClient mongo;
+	protected static MongoDatabase db;
+	protected static MongoCollection<Document> testCollection;
+	protected static MongoCollection<Document> dynamicCollection;
+
+	protected static String dbName = "test";
+	protected static String testCollectionName;
+	protected static String dynamicCollectionName;
+	protected ProducerTemplate template;
+	@Autowired
+	protected CamelContext context;
+
+	@BeforeAll
+	public static void beforeAll() {
+		mongo = MongoClients.create(service.getReplicaSetUrl());
+		db = mongo.getDatabase(dbName);
+	}
+
+	@BeforeEach
+	public void beforeEach() {
+		// Refresh the test collection - drop it and recreate it. We don't do
+		// this for the database because MongoDB would create large
+		// store files each time
+		testCollectionName = "camelTest";
+		testCollection = db.getCollection(testCollectionName, Document.class);
+		testCollection.drop();
+		testCollection = db.getCollection(testCollectionName, Document.class);
+
+		dynamicCollectionName = testCollectionName.concat("Dynamic");
+		dynamicCollection = db.getCollection(dynamicCollectionName, Document.class);
+		dynamicCollection.drop();
+		dynamicCollection = db.getCollection(dynamicCollectionName, Document.class);
+
+		template = context.createProducerTemplate();
+
+		context.getPropertiesComponent().setLocation("classpath:mongodb.test.properties");
+	}
+
+	@AfterEach
+	public void tearDown() throws Exception {
+		testCollection.drop();
+		dynamicCollection.drop();
+	}
+
+	/**
+	 * Useful to simulate the presence of an authenticated user with name {@value #USER} and password {@value #PASSWORD}
+	 */
+	protected void createAuthorizationUser() {
+		createAuthorizationUser("admin", USER, PASSWORD);
+	}
+
+	protected void createAuthorizationUser(String database, String user, String password) {
+		MongoDatabase adminDb = mongo.getDatabase("admin");
+		MongoCollection<Document> usersCollection = adminDb.getCollection("system.users");
+		if (usersCollection.countDocuments(new Document("user", user)) == 0) {
+			MongoDatabase db = mongo.getDatabase(database);
+			Map<String, Object> commandArguments = new LinkedHashMap<>();
+			commandArguments.put("createUser", user);
+			commandArguments.put("pwd", password);
+			String[] roles = {"readWrite"};
+			commandArguments.put("roles", roles);
+			BasicDBObject command = new BasicDBObject(commandArguments);
+			db.runCommand(command);
+		}
+	}
+
+	protected void pumpDataIntoTestCollection() {
+		// there should be 100 of each
+		String[] scientists
+				= {"Einstein", "Darwin", "Copernicus", "Pasteur", "Curie", "Faraday", "Newton", "Bohr", "Galilei", "Maxwell"};
+		for (int i = 1; i <= 1000; i++) {
+			int index = i % scientists.length;
+			Formatter f = new Formatter();
+			String doc
+					= f.format("{\"_id\":\"%d\", \"scientist\":\"%s\", \"fixedField\": \"fixedValue\"}", i, scientists[index])
+					.toString();
+			IOHelper.close(f);
+			testCollection.insertOne(Document.parse(doc));
+		}
+		assertEquals(1000L, testCollection.countDocuments(), "Data pumping of 1000 entries did not complete entirely");
+	}
+
+	protected CamelMongoDbException extractAndAssertCamelMongoDbException(Object result, String message) {
+		assertTrue(result instanceof Throwable, "Result is not an Exception");
+		assertTrue(result instanceof CamelExecutionException, "Result is not an CamelExecutionException");
+		Throwable exc = ((CamelExecutionException) result).getCause();
+		assertTrue(exc instanceof CamelMongoDbException, "Result is not an CamelMongoDbException");
+		CamelMongoDbException camelExc = ObjectHelper.cast(CamelMongoDbException.class, exc);
+		if (message != null) {
+			assertTrue(camelExc.getMessage().contains(message), "CamelMongoDbException doesn't contain desired message string");
+		}
+		return camelExc;
+	}
+
+	protected MockEndpoint getMockEndpoint(String endpoint) {
+		return context.getEndpoint(endpoint, MockEndpoint.class);
+	}
+
+	@Configuration
+	public class MongoConfiguration {
+
+		@Bean
+		public MongoClient mongoClient() {
+			return mongo;
+		}
+
+		@Bean
+		public MongoDbComponent mongoDbComponent() {
+			MongoDbComponent component = new MongoDbComponent();
+			component.setMongoConnection(mongo);
+
+			return component;
+		}
+
+		@Bean
+		public void propertyLocation() {
+			context.getPropertiesComponent().setLocation("classpath:mongodb.test.properties");
+		}
+	}
+}