fix datajack properties for mongo provider

sbtqa · Oct 1, 2018 · 0804d0e · 0804d0e
1 parent bc90e01
commit 0804d0e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/ru/sbtqa/tag/pagefactory/support/data/DataFactory.java b/src/main/java/ru/sbtqa/tag/pagefactory/support/data/DataFactory.java
@@ -1,5 +1,6 @@
 package ru.sbtqa.tag.pagefactory.support.data;
 
+import com.mongodb.DB;
 import com.mongodb.MongoClient;
 import com.mongodb.MongoClientURI;
 import org.slf4j.Logger;
@@ -45,9 +46,9 @@ public static TestDataProvider getDataProvider() throws DataException {
                     );
                     break;
                 case "mongo":
-                    testDataProvider = new MongoDataProvider(
-                            new MongoClient(new MongoClientURI(Props.get("data.uri"))).getDB("data.db"),
-                            Props.get("data.initial.collection")
+                    MongoClient mongoclient = new MongoClient(new MongoClientURI(Props.get("data.uri")));
+                    DB db = mongoclient.getDB(Props.get("data.db"));
+                    testDataProvider = new MongoDataProvider(db, Props.get("data.initial.collection")
                     );
                     break;
                 case "stash":