Contents

# encoding: UTF-8

require File.expand_path("./helper", File.dirname(__FILE__))

class User < Ohm::Model
  attribute :email
  unique :email
  unique :provider

  def self.[](id)
    super(id.to_i)
  end

  def provider
    email[/@(.*?).com/, 1]
  end
end

setup do
  User.create(email: "[e-mail]")
end

test "findability" do |u|
  assert_equal u, User.with(:email, "[e-mail]")
end

test "raises when it already exists during create" do
  assert_raise Ohm::UniqueIndexViolation do
    User.create(email: "[e-mail]")
  end
end

test "raises when it already exists during save" do
  u = User.create(email: "[e-mail]")
  u.email = "[e-mail]"

  assert_raise Ohm::UniqueIndexViolation do
    u.save
  end
end

test "doesn't raise when saving again and again" do |u|
  ex = nil

  begin
    User[u.id].save
  rescue Exception => e
    ex = e
  end

  assert_equal nil, ex
end

test "removes the previous index when changing" do
  u = User.create(email: "[e-mail]")
  u.update(email: "[e-mail]")

  assert_equal nil, User.with(:email, "[e-mail]")
  assert_equal nil, User.key[:unique][:email].hget("[e-mail]")
  assert_equal u, User.with(:email, "[e-mail]")
end

test "removes the previous index when deleting" do |u|
  u.delete

  assert_equal nil, User.with(:email, "[e-mail]")
  assert_equal nil, User.key[:unique][:email].hget("[e-mail]")
end

test "unique virtual attribute" do
  u = User.create(email: "[e-mail]")

  assert_equal u, User.with(:provider, "yahoo")

  # Yahoo should be allowed because this user is the one reserved for it.
  u.update(email: "[e-mail]")

  # `a` is not allowed though.
  assert_raise Ohm::UniqueIndexViolation do
    u.update(email: "[e-mail]")
  end

  # And so is yahoo if we try creating a different user.
  assert_raise Ohm::UniqueIndexViolation do
    User.create(email: "[e-mail]")
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
ohm-1.0.0.rc1	test/uniques.rb
ohm-1.0.0.alpha2	test/uniques.rb
ohm-1.0.0.alpha1	test/uniques.rb