TriplestoreServiceLiveSpec.scala

/*
 * Copyright © 2021 - 2023 Swiss National Data and Service Center for the Humanities and/or DaSCH Service Platform contributors.
 * SPDX-License-Identifier: Apache-2.0
 */

package org.knora.webapi.store.triplestore

import akka.testkit.ImplicitSender

import scala.concurrent.duration._

import org.knora.webapi.CoreSpec
import org.knora.webapi.messages.store.triplestoremessages._
import org.knora.webapi.routing.UnsafeZioRun
import org.knora.webapi.store.triplestore.api.TriplestoreService
import org.knora.webapi.store.triplestore.api.TriplestoreService.Queries.Select
import org.knora.webapi.store.triplestore.api.TriplestoreService.Queries.Update

class TriplestoreServiceLiveSpec extends CoreSpec with ImplicitSender {

  override implicit val timeout: FiniteDuration = 30.seconds

  override lazy val rdfDataObjects: List[RdfDataObject] = List(
    RdfDataObject("test_data/project_data/anything-data.ttl", "http://www.knora.org/data/0001/anything")
  )

  val countTriplesQuery: String =
    """
        SELECT (COUNT(*) AS ?no)
        WHERE
            {
                ?s ?p ?o .
            }
        """

  val namedGraphQuery: String =
    """
        SELECT ?namedGraph ?s ?p ?o ?lang
        WHERE {
                {
              GRAPH ?namedGraph {
                BIND(IRI("http://www.knora.org/ontology/0001/anything#Thing") as ?s)
                ?s ?p ?obj
                BIND(str(?obj) as ?o)
                BIND(lang(?obj) as ?lang)
              }
            }
        }
        """.stripMargin

  val insertQuery: String =
    """
        prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#>
        prefix sub: <http://subotic.org/#>

        INSERT DATA
        {
            GRAPH <http://subotic.org/graph>
            {
                <http://ivan> sub:tries "something" ;
                              sub:hopes "success" ;
                              rdf:type sub:Me .
            }
        }
        """

  val graphDataContent: String =
    """
        prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#>
        prefix jedi: <http://jedi.org/#>

        <http://luke> jedi:tries "force for the first time" ;
                      jedi:hopes "to power the lightsaber" ;
                      rdf:type jedi:Skywalker .
        """

  val checkInsertQuery: String =
    """
        prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#>
        prefix sub: <http://subotic.org/#>

        SELECT *
        WHERE {
            GRAPH <http://subotic.org/graph>
            {
                ?s rdf:type sub:Me .
                ?s ?p ?o .
            }
        }
        """

  val revertInsertQuery: String =
    """
        prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#>
        prefix sub: <http://subotic.org/#>

        WITH <http://subotic.org/graph>
        DELETE { ?s ?p ?o }
        WHERE
        {
            ?s rdf:type sub:Me .
            ?s ?p ?o .
        }
        """

  val searchURI: String = "<http://jena.apache.org/text#query>"

  val textSearchQueryFusekiValueHasString: String =
    s"""
        PREFIX knora-base: <http://www.knora.org/ontology/knora-base#>

        SELECT DISTINCT *
        WHERE {
            ?iri <http://jena.apache.org/text#query> 'test' .
            ?iri knora-base:valueHasString ?literal .
        }
    """

  val textSearchQueryFusekiDRFLabel: String =
    s"""
        PREFIX rdfs: <http://www.w3.org/2000/01/rdf-schema#>

        SELECT DISTINCT *
        WHERE {
            ?iri <http://jena.apache.org/text#query> 'Papa' .
            ?iri rdfs:label ?literal .
        }
    """

  var afterLoadCount: Int         = -1
  var afterChangeCount: Int       = -1
  var afterChangeRevertCount: Int = -1

  "The TriplestoreService" should {

    "reset the data after receiving a 'ResetTriplestoreContent' request" in {
      UnsafeZioRun.runOrThrow(
        TriplestoreService
          .resetTripleStoreContent(rdfDataObjects)
          .timeout(java.time.Duration.ofMinutes(5))
      )

      val msg = UnsafeZioRun.runOrThrow(TriplestoreService.query(Select(countTriplesQuery)))
      afterLoadCount = msg.results.bindings.head.rowMap("no").toInt
      (afterLoadCount > 0) should ===(true)
    }

    "provide data receiving a Named Graph request" in {
      val actual = UnsafeZioRun.runOrThrow(TriplestoreService.query(Select(namedGraphQuery)))
      actual.results.bindings.nonEmpty should ===(true)
    }

    "execute an update" in {
      val countTriplesBefore = UnsafeZioRun.runOrThrow(
        TriplestoreService
          .query(Select(countTriplesQuery))
          .map(_.results.bindings.head.rowMap("no").toInt)
      )
      countTriplesBefore should ===(afterLoadCount)

      UnsafeZioRun.runOrThrow(TriplestoreService.query(Update(insertQuery)))

      val checkInsertActual = UnsafeZioRun.runOrThrow(
        TriplestoreService
          .query(Select(checkInsertQuery))
          .map(_.results.bindings.size)
      )
      checkInsertActual should ===(3)

      afterChangeCount = UnsafeZioRun.runOrThrow(
        TriplestoreService
          .query(Select(countTriplesQuery))
          .map(_.results.bindings.head.rowMap("no").toInt)
      )
      (afterChangeCount - afterLoadCount) should ===(3)
    }

    "revert back " in {
      val countTriplesBefore = UnsafeZioRun.runOrThrow(
        TriplestoreService
          .query(Select(countTriplesQuery))
          .map(_.results.bindings.head.rowMap("no").toInt)
      )
      countTriplesBefore should ===(afterChangeCount)

      UnsafeZioRun.runOrThrow(TriplestoreService.query(Update(revertInsertQuery)))

      val countTriplesQueryActual = UnsafeZioRun.runOrThrow(
        TriplestoreService
          .query(Select(countTriplesQuery))
          .map(_.results.bindings.head.rowMap("no").toInt)
      )
      countTriplesQueryActual should ===(afterLoadCount)

      val checkInsertActual = UnsafeZioRun.runOrThrow(
        TriplestoreService
          .query(Select(checkInsertQuery))
          .map(_.results.bindings.size)
      )
      checkInsertActual should ===(0)
    }

    "execute the search with the lucene index for 'knora-base:valueHasString' properties" in {
      within(1000.millis) {
        val actual = UnsafeZioRun.runOrThrow(
          TriplestoreService
            .query(Select(textSearchQueryFusekiValueHasString))
            .map(_.results.bindings.size)
        )
        actual should ===(3)
      }
    }

    "execute the search with the lucene index for 'rdfs:label' properties" in {
      within(1000.millis) {
        val actual = UnsafeZioRun.runOrThrow(
          TriplestoreService
            .query(Select(textSearchQueryFusekiDRFLabel))
            .map(_.results.bindings.size)
        )
        actual should ===(1)
      }
    }
  }
}