sohm 0.0.1

data/test/json.rb

@@ -0,0 +1,62 @@
+require_relative 'helper'
+require_relative "../lib/ohm/json"
+
+class Venue < Ohm::Model
+  attribute :name
+  list :programmers, :Programmer
+end
+
+class Programmer < Ohm::Model
+  attribute :language
+
+  index :language
+
+  def to_hash
+    super.merge(language: language)
+  end
+end
+
+test "exports model.to_hash to json" do
+  assert_equal Hash.new, JSON.parse(Venue.new.to_json)
+
+  venue = Venue.create(name: "foo")
+  json  = JSON.parse(venue.to_json)
+  assert_equal venue.id, json["id"]
+  assert_equal nil, json["name"]
+
+  programmer = Programmer.create(language: "Ruby")
+  json = JSON.parse(programmer.to_json)
+
+  assert_equal programmer.id, json["id"]
+  assert_equal programmer.language, json["language"]
+end
+
+test "exports a set to json" do
+  Programmer.create(language: "Ruby")
+  Programmer.create(language: "Python")
+
+  expected = [{ id: "1", language: "Ruby" }, { id: "2", language: "Python"}].to_json
+
+  assert_equal expected, Programmer.all.to_json
+end
+
+test "exports a multiset to json" do
+  Programmer.create(language: "Ruby")
+  Programmer.create(language: "Python")
+
+  expected = [{ id: "1", language: "Ruby" }, { id: "2", language: "Python"}].to_json
+  result   = Programmer.find(language: "Ruby").union(language: "Python").to_json
+
+  assert_equal expected, result
+end
+
+test "exports a list to json" do
+  venue = Venue.create(name: "Foo")
+
+  venue.programmers.push(Programmer.create(language: "Ruby"))
+  venue.programmers.push(Programmer.create(language: "Python"))
+
+  expected = [{ id: "1", language: "Ruby" }, { id: "2", language: "Python"}].to_json
+
+  assert_equal expected, venue.programmers.to_json
+end

data/test/list.rb

@@ -0,0 +1,83 @@
+require_relative "helper"
+
+class Post < Ohm::Model
+  list :comments, :Comment
+end
+
+class Comment < Ohm::Model
+end
+
+setup do
+  post = Post.create
+
+  post.comments.push(c1 = Comment.create)
+  post.comments.push(c2 = Comment.create)
+  post.comments.push(c3 = Comment.create)
+
+  [post, c1, c2, c3]
+end
+
+test "include?" do |p, c1, c2, c3|
+  assert p.comments.include?(c1)
+  assert p.comments.include?(c2)
+  assert p.comments.include?(c3)
+end
+
+test "first / last / size / empty?" do |p, c1, c2, c3|
+  assert_equal 3, p.comments.size
+  assert_equal c1, p.comments.first
+  assert_equal c3, p.comments.last
+  assert ! p.comments.empty?
+end
+
+test "replace" do |p, c1, c2, c3|
+  c4 = Comment.create
+
+  p.comments.replace([c4])
+
+  assert_equal [c4], p.comments.to_a
+end
+
+test "push / unshift" do |p, c1, c2, c3|
+  c4 = Comment.create
+  c5 = Comment.create
+
+  p.comments.unshift(c4)
+  p.comments.push(c5)
+
+  assert_equal c4, p.comments.first
+  assert_equal c5, p.comments.last
+end
+
+test "delete" do |p, c1, c2, c3|
+  p.comments.delete(c1)
+  assert_equal 2, p.comments.size
+  assert ! p.comments.include?(c1)
+
+  p.comments.delete(c2)
+  assert_equal 1, p.comments.size
+  assert ! p.comments.include?(c2)
+
+  p.comments.delete(c3)
+  assert p.comments.empty?
+end
+
+test "deleting main model cleans up the collection" do |p, _, _, _|
+  p.delete
+
+  assert_equal 0, Ohm.redis.call("EXISTS", p.key[:comments])
+end
+
+test "#ids returns an array with the ids" do |post, *comments|
+  assert_equal comments.map(&:id), post.comments.ids
+end
+
+test "range" do |p, c1, c2, c3|
+  assert_equal 3, p.comments.range(0, 100).size
+  assert_equal [c1, c2, c3], p.comments.range(0, 2)
+  assert_equal [c1, c2], p.comments.range(0, 1)
+  assert_equal [c2, c3], p.comments.range(1, 2)
+  assert_equal [c1, c2, c3], p.comments.range(0, -1)
+  assert_equal [c1, c2], p.comments.range(0, -2)
+  assert_equal [c2, c3], p.comments.range(1, -1)
+end