Add type argument to QueryMethods module (#108)

This commit is contained in:
Brian J. Cardiff 2019-08-02 11:54:52 -03:00 committed by GitHub
parent 0100b47754
commit adc755e5e4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 30 additions and 30 deletions

View file

@ -22,11 +22,11 @@ class DummyDriver < DB::Driver
@@connections.try &.clear
end
def build_prepared_statement(query)
def build_prepared_statement(query) : DB::Statement
DummyStatement.new(self, query, true)
end
def build_unprepared_statement(query)
def build_unprepared_statement(query) : DB::Statement
DummyStatement.new(self, query, false)
end
@ -102,13 +102,13 @@ class DummyDriver < DB::Driver
raise DB::Error.new(query) if query == "syntax error"
end
protected def perform_query(args : Enumerable)
protected def perform_query(args : Enumerable) : DB::ResultSet
@connection.as(DummyConnection).check
set_params args
DummyResultSet.new self, @query
end
protected def perform_exec(args : Enumerable)
protected def perform_exec(args : Enumerable) : DB::ExecResult
@connection.as(DummyConnection).check
set_params args
raise DB::Error.new("forced exception due to query") if @query == "raise"
@ -161,16 +161,16 @@ class DummyDriver < DB::Driver
@@last_result_set.not_nil!
end
def move_next
def move_next : Bool
@values = @top_values.shift?
!!@values
end
def column_count
def column_count : Int32
@column_count
end
def column_name(index)
def column_name(index) : String
"c#{index}"
end