fix: assertions

Author: shortcuts

algoliasearch_django/models.py

@@ -324,6 +324,7 @@ def save_record(self, instance, update_fields=None, **kwargs):
                 )
             else:
                 obj = self.get_raw_record(instance)
+                print(obj)
                 self.__client.save_objects(
                     index_name=self.index_name, objects=[obj], wait_for_tasks=True
                 )
@@ -391,9 +392,9 @@ def update_records(self, qs, batch_size=1000, **kwargs):
     def raw_search(self, query="", params=None):
         """Performs a search query and returns the parsed JSON."""
         if params is None:
-            params = SearchParamsObject()
+            params = SearchParamsObject().to_dict()
 
-        params.query = query
+        params["query"] = query
 
         try:
             return self.__client.search_single_index(self.index_name, params).to_dict()

tests/test_engine.py

@@ -54,7 +54,7 @@ def test_is_register(self):
 
     def test_get_adapter(self):
         self.engine.register(Website)
-        self.assertEquals(AlgoliaIndex, self.engine.get_adapter(Website).__class__)
+        self.assertEqual(AlgoliaIndex, self.engine.get_adapter(Website).__class__)
 
     def test_get_adapter_exception(self):
         with self.assertRaises(RegistrationError):
@@ -63,7 +63,7 @@ def test_get_adapter_exception(self):
     def test_get_adapter_from_instance(self):
         self.engine.register(Website)
         instance = Website()
-        self.assertEquals(
+        self.assertEqual(
             AlgoliaIndex, self.engine.get_adapter_from_instance(instance).__class__
         )
 

tests/test_index.py

@@ -252,7 +252,7 @@ class WebsiteIndex(AlgoliaIndex):
             "consequence": {"params": {"query": "other text"}},
         }
 
-        self.index.__client.save_rule(self.index.index_name, rule["objectID"], rule)
+        self.client.save_rule(self.index.index_name, rule["objectID"], rule)
 
         # When reindexing with no settings on the instance
         self.index = WebsiteIndex(Website, self.client, settings.ALGOLIA)