diff options
Diffstat (limited to 'test/support/fake_record.rb')
-rw-r--r-- | test/support/fake_record.rb | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/test/support/fake_record.rb b/test/support/fake_record.rb new file mode 100644 index 0000000000..ef3cc6a291 --- /dev/null +++ b/test/support/fake_record.rb @@ -0,0 +1,89 @@ +module FakeRecord + class Column < Struct.new(:name, :type) + end + + class Connection + attr_reader :tables + + def initialize + @tables = %w{ users photos developers } + @columns = { + 'users' => [ + Column.new('id', :integer), + Column.new('name', :string) + ] + } + @primary_keys = { + 'users' => 'id' + } + end + + def primary_key name + @primary_keys[name.to_s] + end + + def table_exists? name + @tables.include? name.to_s + end + + def columns name, message = nil + @columns[name.to_s] + end + + def quote_table_name name + "\"#{name.to_s}\"" + end + + def quote_column_name name + "\"#{name.to_s}\"" + end + + def quote thing, column = nil + if column && column.type == :integer + return 'NULL' if thing.nil? + return thing.to_i + end + + case thing + when true + "'t'" + when false + "'f'" + when nil + 'NULL' + when Numeric + thing + else + "'#{thing}'" + end + end + end + + class ConnectionPool + class Spec < Struct.new(:config) + end + + attr_reader :spec, :connection + + def initialize + @spec = Spec.new(:adapter => 'sqlite3') + @connection = Connection.new + end + + def with_connection + yield connection + end + end + + class Base + attr_accessor :connection_pool + + def initialize + @connection_pool = ConnectionPool.new + end + + def connection + connection_pool.connection + end + end +end |