compile

cognitedata · Oct 31, 2023 · 92bc1e6 · 92bc1e6
1 parent a281008
commit 92bc1e6
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/src/main/scala/cognite/spark/v1/DefaultSource.scala b/src/main/scala/cognite/spark/v1/DefaultSource.scala
@@ -346,7 +346,14 @@ object DefaultSource {
       }
       clientId <- parameters.get("clientId")
       clientSecret <- parameters.get("clientSecret")
-      clientCredentials = OAuth2.ClientCredentials(tokenUri, clientId, clientSecret, scopes, audience)
+      project <- parameters.get("project")
+      clientCredentials = OAuth2.ClientCredentials(
+        tokenUri,
+        clientId,
+        clientSecret,
+        scopes,
+        project,
+        audience)
     } yield CdfSparkAuth.OAuth2ClientCredentials(clientCredentials)
 
     val session = for {

diff --git a/src/test/scala/cognite/spark/v1/DefaultSourceTest.scala b/src/test/scala/cognite/spark/v1/DefaultSourceTest.scala
@@ -75,6 +75,7 @@ class DefaultSourceTest extends WordSpec with Matchers {
               "value-ClientId",
               "value-ClientSecret",
               List("value-Scopes"),
+              "value-Project",
               Some("value-Audience")))
         )
       }

diff --git a/src/test/scala/cognite/spark/v1/SparkTest.scala b/src/test/scala/cognite/spark/v1/SparkTest.scala
@@ -58,6 +58,7 @@ trait SparkTest {
     clientSecret = OIDCWrite.clientSecret,
     scopes = List(OIDCWrite.scopes),
     audience = Some(OIDCWrite.audience),
+    cdfProjectName = OIDCWrite.project,
   )
   implicit val sttpBackend: SttpBackend[IO, Any] = CdpConnector.retryingSttpBackend(15, 30)
 
@@ -120,7 +121,8 @@ trait SparkTest {
     tokenUri = uri"$readTokenUri",
     clientId = readClientId,
     clientSecret = readClientSecret,
-    scopes = List("https://api.cognitedata.com/.default")
+    scopes = List("https://api.cognitedata.com/.default"),
+    cdfProjectName = readProject
   )
 
   def dataFrameReaderUsingOidc: DataFrameReader =
@@ -159,7 +161,8 @@ trait SparkTest {
       tokenUri = bluefieldTokenUri,
       clientId = bluefieldClientId,
       clientSecret = bluefieldClientSecret,
-      scopes = List("https://bluefield.cognitedata.com/.default")
+      scopes = List("https://bluefield.cognitedata.com/.default"),
+      cdfProjectName = "extractor-bluefield-testing"
     )
 
     val authProvider =