activerecord/test/cases/arel/crud_test.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

# frozen_string_literal: true

require_relative "helper"

module Arel
  class FakeCrudder < SelectManager
    class FakeEngine
      attr_reader :calls, :connection_pool, :spec, :config

      def initialize
        @calls = []
        @connection_pool = self
        @spec = self
        @config = { adapter: "sqlite3" }
      end

      def connection; self end

      def method_missing(name, *args)
        @calls << [name, args]
      end
    end

    include Crud

    attr_reader :engine
    attr_accessor :ctx

    def initialize(engine = FakeEngine.new)
      super
    end
  end

  describe "crud" do
    describe "insert" do
      it "should call insert on the connection" do
        table = Table.new :users
        fc = FakeCrudder.new
        fc.from table
        im = fc.compile_insert [[table[:id], "foo"]]
        assert_instance_of Arel::InsertManager, im
      end
    end

    describe "update" do
      it "should call update on the connection" do
        table = Table.new :users
        fc = FakeCrudder.new
        fc.from table
        stmt = fc.compile_update [[table[:id], "foo"]], Arel::Attributes::Attribute.new(table, "id")
        assert_instance_of Arel::UpdateManager, stmt
      end
    end

    describe "delete" do
      it "should call delete on the connection" do
        table = Table.new :users
        fc = FakeCrudder.new
        fc.from table
        stmt = fc.compile_delete
        assert_instance_of Arel::DeleteManager, stmt
      end
    end
  end
end