From fa222f57195c710c59ac84898d9853279fe43cac Mon Sep 17 00:00:00 2001
From: azhwani <>
Date: Wed, 10 Jun 2020 18:58:21 +0100
Subject: [PATCH] updating jndi.properties

---
 .../jndi/datasource/SimpleJNDIUnitTest.java   | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 persistence-modules/spring-persistence-simple-2/src/test/java/com/baeldung/jndi/datasource/SimpleJNDIUnitTest.java

diff --git a/persistence-modules/spring-persistence-simple-2/src/test/java/com/baeldung/jndi/datasource/SimpleJNDIUnitTest.java b/persistence-modules/spring-persistence-simple-2/src/test/java/com/baeldung/jndi/datasource/SimpleJNDIUnitTest.java
new file mode 100644
index 0000000000..37f33b1192
--- /dev/null
+++ b/persistence-modules/spring-persistence-simple-2/src/test/java/com/baeldung/jndi/datasource/SimpleJNDIUnitTest.java
@@ -0,0 +1,39 @@
+package com.baeldung.jndi.datasource;
+
+import static org.junit.Assert.assertEquals;
+
+import javax.naming.Context;
+import javax.naming.InitialContext;
+import javax.sql.DataSource;
+
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+public class SimpleJNDIUnitTest {
+
+    private InitialContext initContext;
+
+    @BeforeEach
+    public void setup() throws Exception {
+        this.initContext = new InitialContext();
+    }
+
+    @Test
+    public void whenMockJndiDataSource_thenReturnJndiDataSource() throws Exception {
+        String dsString = "org.h2.Driver::::jdbc:jdbc:h2:mem:testdb::::sa";
+        Context envContext = (Context) this.initContext.lookup("java:/comp/env");
+        DataSource ds = (DataSource) envContext.lookup("datasource/ds");
+
+        assertEquals(dsString, ds.toString());
+    }
+
+    @AfterEach
+    public void tearDown() throws Exception {
+        if (this.initContext != null) {
+            this.initContext.close();
+            this.initContext = null;
+        }
+    }
+
+}