aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/cases/connection_adapters/schema_cache_test.rb
blob: 727cab77f552725c685febaa5f894c76d5074b52 (plain) (blame)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# frozen_string_literal: true

require "cases/helper"

module ActiveRecord
  module ConnectionAdapters
    class SchemaCacheTest < ActiveRecord::TestCase
      def setup
        connection = ActiveRecord::Base.connection
        @cache     = SchemaCache.new connection
      end

      def test_primary_key
        assert_equal "id", @cache.primary_keys("posts")
      end

      def test_yaml_dump_and_load
        @cache.columns("posts")
        @cache.columns_hash("posts")
        @cache.data_sources("posts")
        @cache.primary_keys("posts")

        new_cache = YAML.load(YAML.dump(@cache))
        assert_no_queries do
          assert_equal 12, new_cache.columns("posts").size
          assert_equal 12, new_cache.columns_hash("posts").size
          assert new_cache.data_sources("posts")
          assert_equal "id", new_cache.primary_keys("posts")
        end
      end

      def test_yaml_loads_5_1_dump
        body = File.open(schema_dump_path).read
        cache = YAML.load(body)

        assert_no_queries do
          assert_equal 11, cache.columns("posts").size
          assert_equal 11, cache.columns_hash("posts").size
          assert cache.data_sources("posts")
          assert_equal "id", cache.primary_keys("posts")
        end
      end

      def test_primary_key_for_non_existent_table
        assert_nil @cache.primary_keys("omgponies")
      end

      def test_caches_columns
        columns = @cache.columns("posts")
        assert_equal columns, @cache.columns("posts")
      end

      def test_caches_columns_hash
        columns_hash = @cache.columns_hash("posts")
        assert_equal columns_hash, @cache.columns_hash("posts")
      end

      def test_clearing
        @cache.columns("posts")
        @cache.columns_hash("posts")
        @cache.data_sources("posts")
        @cache.primary_keys("posts")

        @cache.clear!

        assert_equal 0, @cache.size
      end

      def test_dump_and_load
        @cache.columns("posts")
        @cache.columns_hash("posts")
        @cache.data_sources("posts")
        @cache.primary_keys("posts")

        @cache = Marshal.load(Marshal.dump(@cache))

        assert_no_queries do
          assert_equal 12, @cache.columns("posts").size
          assert_equal 12, @cache.columns_hash("posts").size
          assert @cache.data_sources("posts")
          assert_equal "id", @cache.primary_keys("posts")
        end
      end

      def test_data_source_exist
        assert @cache.data_source_exists?("posts")
        assert_not @cache.data_source_exists?("foo")
      end

      def test_clear_data_source_cache
        @cache.clear_data_source_cache!("posts")
      end

      test "#columns_hash? is populated by #columns_hash" do
        assert_not @cache.columns_hash?("posts")

        @cache.columns_hash("posts")

        assert @cache.columns_hash?("posts")
      end

      test "#columns_hash? is not populated by #data_source_exists?" do
        assert_not @cache.columns_hash?("posts")

        @cache.data_source_exists?("posts")

        assert_not @cache.columns_hash?("posts")
      end

      private

        def schema_dump_path
          "test/assets/schema_dump_5_1.yml"
        end
    end
  end
end