rebel/test/test_exec.rb

require 'minitest/autorun'
require 'helper'
require 'rebel'

class TestExec < Minitest::Test
  include Rebel::SQL

  def setup
    @conn = memdb
  end

  def test_create_table
    assert_raises(SQLite3::SQLException) { conn.execute('SELECT * FROM foo') }
    create_table :foo, id: 'INT', col: 'VARCHAR(255)'
    assert_equal(conn.execute('SELECT * FROM foo'), [])
  end

  def test_drop_table
    create_table :foo, id: 'INT', col: 'VARCHAR(255)'
    assert_equal(conn.execute('SELECT * FROM foo'), [])
    drop_table :foo
    assert_raises(SQLite3::SQLException) { conn.execute('SELECT * FROM foo') }
  end

  def test_insert_into
    create_table :foo, id: 'INT', col: 'VARCHAR(255)'
    insert_into :foo, id: 1, col: 'whatevs'
    assert_equal(conn.execute('SELECT * FROM foo'), [[1, 'whatevs']])
  end

  def test_insert_into_with_many_values
    create_table :foo, id: 'INT', col: 'VARCHAR(255)'
    insert_into :foo,
                { id: 1, col: 'more' },
                { id: 2, col: 'rows' },
                { id: 3, col: 'for the win' }
    assert_equal(conn.execute('SELECT * FROM foo'), [
      [1, 'more'],
      [2, 'rows'],
      [3, 'for the win'],
    ])
  end

  def test_select
    create_table :foo, id: 'INT', col: 'VARCHAR(255)'
    insert_into :foo, id: 1, col: 'whatevs'
    assert_equal(select('*', from: :foo), [[1, 'whatevs']])
  end

  def test_limit
    create_table :foo, id: 'INT', col: 'VARCHAR(255)'
    insert_into :foo, id: 1, col: 'whatevs'
    insert_into :foo, id: 2, col: 'something else'
    assert_equal(select('*', from: :foo, limit: 1), [[1, 'whatevs']])
  end

  def test_order_by
    create_table :foo, id: 'INT', col: 'VARCHAR(255)', value: 'VARCHAR(255)'
    insert_into :foo, id: 1, value: '2', col: 'whatevs'
    insert_into :foo, id: 2, value: '2', col: 'something'
    insert_into :foo, id: 3, value: '1', col: 'else'
    assert_equal(select(:id, from: :foo, order_by: :col), [[3], [2], [1]])
    assert_equal(select(:id, from: :foo, order_by: {id: :desc}), [[3], [2], [1]])
    assert_equal(select(:id, from: :foo, order_by: [value: :asc, id: :asc]), [[3], [1], [2]])
  end
end