Database class for Sequel's mock adapter.
Map of database type names to module names, used for handling mock adapters for specific database types.
Procs to run for specific database types to get the mock adapter to work with the shared adapter
Set the autogenerated primary key integer to be returned when running an insert query. Argument types supported:
Return nil for all inserts
Starting integer for next insert, with futher inserts getting an incremented value
First insert gets the first value in the array, second gets the second value, etc.
Called with the insert SQL query, uses the value returned
Should be an Exception subclass, will create a new instance an raise it wrapped in a DatabaseError.
Set the columns to set in the dataset when the dataset fetches rows. Argument types supported:
Set no columns
Array of Symbols: Used for all datasets Array (otherwise): First retrieval gets the first value in the
array, second gets the second value, etc.
Called with the select SQL query, uses the value returned, which should be an array of symbols
Set the hashes to yield by execute when retrieving rows. Argument types supported:
Yield no rows
Always yield a single row with this hash
Yield separately for each hash in this array
First retrieval gets the first value in the array, second gets the second value, etc.
Called with the select SQL query, uses the value returned, which should be a hash or array of hashes.
Should be an Exception subclass, will create a new instance an raise it wrapped in a DatabaseError.
Set the number of rows to return from update or delete. Argument types supported:
Return 0 for all updates and deletes
Used for all updates and deletes
First update/delete gets the first value in the array, second gets the second value, etc.
Called with the update/delete SQL query, uses the value returned.
Should be an Exception subclass, will create a new instance an raise it wrapped in a DatabaseError.
Mock the server version, useful when using the shared adapters
Additional options supported:
Call autoid= with the value
Call columns= with the value
Call fetch= with the value
Call numrows= with the value
A module the object is extended with.
The array to store the SQL queries in.
# File lib/sequel/adapters/mock.rb, line 142 def initialize(opts={}) super opts = @opts @sqls = opts[:sqls] || [] if mod_name = SHARED_ADAPTERS[opts[:host]] @shared_adapter = true require "sequel/adapters/shared/#{opts[:host]}" extend Sequel.const_get(mod_name)::DatabaseMethods extend_datasets Sequel.const_get(mod_name)::DatasetMethods if pr = SHARED_ADAPTER_SETUP[opts[:host]] pr.call(self) end else @shared_adapter = false end self.autoid = opts[:autoid] self.columns = opts[:columns] self.fetch = opts[:fetch] self.numrows = opts[:numrows] extend(opts[:extend]) if opts[:extend] sqls end
Return a related Connection option connecting to the given shard.
# File lib/sequel/adapters/mock.rb, line 166 def connect(server) Connection.new(self, server, server_opts(server)) end
# File lib/sequel/adapters/mock.rb, line 170 def disconnect_connection(c) end
Store the sql used for later retrieval with sqls, and return the appropriate value using either the autoid, fetch, or numrows methods.
# File lib/sequel/adapters/mock.rb, line 176 def execute(sql, opts={}, &block) synchronize(opts[:server]){|c| _execute(c, sql, opts, &block)} end
Store the sql used, and return the value of the numrows method.
# File lib/sequel/adapters/mock.rb, line 182 def execute_dui(sql, opts={}) execute(sql, opts.merge(:meth=>:numrows)) end
Store the sql used, and return the value of the autoid method.
# File lib/sequel/adapters/mock.rb, line 187 def execute_insert(sql, opts={}) execute(sql, opts.merge(:meth=>:autoid)) end
Enable use of savepoints.
# File lib/sequel/adapters/mock.rb, line 200 def supports_savepoints? shared_adapter? ? super : true end
# File lib/sequel/adapters/mock.rb, line 206 def _autoid(sql, v, ds=nil) case v when Integer if ds ds.autoid += 1 if ds.autoid.is_a?(Integer) else @autoid += 1 end v else _nextres(v, sql, nil) end end
# File lib/sequel/adapters/mock.rb, line 220 def _execute(c, sql, opts={}, &block) sql += " -- args: #{opts[:arguments].inspect}" if opts[:arguments] sql += " -- #{@opts[:append]}" if @opts[:append] sql += " -- #{c.server.is_a?(Symbol) ? c.server : c.server.inspect}" if c.server != :default log_info(sql) unless opts[:log] == false @sqls << sql ds = opts[:dataset] begin if block columns(ds, sql) if ds _fetch(sql, (ds._fetch if ds) || @fetch, &block) elsif meth = opts[:meth] if meth == :numrows _numrows(sql, (ds.numrows if ds) || @numrows) else v = ds.autoid if ds _autoid(sql, v || @autoid, (ds if v)) end end rescue => e raise_error(e) end end
# File lib/sequel/adapters/mock.rb, line 245 def _fetch(sql, f, &block) case f when Hash yield f.dup when Array if f.all?{|h| h.is_a?(Hash)} f.each{|h| yield h.dup} else _fetch(sql, f.shift, &block) end when Proc h = f.call(sql) if h.is_a?(Hash) yield h.dup elsif h h.each{|h1| yield h1.dup} end when Class if f < Exception raise f else raise Error, "Invalid @autoid/@numrows attribute: #{v.inspect}" end when nil # nothing else raise Error, "Invalid @fetch attribute: #{f.inspect}" end end
# File lib/sequel/adapters/mock.rb, line 275 def _nextres(v, sql, default) case v when Integer v when Array v.empty? ? default : _nextres(v.shift, sql, default) when Proc v.call(sql) when Class if v < Exception raise v else raise Error, "Invalid @autoid/@numrows attribute: #{v.inspect}" end when nil default else raise Error, "Invalid @autoid/@numrows attribute: #{v.inspect}" end end
# File lib/sequel/adapters/mock.rb, line 296 def _numrows(sql, v) _nextres(v, sql, 0) end
# File lib/sequel/adapters/mock.rb, line 300 def columns(ds, sql, cs=@columns) case cs when Array unless cs.empty? if cs.all?{|c| c.is_a?(Symbol)} ds.columns(*cs) else columns(ds, sql, cs.shift) end end when Proc ds.columns(*cs.call(sql)) when nil # nothing else raise Error, "Invalid @columns attribute: #{cs.inspect}" end end
# File lib/sequel/adapters/mock.rb, line 323 def identifier_input_method_default shared_adapter? ? super : nil end
# File lib/sequel/adapters/mock.rb, line 327 def identifier_output_method_default shared_adapter? ? super : nil end
# File lib/sequel/adapters/mock.rb, line 319 def quote_identifiers_default shared_adapter? ? super : false end