Skip to content

Commit 158cc22

Browse files
committed
fix
1 parent 4020a5e commit 158cc22

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

spec/umbrellio_utils/database_spec.rb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,14 +184,15 @@
184184
let(:tokens_data) { Array.new(10) { |i| { user_id: users[i].id } } }
185185

186186
let!(:db_logger) do
187-
Class.new(Logger) do
187+
logger = Class.new(Logger) do
188188
attr_accessor :queries
189189

190190
def add(_, _, msg)
191191
self.queries ||= []
192192
queries << msg if msg.include?('FROM "users"')
193193
end
194-
end.new(nil).tap { |x| DB.loggers << x }
194+
end.new(nil)
195+
logger.tap { |x| DB.loggers << x }
195196
end
196197

197198
subject(:result_users) do
@@ -205,6 +206,7 @@ def add(_, _, msg)
205206
end
206207

207208
before { UserToken.multi_insert(tokens_data) }
209+
after { DB.loggers.pop }
208210

209211
it "preloads users" do
210212
expect(result_users.map(&:id)).to eq(users.map(&:id).reverse)

0 commit comments

Comments
 (0)