BUG23568257: Add fetch_one method to mysqlx.result

amitab · amitab · commit 5b1279ef9cb0 · 2016-07-15T14:33:42.000+05:30
Added test cases for regression.
diff --git a/lib/mysqlx/result.py b/lib/mysqlx/result.py
@@ -688,6 +688,17 @@ def _page_in_items(self):
             count += 1
         return count
 
+    def fetch_one(self):
+        """ Fetch one item.
+
+        Returns:
+            Row/DbDoc: one result item.
+        """
+        if self._closed:
+            return None
+
+        return self._read_item(False)
+
     def fetch_all(self):
         """Fetch all items.
 
diff --git a/tests/test_mysqlx_crud.py b/tests/test_mysqlx_crud.py
@@ -275,6 +275,11 @@ def test_results(self):
         docs1 = result1.fetch_all()
         self.assertEqual(4, len(docs1))
 
+        result3 = collection.find("age > 28").sort("age DESC").execute()
+        self.assertEqual("Betty", result3.fetch_one()["name"])
+        self.assertEqual("Wilma", result3.fetch_one()["name"])
+        self.assertEqual(None, result3.fetch_one())
+
     # def test_create_index(self):
     #     collection_name = "collection_test"
     #     collection = self.schema.create_collection(collection_name)
@@ -517,6 +522,23 @@ def test_count(self):
         self.assertEqual(table.count(), 1)
         self.schema.drop_table(table_name)
 
+    def test_results(self):
+        table_name = "{0}.test".format(self.schema_name)
+
+        self.node_session.sql("CREATE TABLE {0}(age INT, name VARCHAR(50))"
+                              "".format(table_name)).execute()
+        self.node_session.sql("INSERT INTO {0} VALUES (21, 'Fred')"
+                              "".format(table_name)).execute()
+        self.node_session.sql("INSERT INTO {0} VALUES (28, 'Barney')"
+                              "".format(table_name)).execute()
+
+        table = self.schema.get_table("test")
+        result = table.select().execute()
+
+        self.assertEqual("Fred", result.fetch_one()["name"])
+        self.assertEqual("Barney", result.fetch_one()["name"])
+        self.assertEqual(None, result.fetch_one())
+
     def test_multiple_resultsets(self):
         self.node_session.sql("CREATE PROCEDURE {0}.spProc() BEGIN SELECT 1; "
                               "SELECT 2; SELECT 'a'; END"
