From 021efae95bef1b3baa0b5227c7f9365864b3cde6 Mon Sep 17 00:00:00 2001 From: Jorge Bejar Date: Sun, 3 Nov 2013 18:21:21 -0200 Subject: [PATCH] Change hardcoded class in favor of using provided serializer class in AR integration tests --- test/integration/active_record/active_record_test.rb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/integration/active_record/active_record_test.rb b/test/integration/active_record/active_record_test.rb index 9d9ac637..833795c2 100644 --- a/test/integration/active_record/active_record_test.rb +++ b/test/integration/active_record/active_record_test.rb @@ -58,17 +58,17 @@ module ActiveModel private - def embed(klass, options = {}) - old_assocs = Hash[ARPostSerializer._associations.to_a.map { |(name, association)| [name, association.dup] }] + def embed(serializer_class, options = {}) + old_assocs = Hash[serializer_class._associations.to_a.map { |(name, association)| [name, association.dup] }] - ARPostSerializer._associations.each_value do |association| + serializer_class._associations.each_value do |association| association.embed = options[:embed] association.embed_in_root = options[:embed_in_root] end yield ensure - ARPostSerializer._associations = old_assocs + serializer_class._associations = old_assocs end end end