redgraph 0.1.0 → 0.2.0

data/test/node_model_class_methods_test.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require "test_helper"
+
+class NodeModelClassMethodsTest < Minitest::Test
+  include TestHelpers
+
+  GRAPH = Redgraph::Graph.new("movies", url: $REDIS_URL)
+
+  def setup
+    @graph = GRAPH
+  end
+
+  def teardown
+    @graph.delete
+  end
+
+  # test classes
+  #
+
+  class Actor
+    include Redgraph::NodeModel
+    self.graph = GRAPH
+    attribute :name
+  end
+
+  # tests
+  #
+
+  def test_count
+    quick_add_node(label: 'actor', properties: {_type: Actor.name, name: "Al Pacino"})
+    quick_add_node(label: 'actor', properties: {_type: Actor.name, name: "John Travolta"})
+    assert_equal(2, Actor.count)
+    assert_equal(1, Actor.count(properties: {name: "Al Pacino"}))
+  end
+
+  def test_all
+    al = Actor.new(name: "Al Pacino").add_to_graph
+    john = Actor.new(name: "John Travolta").add_to_graph
+
+    items = Actor.all
+    assert_equal(2, items.size)
+    assert_includes(items, al)
+    assert_includes(items, john)
+
+    items = Actor.all(properties: {name: "Al Pacino"})
+    assert_equal(1, items.size)
+    assert_includes(items, al)
+  end
+
+  def test_find
+    al = quick_add_node(label: 'actor', properties: {name: "Al Pacino"})
+    item = Actor.find(al.id)
+
+    assert_equal(Actor, item.class)
+    assert_predicate(item, :persisted?)
+    assert_equal(al.id, item.id)
+    assert_equal("Al Pacino", item.name)
+  end
+
+  def test_find_bad_id
+    quick_add_node(label: 'actor', properties: {name: "Al Pacino"})
+    item = Actor.find("-1")
+    assert_nil(item)
+  end
+end

data/test/node_model_labels_test.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "test_helper"
+
+class NodeModelLabelsTest < Minitest::Test
+  include TestHelpers
+
+  GRAPH = Redgraph::Graph.new("movies", url: $REDIS_URL)
+
+  def setup
+    @graph = GRAPH
+  end
+
+  def teardown
+    @graph.delete
+  end
+
+  # test classes
+  #
+
+  class Actor
+    include Redgraph::NodeModel
+    self.graph = GRAPH
+    attribute :name
+  end
+
+  class Artist
+    include Redgraph::NodeModel
+    self.label = "_artist"
+  end
+
+  class Painter < Artist
+  end
+
+  class Pianist < Artist
+    self.label = "pianist"
+  end
+
+  # tests
+  #
+
+  def test_label
+    assert_equal("actor", Actor.label)
+  end
+
+  def test_custom_label
+    assert_equal("_artist", Artist.label)
+  end
+
+  def test_default_label_when_inherited
+    assert_equal("painter", Painter.label)
+  end
+
+  def test_custom_label_when_inherited
+    assert_equal("pianist", Pianist.label)
+  end
+end

data/test/node_model_test.rb

