[STORE] Fix up tests to work with default document_type and klass on Repository object

Author: estolfo

elasticsearch-persistence/lib/elasticsearch/persistence/repository/naming.rb

@@ -14,8 +14,12 @@ module Naming
 
         # Get or set the class used to initialize domain objects when deserializing them
         #
-        def klass name=nil
-          @klass = name || @klass
+        def klass(name=nil)
+          if name
+            @klass = name
+          else
+            @klass
+          end
         end
 
         # Set the class used to initialize domain objects when deserializing them

elasticsearch-persistence/test/integration/repository/custom_class_test.rb

@@ -30,6 +30,7 @@ class ::MyNotesRepository
             include Elasticsearch::Persistence::Repository
 
             klass MyNote
+            document_type 'my_note'
 
             settings number_of_shards: 1 do
               mapping do
@@ -45,6 +46,8 @@ def deserialize(document)
           end
 
           @repository = MyNotesRepository.new
+          @repository.klass = MyNotesRepository.klass
+          @repository.document_type = MyNotesRepository.document_type
 
           @repository.client.cluster.health wait_for_status: 'yellow'
         end

elasticsearch-persistence/test/integration/repository/default_class_test.rb

@@ -19,6 +19,8 @@ def to_hash
       context "The default repository class" do
         setup do
           @repository = Elasticsearch::Persistence::Repository.new
+          @repository.klass = ::Note
+          @repository.document_type = 'note'
           @repository.client.cluster.health wait_for_status: 'yellow'
         end
 
@@ -105,6 +107,7 @@ def to_hash
         end
 
         should "save and find a plain hash" do
+          @repository.klass = Hash
           @repository.save id: 1, title: 'Hash'
           result = @repository.find(1)
           assert_equal 'Hash', result['_source']['title']

elasticsearch-persistence/test/integration/repository/virtus_model_test.rb

@@ -25,6 +25,7 @@ def set_id(id)
           @repository = Elasticsearch::Persistence::Repository.new do
             index :pages
             klass Page
+            document_type 'page'
 
             def deserialize(document)
               page = klass.new document['_source']