created CouchRest::Model mixin

This commit is contained in:
Chris Anderson 2008-09-29 09:55:40 -07:00
parent ef3055c222
commit ce3a3258bc
3 changed files with 141 additions and 0 deletions

View file

@ -26,6 +26,7 @@ require 'couchrest/monkeypatches'
module CouchRest module CouchRest
autoload :Server, 'couchrest/core/server' autoload :Server, 'couchrest/core/server'
autoload :Database, 'couchrest/core/database' autoload :Database, 'couchrest/core/database'
autoload :Model, 'couchrest/core/model'
autoload :Pager, 'couchrest/helper/pager' autoload :Pager, 'couchrest/helper/pager'
autoload :FileManager, 'couchrest/helper/file_manager' autoload :FileManager, 'couchrest/helper/file_manager'
autoload :Streamer, 'couchrest/helper/streamer' autoload :Streamer, 'couchrest/helper/streamer'

View file

@ -0,0 +1,73 @@
module CouchRest
module Model
class << self
attr_accessor :default_database
end
# instance methods on the model classes
module InstanceMethods
attr_accessor :doc
def initialize doc = {}
self.doc = doc
unless doc['_id'] && doc['_rev']
init_doc
end
end
def database
self.class.database
end
def id
doc['_id']
end
def rev
doc['_rev']
end
def save
result = database.save doc
if result['ok']
doc['_id'] = result['id']
doc['_rev'] = result['rev']
end
result['ok']
end
private
def init_doc
doc['type'] = self.class.to_s
end
end # module InstanceMethods
# these show up as class methods on models that include CouchRest::Model
module ClassMethods
def use_database db
@database = db
end
def database
@database || CouchRest::Model.default_database
end
def uniq_id method
before_create do |model|
model.doc['_id'] = model.send(method)
end
end
end # module ClassMethods
# bookkeeping section
# load the code into the model class
def self.included(klass)
klass.extend ClassMethods
klass.send(:include, InstanceMethods)
end
end # module Model
end # module CouchRest

View file

@ -0,0 +1,67 @@
require File.dirname(__FILE__) + '/../../spec_helper'
class Basic
include CouchRest::Model
end
class Article
include CouchRest::Model
use_database CouchRest.database!('http://localhost:5984/couchrest-model-test')
uniq_id :slug
end
describe CouchRest::Model do
before(:all) do
@cr = CouchRest.new(COUCHHOST)
@db = @cr.database(TESTDB)
@db.delete! rescue nil
@db = @cr.create_db(TESTDB) rescue nil
CouchRest::Model.default_database = CouchRest.database!('http://localhost:5984/couchrest-test')
end
it "should use the default database" do
Basic.database.info['db_name'].should == 'couchrest-test'
end
it "should override the default db" do
Article.database.info['db_name'].should == 'couchrest-model-test'
end
describe "saving a model" do
before(:all) do
@obj = Basic.new
@obj.save.should == true
end
it "should save the doc" do
doc = @obj.database.get @obj.id
doc['_id'].should == @obj.id
end
it "should be set for resaving" do
rev = @obj.rev
@obj.doc['another-key'] = "some value"
@obj.save
@obj.rev.should_not == rev
end
it "should set the id" do
@obj.id.should be_an_instance_of String
end
it "should set the type" do
@obj.doc['type'].should == 'Basic'
end
end
describe "saving a model with a uniq_id configured" do
before(:all) do
@art = Article.new
end
it "should require the slug" do
lambda{@art.save}.should raise_error
@art.slug = 'this-becomes-the-id'
@art.save.should == true
end
end
end