#NEWLINE# @ patch ( 'nefertari.elasticsearch.ES._bulk' ) #NEWLINE# @ patch ( 'nefertari.elasticsearch.ES.api.mget' ) #NEWLINE# def test_index_missing_documents( self , mock_mget , mock_bulk ) : #NEWLINE# #INDENT# obj = es . ES ( 'Foo' , 'foondex' ) #NEWLINE# documents = [ { #NEWLINE# #INDENT# '_pk' : 1 , #NEWLINE# 'name' : 'foo' , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# '_pk' : 2 , #NEWLINE# 'name' : 'bar' , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# '_pk' : 3 , #NEWLINE# 'name' : 'baz' , #NEWLINE# #UNINDENT# } ] #NEWLINE# mock_mget . return_value = { #NEWLINE# #INDENT# 'docs' : [ { #NEWLINE# #INDENT# '_id' : '1' , #NEWLINE# 'name' : 'foo' , #NEWLINE# 'found' : False , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# '_id' : '2' , #NEWLINE# 'name' : 'bar' , #NEWLINE# 'found' : True , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# '_id' : '3' , #NEWLINE# 'name' : 'baz' , #NEWLINE# #UNINDENT# } ] , #NEWLINE# #UNINDENT# } #NEWLINE# obj . index_missing_documents ( self ) #NEWLINE# mock_mget . assert_called_once_with ( index = 'foondex' , doc_type = 'Foo' , fields = [ '_id' ] , body = { #NEWLINE# #INDENT# 'ids' : [ 1 , 2 , 3 ] , #NEWLINE# #UNINDENT# } ) #NEWLINE# mock_bulk . assert_called_once_with ( 'index' , [ { #NEWLINE# #INDENT# '_pk' : 1 , #NEWLINE# 'name' : 'foo' , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# '_pk' : 3 , #NEWLINE# 'name' : 'baz' , #NEWLINE# #UNINDENT# } ] , None )