Fixes #82

AbsaOSS · Mar 10, 2023 · f01d273 · f01d273
1 parent bfd34ea
commit f01d273
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/spark-commons/src/main/scala/za/co/absa/spark/commons/OncePerSparkSession.scala b/spark-commons/src/main/scala/za/co/absa/spark/commons/OncePerSparkSession.scala
@@ -41,9 +41,9 @@ abstract class OncePerSparkSession() extends Serializable {
   }
 
   def register(implicit spark: SparkSession): Boolean = {
-    val created = OncePerSparkSession.registerMe(this, spark)
+    val registered = OncePerSparkSession.registerMe(this, spark)
 
-    if (created == this) true
+    if (registered == this) true
     else false
   }
 

diff --git a/spark-commons/src/test/scala/za/co/absa/spark/commons/OncePerSparkSessionTest.scala b/spark-commons/src/test/scala/za/co/absa/spark/commons/OncePerSparkSessionTest.scala
@@ -27,7 +27,7 @@ class OncePerSparkSessionTest extends AnyFunSuite with MockitoSugar with SparkTe
     var libraryBInitCounter = 0
 
     val anotherSpark: SparkSession =  mock[SparkSession]
-    class UDFLibraryA()(implicit sparkToRegisterTo: Option[SparkSession]) extends OncePerSparkSession()(sparkToRegisterTo) {
+    class UDFLibraryA()(implicit sparkToRegisterTo: Option[SparkSession]) extends OncePerSparkSession() {
       override def register(implicit spark: Option[SparkSession]): Unit = {
         libraryAInitCounter += 1
       }