From cd3e5e9de2ece6b7f802951f8c0bbd21339ebe73 Mon Sep 17 00:00:00 2001
From: Santiago Pastorino <santiago@wyeworks.com>
Date: Tue, 20 Aug 2013 14:33:33 -0300
Subject: [PATCH] Assign the association in the setup method

---
 .../active_model/serializer/has_one_test.rb   | 21 +++++++++----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/test/unit/active_model/serializer/has_one_test.rb b/test/unit/active_model/serializer/has_one_test.rb
index c2e770f2..e49fd0b0 100644
--- a/test/unit/active_model/serializer/has_one_test.rb
+++ b/test/unit/active_model/serializer/has_one_test.rb
@@ -7,16 +7,15 @@ module ActiveModel
       def setup
         @user = User.new({ name: 'Name 1', email: 'mail@server.com', gender: 'M' })
         @user_serializer = UserSerializer.new(@user)
-        @user_serializer.class._associations[0].include = false
-        @user_serializer.class._associations[0].embed = :ids
+        @association = UserSerializer._associations[0]
+        @association.include = false
+        @association.embed = :ids
       end
 
       def test_associations_definition
-        associations = @user_serializer.class._associations
-
-        assert_equal 1, associations.length
-        assert_kind_of Association::HasOne, associations[0]
-        assert_equal 'profile', associations[0].name
+        assert_equal 1, UserSerializer._associations.length
+        assert_kind_of Association::HasOne, @association
+        assert_equal 'profile', @association.name
       end
 
       def test_associations_embedding_ids_serialization_using_serializable_hash
@@ -32,28 +31,28 @@ module ActiveModel
       end
 
       def test_associations_embedding_objects_serialization_using_serializable_hash
-        @user_serializer.class._associations[0].embed = :objects
+        @association.embed = :objects
         assert_equal({
           'name' => 'Name 1', 'email' => 'mail@server.com', 'profile' => { 'name' => 'N1', 'description' => 'D1' }
         }, @user_serializer.serializable_hash)
       end
 
       def test_associations_embedding_objects_serialization_using_as_json
-        @user_serializer.class._associations[0].embed = :objects
+        @association.embed = :objects
         assert_equal({
           'name' => 'Name 1', 'email' => 'mail@server.com', 'profile' => { 'name' => 'N1', 'description' => 'D1' }
         }, @user_serializer.as_json)
       end
 
       def test_associations_embedding_ids_including_objects_serialization_using_serializable_hash
-        @user_serializer.class._associations[0].include = true
+        @association.include = true
         assert_equal({
           'name' => 'Name 1', 'email' => 'mail@server.com', 'profile_id' => @user.profile.object_id, 'profile' => { 'name' => 'N1', 'description' => 'D1' }
         }, @user_serializer.serializable_hash)
       end
 
       def test_associations_embedding_ids_including_objects_serialization_using_as_json
-        @user_serializer.class._associations[0].include = true
+        @association.include = true
         assert_equal({
           'name' => 'Name 1', 'email' => 'mail@server.com', 'profile_id' => @user.profile.object_id, 'profile' => { 'name' => 'N1', 'description' => 'D1' }
         }, @user_serializer.as_json)