Fix mock opensearch tests

tests/test_opensearch_mock.py

@@ -78,10 +78,9 @@ def test_push(self, abcd):
         assert isinstance(atoms_2, Atoms)
         atoms_2.set_cell([1, 1, 1])
 
+        abcd.refresh()
         result = AtomsModel(
-            None,
-            None,
-            abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
+            dict=abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
         ).to_ase()
         assert atoms_1 == result
         assert atoms_2 != result
@@ -117,17 +116,14 @@ def test_bulk(self, abcd):
         atoms_list.append(atoms_1)
         atoms_list.append(atoms_2)
         abcd.push(atoms_list)
+        abcd.refresh()
         assert abcd.count() == 2
 
         result_1 = AtomsModel(
-            None,
-            None,
-            abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
+            dict=abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
         ).to_ase()
         result_2 = AtomsModel(
-            None,
-            None,
-            abcd.client.search(index="test_index")["hits"]["hits"][1]["_source"],
+            dict=abcd.client.search(index="test_index")["hits"]["hits"][1]["_source"],
         ).to_ase()
         assert atoms_1 == result_1
         assert atoms_2 == result_2
@@ -151,4 +147,5 @@ def test_count(self, abcd):
         atoms.set_cell([1, 1, 1])
         abcd.push(atoms)
         abcd.push(atoms)
+        abcd.refresh()
         assert abcd.count() == 2