add test for .one()

mambocab · mambocab · commit bff974fa7a90 · 2018-02-28T16:33:47.000-05:00
diff --git a/tests/unit/test_resultset.py b/tests/unit/test_resultset.py
@@ -187,3 +187,10 @@ def test_was_applied(self):
         for applied in (True, False):
             rs = ResultSet(Mock(row_factory=row_factory), [{'[applied]': applied}])
             self.assertEqual(rs.was_applied, applied)
+
+    def test_one(self):
+        # no pages
+        first, second = Mock(), Mock()
+        rs = ResultSet(Mock(has_more_pages=False), [first, second])
+
+        self.assertEqual(rs.one(), first)
