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
|
require_relative "spec_helper"
describe "pg_static_cache_updater extension" do
before do
@db = Sequel.connect('mock://postgres')
@db.extend_datasets{def quote_identifiers?; false end}
def @db.listen(chan, opts={})
execute("LISTEN #{chan}")
yield(*opts[:yield])
end
@db.extension(:pg_static_cache_updater)
@model = Class.new(Sequel::Model(@db[:table]))
@model.plugin :static_cache
@db.sqls
end
it "#create_static_cache_update_function should create a function in the database" do
@db.create_static_cache_update_function
@db.sqls.first.gsub(/\s+/, ' ').must_equal " CREATE FUNCTION sequel_static_cache_update() RETURNS trigger LANGUAGE plpgsql AS 'BEGIN PERFORM pg_notify(''sequel_static_cache_update'', TG_RELID::text); RETURN NULL; END ' "
end
it "#create_static_cache_update_function should support :channel_name and :function_name options" do
@db.create_static_cache_update_function(:channel_name=>'foo', :function_name=>'bar')
@db.sqls.first.gsub(/\s+/, ' ').must_equal " CREATE FUNCTION bar() RETURNS trigger LANGUAGE plpgsql AS 'BEGIN PERFORM pg_notify(''foo'', TG_RELID::text); RETURN NULL; END ' "
end
it "#create_static_cache_update_trigger should create a trigger for the database table" do
@db.create_static_cache_update_trigger(:tab)
@db.sqls.first.gsub(/\s+/, ' ').must_equal "CREATE TRIGGER sequel_static_cache_update AFTER INSERT OR UPDATE OR DELETE ON tab EXECUTE PROCEDURE sequel_static_cache_update()"
end
it "#create_static_cache_update_trigger should support :trigger_name and :function_name options" do
@db.create_static_cache_update_trigger(:tab, :trigger_name=>'foo', :function_name=>'bar')
@db.sqls.first.gsub(/\s+/, ' ').must_equal "CREATE TRIGGER foo AFTER INSERT OR UPDATE OR DELETE ON tab EXECUTE PROCEDURE bar()"
end
it "#default_static_cache_update_name should return the default name for function, trigger, and channel" do
@db.default_static_cache_update_name.must_equal :sequel_static_cache_update
end
it "#listen_for_static_cache_updates should listen for changes to model tables and reload model classes" do
@db.fetch = {:v=>1234}
@db.listen_for_static_cache_updates([@model], :yield=>[nil, nil, 1234]).join
@db.sqls.must_equal ["SELECT CAST(CAST('table' AS regclass) AS oid) AS v LIMIT 1", "LISTEN sequel_static_cache_update", "SELECT * FROM table"]
end
it "#listen_for_static_cache_updates should not reload model classes if oid doesn't match" do
@db.fetch = {:v=>1234}
@db.listen_for_static_cache_updates([@model], :yield=>[nil, nil, 12345]).join
@db.sqls.must_equal ["SELECT CAST(CAST('table' AS regclass) AS oid) AS v LIMIT 1", "LISTEN sequel_static_cache_update"]
end
it "#listen_for_static_cache_updates should support a single model argument" do
@db.fetch = {:v=>1234}
@db.listen_for_static_cache_updates(@model, :yield=>[nil, nil, 1234]).join
@db.sqls.must_equal ["SELECT CAST(CAST('table' AS regclass) AS oid) AS v LIMIT 1", "LISTEN sequel_static_cache_update", "SELECT * FROM table"]
end
it "#listen_for_static_cache_updates should support the :channel_name option" do
@db.fetch = {:v=>1234}
@db.listen_for_static_cache_updates([@model], :yield=>[nil, nil, 12345], :channel_name=>:foo).join
@db.sqls.must_equal ["SELECT CAST(CAST('table' AS regclass) AS oid) AS v LIMIT 1", "LISTEN foo"]
end
it "#listen_for_static_cache_updates should raise an error if given an empty array" do
@db.fetch = {:v=>1234}
proc{@db.listen_for_static_cache_updates([])}.must_raise(Sequel::Error)
end
it "#listen_for_static_cache_updates should raise an error if one of the models is not using the static cache plugin" do
@db.fetch = {:v=>1234}
proc{@db.listen_for_static_cache_updates(Class.new(Sequel::Model(@db[:table])))}.must_raise(Sequel::Error)
end
it "#listen_for_static_cache_updates should raise an error if the database doesn't respond to listen" do
@db = Sequel.mock(:host=>'postgres')
@db.extension(:pg_static_cache_updater)
@db.fetch = {:v=>1234}
proc{@db.listen_for_static_cache_updates(Class.new(Sequel::Model(@db[:table])))}.must_raise(Sequel::Error)
end
it "#listen_for_static_cache_updates should handle a :before_thread_exit option" do
a = []
@db.listen_for_static_cache_updates([@model], :yield=>[nil, nil, 12345], :before_thread_exit=>proc{a << 1}).join
a.must_equal [1]
end
it "#listen_for_static_cache_updates should call :before_thread_exit option even if listen raises an exception" do
a = []
@db.listen_for_static_cache_updates([@model], :yield=>[nil, nil, 12345], :after_listen=>proc{raise ArgumentError}, :before_thread_exit=>proc{a << 1}).join
a.must_equal [1]
end
end
|
