From 09d6595514029345c2bdf839768d008a2ed92c13 Mon Sep 17 00:00:00 2001
From: Kapil Khandelwal <kapilbk1996@gmail.com>
Date: Tue, 19 Apr 2022 05:05:55 +0530
Subject: [PATCH] Bael 5369 new (#12073)

* BAEL-5369: Checking Connection to MongoDB

* BAEL-5369:- Minor Fix Checking Connection to MongoDB
---
 .../com/baeldung/mongo/ConnectionCheck.java   |  8 ++---
 .../mongo/ConnectionCheckLiveTest.java        | 31 +++----------------
 2 files changed, 8 insertions(+), 31 deletions(-)

diff --git a/persistence-modules/java-mongodb-2/src/main/java/com/baeldung/mongo/ConnectionCheck.java b/persistence-modules/java-mongodb-2/src/main/java/com/baeldung/mongo/ConnectionCheck.java
index 240e54ab99..df17e27351 100644
--- a/persistence-modules/java-mongodb-2/src/main/java/com/baeldung/mongo/ConnectionCheck.java
+++ b/persistence-modules/java-mongodb-2/src/main/java/com/baeldung/mongo/ConnectionCheck.java
@@ -7,7 +7,7 @@ import com.mongodb.ServerAddress;
 
 public class ConnectionCheck {
 
-    public static void checkingConnection() {
+    public static MongoClient checkingConnection() {
 
         MongoClientOptions.Builder builder = MongoClientOptions.builder();
 
@@ -27,9 +27,10 @@ public class ConnectionCheck {
             System.out.println(db.getStats());
         } catch (Exception e) {
             System.out.println("MongoDB Server is Down");
-            mongoClient.close();
         }
 
+        return mongoClient;
+
     }
 
     public static void main(String[] args) {
@@ -42,5 +43,4 @@ public class ConnectionCheck {
 
     }
 
-}
-
+}
\ No newline at end of file
diff --git a/persistence-modules/java-mongodb-2/src/test/java/com/baeldung/mongo/ConnectionCheckLiveTest.java b/persistence-modules/java-mongodb-2/src/test/java/com/baeldung/mongo/ConnectionCheckLiveTest.java
index 69f356997c..32014a4473 100644
--- a/persistence-modules/java-mongodb-2/src/test/java/com/baeldung/mongo/ConnectionCheckLiveTest.java
+++ b/persistence-modules/java-mongodb-2/src/test/java/com/baeldung/mongo/ConnectionCheckLiveTest.java
@@ -3,45 +3,22 @@ package com.baeldung.mongo;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
 
-import java.io.IOException;
-
-import org.junit.BeforeClass;
 import org.junit.Test;
 
+import com.baeldung.ConnectionCheck;
 import com.mongodb.MongoClient;
-import com.mongodb.MongoClientOptions;
-import com.mongodb.ServerAddress;
 
 public class ConnectionCheckLiveTest {
 
-    private static MongoClient mongoClient;
-    private static MongoClientOptions.Builder builder;
-    private static ServerAddress ServerAddress;
-
-    @BeforeClass
-    public static void setup() throws IOException {
-        if (mongoClient == null) {
-
-            builder = MongoClientOptions.builder();
-            builder.connectionsPerHost(100);
-            builder.maxWaitTime(60000);
-            builder.connectTimeout(1500);
-            builder.socketTimeout(60000);
-            builder.socketKeepAlive(true);
-
-            ServerAddress = new ServerAddress("localhost", 27017);
-            mongoClient = new MongoClient(ServerAddress, builder.build());
-
-        }
-    }
+    ConnectionCheck ConnectionCheck = new ConnectionCheck();
 
     @Test
     public void givenMongoClient_whenConnectionCheck_thenCheckingForConnectionPoint() {
 
+        MongoClient mongoClient = ConnectionCheck.checkingConnection();
         String connectionPoint = mongoClient.getConnectPoint();
         assertNotNull(connectionPoint);
         assertFalse(connectionPoint.isEmpty());
 
     }
-}
-
+}
\ No newline at end of file