@@ -0,0 +1,126 @@
+# frozen_string_literal: true
+
+require "test_helper"
+
+class NodeModelTest < Minitest::Test
+  include TestHelpers
+
+  GRAPH = Redgraph::Graph.new("movies", url: $REDIS_URL)
+
+  def setup
+    @graph = GRAPH
+  end
+
+  def teardown
+    @graph.delete
+  end
+
+  # test classes
+  #
+
+  class Animal
+    include Redgraph::NodeModel
+    attribute :name
+    self.graph = "pippo"
+  end
+
+  class Dog < Animal
+  end
+
+  class Actor
+    include Redgraph::NodeModel
+    self.graph = GRAPH
+    attribute :name
+  end
+
+  class Film
+    include Redgraph::NodeModel
+    self.graph = GRAPH
+    attribute :name
+    attribute :year
+  end
+
+  # tests
+  #
+
+  def test_graph_accessor
+    assert_equal("pippo", Animal.graph)
+    assert_equal("pippo", Animal.new.graph)
+  end
+
+  def test_class_inheritance
+    assert_equal("pippo", Dog.graph)
+    assert_equal("dog", Dog.label)
+    assert_equal([:id, :name], Dog.attribute_names)
+  end
+
+  def test_attribute_names
+    assert_equal([:id, :name, :year], Film.attribute_names)
+
+    film = Film.new(name: "Star Wars", year: 1977)
+    assert_equal("Star Wars", film.name)
+    assert_equal(1977, film.year)
+  end
+
+  def test_attributes
+    film = Film.new(name: "Star Wars", year: 1977)
+    assert_equal({id: nil, name: "Star Wars", year: 1977}, film.attributes)
+  end
+
+  def test_add_to_graph
+    assert_equal(0, Film.count)
+    film = Film.new(name: "Star Wars", year: 1977)
+    item = film.add_to_graph
+    assert_predicate(item, :persisted?)
+
+    assert_equal(1, Film.count)
+  end
+
+  def test_merge_into_graph
+    film = Film.new(name: "Star Wars", year: 1977)
+    item = film.add_to_graph(allow_duplicates: false)
+    assert_predicate(item, :persisted?)
+
+    assert_equal(1, Film.count)
+
+    film = Film.new(name: "Star Wars", year: 1977)
+    item = film.add_to_graph(allow_duplicates: false)
+    assert_predicate(item, :persisted?)
+
+    assert_equal(1, Film.count)
+  end
+
+  def test_add_relation
+    film = Film.new(name: "Star Wars", year: 1977).add_to_graph
+    actor = Actor.new(name: "Harrison Ford").add_to_graph
+    edge = actor.add_relation(type: "ACTED_IN", node: film, properties: {role: "Han Solo"})
+
+    assert_predicate(edge, :persisted?)
+    assert_equal("ACTED_IN", edge.type)
+  end
+
+  def test_add_relation_with_duplicate_control
+    film = Film.new(name: "Star Wars", year: 1977).add_to_graph
+    actor = Actor.new(name: "Harrison Ford").add_to_graph
+
+    actor.add_relation(type: "ACTED_IN", node: film, properties: {role: "Han Solo"}, allow_duplicates: true)
+    assert_equal(1, @graph.count_edges)
+
+    actor.add_relation(type: "ACTED_IN", node: film, properties: {role: "Han Solo"}, allow_duplicates: true)
+    assert_equal(2, @graph.count_edges)
+
+    actor.add_relation(type: "ACTED_IN", node: film, properties: {role: "Han Solo"}, allow_duplicates: false)
+    assert_equal(2, @graph.count_edges)
+  end
+
+  def test_casting_query
+    Film.new(name: "Star Wars", year: 1977).add_to_graph
+    Actor.new(name: "Harrison Ford").add_to_graph
+
+    items = Film.query("MATCH (node) RETURN node ORDER BY node.name")
+    assert_equal(2, items.size)
+    assert_kind_of(Actor, items[0][0])
+    assert_kind_of(Film, items[1][0])
+  end
+
+end

data/test/test_helper.rb

@@ -1,5 +1,14 @@
 # frozen_string_literal: true
 
+if ENV['COVERAGE']
+  require 'simplecov'
+
+  SimpleCov.start do
+    enable_coverage :branch
+    add_filter %r{^/test/}
+  end
+end
+
 $LOAD_PATH.unshift File.expand_path("../lib", __dir__)
 require "redgraph"
 
@@ -10,3 +19,15 @@ unless $REDIS_URL = ENV['TEST_REDIS_URL']
   puts "To run the tests you need to define the TEST_REDIS_URL environment variable"
   exit(1)
 end
+
+module TestHelpers
+  private
+
+  def quick_add_node(label:, properties:)
+    @graph.add_node(Redgraph::Node.new(label: label, properties: properties))
+  end
+
+  def quick_add_edge(type:, src:, dest:, properties:)
+    @graph.add_edge(Redgraph::Edge.new(type: type, src: src, dest: dest, properties: properties))
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: redgraph
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Paolo Zaccagnini
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-04-10 00:00:00.000000000 Z
+date: 2021-04-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis
@@ -24,6 +24,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '4'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
 description:
 email:
 - hi@pzac.net
@@ -44,14 +58,24 @@ files:
 - lib/redgraph.rb
 - lib/redgraph/edge.rb
 - lib/redgraph/graph.rb
+- lib/redgraph/graph/edge_methods.rb
+- lib/redgraph/graph/node_methods.rb
 - lib/redgraph/node.rb
+- lib/redgraph/node_model.rb
+- lib/redgraph/node_model/class_methods.rb
 - lib/redgraph/query_response.rb
+- lib/redgraph/util.rb
 - lib/redgraph/version.rb
 - redgraph.gemspec
 - test/graph_connection_test.rb
+- test/graph_edge_methods_test.rb
 - test/graph_manipulation_test.rb
+- test/graph_node_methods_test.rb
 - test/graph_queries_test.rb
 - test/graph_test.rb
+- test/node_model_class_methods_test.rb
+- test/node_model_labels_test.rb
+- test/node_model_test.rb
 - test/redgraph_test.rb
 - test/test_helper.rb
 homepage: https://github.com/pzac/redgraph
@@ -60,7 +84,7 @@ licenses:
 metadata:
   homepage_uri: https://github.com/pzac/redgraph
   source_code_uri: https://github.com/pzac/redgraph
-  changelog_uri: https://github.com/pzac/redgraph/CHANGELOG.md
+  changelog_uri: https://github.com/pzac/redgraph/blob/master/CHANGELOG.md
 post_install_message:
 rdoc_options: []
 require_paths: