adding possibility to search on a endpoint only object + changing resource

Author: icepick4

onepyece/interface.py

@@ -72,6 +72,12 @@ def update_dict_recursively(self, data_object):
             else:
                 self.__dict__[key] = value
 
-    def search(self, search=None, resource=None):
+    def search(self, search, resource=None):
         if self.search_term is None and self.resource is None:
-            pass
+            return API(self.endpoint, search, resource)
+        raise ValueError(f"You can't instantiate a search API object because you already have {self.search_term} as search and {self.resource} as resource")
+
+    def edit_resource(self, resource):
+        if self.search is None:
+            raise ValueError('You can\'t change the resource without search use the method search()')
+        self.__dict__ = API(self.endpoint, self.search_term, resource).__dict__

tests/test_interface.py

@@ -58,3 +58,21 @@ def test_iterate_object(self):
     def test_recursive_dict(self):
         api_object = interface.API('hakis', 'characters_id', 1)
         self.assertIsInstance(api_object[0].characters.haki_character, interface.API)
+
+    def test_search_method(self):
+        api_object = interface.API('characters').search('job', 'capitaine')
+        self.assertEqual(api_object[1].french_name, 'Cavendish')
+
+    def test_search_method_with_search(self):
+        with self.assertRaises(ValueError):
+            interface.API('characters', 'count').search('job', 'capitaine')
+
+    def test_search_method_with_search_and_resource(self):
+        with self.assertRaises(ValueError):
+            interface.API('characters', 'status', 'décédé').search('job', 'capitaine')
+
+    def test_edit_resource(self):
+        api_object = interface.API('characters', 'name', 'luffy')
+        self.assertEqual(api_object.french_name, 'Monkey D Luffy')
+        api_object.edit_resource('zoro')
+        self.assertEqual(api_object.french_name, 'Roronoa Zoro')