From eda84c24b93754aaadfdac4195d921392ce763a7 Mon Sep 17 00:00:00 2001
From: anuragkumawat <anuragkumawat7@gmail.com>
Date: Sun, 10 Jul 2022 13:40:05 +0530
Subject: [PATCH]  JAVA-12627 Fix DatasetUnitTest in apache-spark module

---
 .../CustomerToDataFrameConverterAppUnitTest.java           | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/apache-spark/src/test/java/com/baeldung/dataframes/CustomerToDataFrameConverterAppUnitTest.java b/apache-spark/src/test/java/com/baeldung/dataframes/CustomerToDataFrameConverterAppUnitTest.java
index 06c8f66bcd..c16ca34fdc 100644
--- a/apache-spark/src/test/java/com/baeldung/dataframes/CustomerToDataFrameConverterAppUnitTest.java
+++ b/apache-spark/src/test/java/com/baeldung/dataframes/CustomerToDataFrameConverterAppUnitTest.java
@@ -8,6 +8,8 @@ import java.util.List;
 
 import org.apache.spark.sql.Dataset;
 import org.apache.spark.sql.Row;
+import org.apache.spark.sql.SparkSession;
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.Test;
 
 class CustomerToDataFrameConverterAppUnitTest {
@@ -59,4 +61,9 @@ class CustomerToDataFrameConverterAppUnitTest {
         assertEquals( "Male", row2.getAs("gender"));
         assertEquals( 1200, (int)row2.getAs("transaction_amount"));
     }
+    
+    @AfterEach
+    public void cleanup() {
+    	SparkSession.getActiveSession().get().close();
+    }
 }