1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
require_relative 'test_helper'
class DatabaseMethodTest < MiniTest::Test
class TestModel < CouchRest::Model::Base
include CouchRest::Model::DatabaseMethod
use_database_method :db_name
property :dbname, String
property :confirm, String
def db_name
"test_db_#{self[:dbname]}"
end
end
def test_instance_method
doc1 = TestModel.new({:dbname => 'one'})
doc1.database.create!
assert doc1.database.root.ends_with?('test_db_one')
assert doc1.save
doc1.update_attributes(:confirm => 'yep')
doc2 = TestModel.new({:dbname => 'two'})
doc2.database.create!
assert doc2.database.root.ends_with?('test_db_two')
assert doc2.save
doc2.confirm = 'sure'
doc2.save!
doc1_copy = CouchRest.get([doc1.database.root, doc1.id].join('/'))
assert_equal "yep", doc1_copy["confirm"]
doc2_copy = CouchRest.get([doc2.database.root, doc2.id].join('/'))
assert_equal "sure", doc2_copy["confirm"]
doc1.database.delete!
doc2.database.delete!
end
def test_switch_db
doc_red = TestModel.new({:dbname => 'red', :confirm => 'rose'})
doc_red.database.create!
root = doc_red.database.root
doc_blue = doc_red.clone
doc_blue.dbname = 'blue'
doc_blue.database!
doc_blue.save!
doc_blue_copy = CouchRest.get([root.sub('red','blue'), doc_blue.id].join('/'))
assert_equal "rose", doc_blue_copy["confirm"]
doc_red.database.delete!
doc_blue.database.delete!
end
#
# A test scenario for database_method in which some user accounts
# are stored in a seperate temporary database (so that the test
# accounts don't bloat the normal database).
#
class User < CouchRest::Model::Base
include CouchRest::Model::DatabaseMethod
use_database_method :db_name
property :login, String
before_save :create_db
class << self
def get(id, db = database)
result = super(id, db)
if result.nil?
return super(id, choose_database('test-user'))
else
return result
end
end
alias :find :get
end
protected
def self.db_name(login = nil)
if !login.nil? && login =~ /test-user/
'tmp_users'
else
'users'
end
end
def db_name
self.class.db_name(self.login)
end
def create_db
unless database_exists?(db_name)
self.database!
end
end
end
def test_tmp_user_db
user1 = User.new({:login => 'test-user-1'})
assert user1.save
assert User.find(user1.id), 'should find user in tmp_users'
assert_equal user1.login, User.find(user1.id).login
assert_equal 'test-user-1', User.server.database('couchrest_tmp_users').get(user1.id)['login']
assert_raises CouchRest::NotFound do
User.server.database('couchrest_users').get(user1.id)
end
end
end
|
