# frozen_string_literal: true

require "test_helper"

class GraphQueriesTest < Minitest::Test
  def setup
    @graph = Redgraph::Graph.new("movies", url: $REDIS_URL)
    @actor = Redgraph::Node.new(label: 'actor', properties: {name: "Al Pacino"})
    @graph.add_node(@actor)

    @other_actor = Redgraph::Node.new(label: 'actor', properties: {name: "John Travolta"})
    refute_nil(@actor.id)
  end

  def teardown
    @graph.delete
  end

  def test_find_node_by_id
    node = @graph.find_node_by_id(@actor.id)

    refute_nil(node)
    assert_equal("actor", node.label)
    assert_equal("Al Pacino", node.properties["name"])
    assert_equal(@actor.id, node.id)
  end

  def test_get_all_nodes
    @graph.add_node(@other_actor)

    actors = @graph.nodes

    assert_equal(2, actors.size)
    assert_includes(actors, @actor)
    assert_includes(actors, @other_actor)
  end

  def test_get_all_nodes_by_label
    @graph.add_node(@other_actor)
    film = Redgraph::Node.new(label: 'film', properties: {name: "Scarface"})
    @graph.add_node(film)

    actors = @graph.nodes(label: 'actor')
    assert_equal(2, actors.size)
    assert_includes(actors, @actor)
    assert_includes(actors, @other_actor)

    films = @graph.nodes(label: 'film')
    assert_equal(1, films.size)
    assert_includes(films, film)
  end

  def test_find_node_by_wrong_id
    node = @graph.find_node_by_id("-1")

    assert_nil(node)
  end
end