defmodule NS do
  use RDF.Vocabulary.Namespace

  defvocab MF,
           base_iri: "http://www.w3.org/2001/sw/DataAccess/tests/test-manifest#",
           terms: [], strict: false

  defvocab RDFT,
           base_iri: "http://www.w3.org/ns/rdftest#",
           terms: [], strict: false
end

alias NS.{MF, RDFT}
alias RDF.NS.RDFS
alias RDF.Query.BGP

test_graph = RDF.Turtle.read_file!("test/data/TURTLE-TESTS/manifest.ttl", base: "http://www.w3.org/2013/TurtleTests/")

all_query = %BGP{triple_patterns: [{:s, :p, :o}]}
Benchee.run(%{
  "take 1 from BGP.Simple" => fn -> BGP.Simple.stream(all_query, test_graph) |> Enum.take(1) end,
  "take 1 from BGP.Stream" => fn -> BGP.Stream.stream(all_query, test_graph) |> Enum.take(1) end,
})


# rdft:approval rdft:Approved - count: 287
approved_query = %BGP{triple_patterns: [
  {:test_case, RDFT.approval, RDF.iri(RDFT.Approved)},
  {:test_case, MF.name, :name},
  {:test_case, RDFS.comment, :comment},
]}
# rdft:approval rdft:Proposed - count: 4
proposed_query = %BGP{triple_patterns: [
  {:test_case, RDFT.approval, RDF.iri(RDFT.Proposed)},
  {:test_case, MF.name, :name},
  {:test_case, RDFS.comment, :comment},
]}
Benchee.run(%{
  "APPROVED from BGP.Simple" => fn -> BGP.Simple.execute(approved_query, test_graph) end,
  "PROPOSED from BGP.Simple" => fn -> BGP.Simple.execute(proposed_query, test_graph) end,

  "APPROVED from BGP.Stream (consumed)"   => fn -> BGP.Stream.execute(approved_query, test_graph) end,
  "PROPOSED from BGP.Stream (consumed)"   => fn -> BGP.Stream.execute(proposed_query, test_graph) end,
  "APPROVED from BGP.Stream (unconsumed)" => fn -> BGP.Stream.stream(approved_query, test_graph) end,
  "PROPOSED from BGP.Stream (unconsumed)" => fn -> BGP.Stream.stream(proposed_query, test_graph) end,
  "APPROVED from BGP.Stream (1 consumed)" => fn -> BGP.Stream.stream(approved_query, test_graph) |> Enum.take(1) end
})