Class: Lafcadio::ObjectStore::MockDbBridge

Inherits:
Object
  • Object
show all
Defined in:
lib/lafcadio/mock.rb

Overview

:nodoc:

Defined Under Namespace

Classes: Transaction

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ MockDbBridge

Returns a new instance of MockDbBridge.



10
11
12
13
14
15
 
# File 'lib/lafcadio/mock.rb', line 10

def initialize
	@objects = {}
	@next_pk_ids = {}
	@queries = []
	@transaction = nil
end

Instance Attribute Details

#last_pk_id_inserted ⇒ Object (readonly)

Returns the value of attribute last_pk_id_inserted.



7
8
9
 
# File 'lib/lafcadio/mock.rb', line 7

def last_pk_id_inserted
  @last_pk_id_inserted
end

#next_pk_ids ⇒ Object

Returns the value of attribute next_pk_ids.



8
9
10
 
# File 'lib/lafcadio/mock.rb', line 8

def next_pk_ids
  @next_pk_ids
end

#objects ⇒ Object

Returns the value of attribute objects.



8
9
10
 
# File 'lib/lafcadio/mock.rb', line 8

def objects
  @objects
end

#queries(domain_class = nil) ⇒ Object

Returns the value of attribute queries.



8
9
10
 
# File 'lib/lafcadio/mock.rb', line 8

def queries
  @queries
end

Instance Method Details

#all(domain_class) ⇒ Object 



17
18
19
 
# File 'lib/lafcadio/mock.rb', line 17

def all( domain_class )
	@objects[domain_class] ? @objects[domain_class].values : []
end

#commit(db_object) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/lafcadio/mock.rb', line 21

def commit(db_object)
	if db_object.pk_id and !db_object.pk_id.is_a?( Integer )
		raise ArgumentError
	end
	if @transaction
		@transaction << db_object
	else
		objects_by_domain_class = objects_by_domain_class(
			db_object.domain_class
		)
		if db_object.delete
			objects_by_domain_class.delete( db_object.pk_id )
		else
			object_pk_id = pre_commit_pk_id( db_object )
			objects_by_domain_class[object_pk_id] = db_object
		end
	end
end

#group_query(query) ⇒ Object 



40
41
42
 
# File 'lib/lafcadio/mock.rb', line 40

def group_query( query )
	query.collect objects_by_domain_class( query.domain_class ).values
end

#next_pk_id(domain_class) ⇒ Object 



44
45
46
47
48
49
 
# File 'lib/lafcadio/mock.rb', line 44

def next_pk_id( domain_class )
	dobjs = objects_by_domain_class( domain_class ).values
	dobjs.inject( 0 ) { |memo, obj|
		memo > obj.pk_id ? memo : obj.pk_id
	} + 1
end

#objects_by_domain_class(domain_class) ⇒ Object 



51
52
53
54
 
# File 'lib/lafcadio/mock.rb', line 51

def objects_by_domain_class( domain_class )
	@objects[domain_class] = {} unless @objects[domain_class]
	@objects[domain_class]
end

#pre_commit_pk_id(domain_object) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/lafcadio/mock.rb', line 56

def pre_commit_pk_id( domain_object )
	@next_pk_ids = {} unless @next_pk_ids
	if (next_pk_id = @next_pk_ids[domain_object.domain_class])
		@next_pk_ids[domain_object.domain_class] = nil
		@last_pk_id_inserted = next_pk_id
	elsif domain_object.pk_id
		domain_object.pk_id
	else
		pk_ids = objects_by_domain_class( domain_object.domain_class ).keys
		@last_pk_id_inserted = pk_ids.max ? pk_ids.max + 1 : 1
	end
end

#query_count(sql) ⇒ Object 



75
76
77
 
# File 'lib/lafcadio/mock.rb', line 75

def query_count( sql )
	@queries.select { |qry| qry.to_sql == sql }.size
end

#rollback ⇒ Object 



79
 
# File 'lib/lafcadio/mock.rb', line 79

def rollback; end

#select_dobjs(query) ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/lafcadio/mock.rb', line 81

def select_dobjs( query )
	@queries << query
	domain_class = query.domain_class
	all = all domain_class
	if @transaction
		from_tr = @transaction.select { |dobj|
			dobj.domain_class == query.domain_class
		}
		all = all.concat from_tr
	end
	objects = all.select { |dobj| query.dobj_satisfies?( dobj ) }
	query.order_and_limit_collection objects
end

#set_next_pk_id(domain_class, npi) ⇒ Object 



95
96
97
98
 
# File 'lib/lafcadio/mock.rb', line 95

def set_next_pk_id( domain_class, npi )
	@next_pk_ids = {} unless @next_pk_ids
	@next_pk_ids[ domain_class ] = npi
end

#transaction(action) ⇒ Object 



100
101
102
103
104
105
106
107
 
# File 'lib/lafcadio/mock.rb', line 100

def transaction( action )
	tr = MockDbBridge::Transaction.new
	@transaction = tr
	action.call tr
	@transaction = nil
	tr.uniq!
	tr.each do |dobj_to_commit| commit( dobj_to_commit ); end
end

#transactional_clone ⇒ Object 



109
110
111
112
113
114
115
 
# File 'lib/lafcadio/mock.rb', line 109

def transactional_clone
	tc = clone
	tc.objects = objects.clone
	tc.next_pk_ids = next_pk_ids.clone
	tc.queries = queries.clone
	tc
end