From 28b788fa767d89444181464fdfb4853cb75747c0 Mon Sep 17 00:00:00 2001
From: Mike Shoup <mike@shoup.io>
Date: Fri, 28 Jun 2019 10:50:41 -0600
Subject: [PATCH] Add get_view function

---
 src/humulus/couch.py | 6 ++++++
 tests/test_couch.py  | 8 +++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/humulus/couch.py b/src/humulus/couch.py
index ea4bd7b..6fd1da1 100644
--- a/src/humulus/couch.py
+++ b/src/humulus/couch.py
@@ -21,6 +21,7 @@ from pathlib import Path
 
 import click
 from cloudant import CouchDB
+from cloudant.view import View
 from flask import abort, current_app, g
 from flask.cli import with_appcontext
 from slugify import slugify
@@ -151,3 +152,8 @@ def put_designs():
             doc.save()
         else:
             put_doc(data)
+
+
+def get_view(doc_name, view_name):
+    """Return a cloudant.View object matching the specified name."""
+    return View(get_doc(doc_name), view_name)
diff --git a/tests/test_couch.py b/tests/test_couch.py
index 127add2..d276125 100644
--- a/tests/test_couch.py
+++ b/tests/test_couch.py
@@ -15,7 +15,7 @@
 import uuid
 from pathlib import Path
 
-from humulus.couch import get_doc, put_doc, update_doc, put_designs
+from humulus.couch import *
 
 
 def test_put_doc(app):
@@ -90,3 +90,9 @@ def test_put_designs(app, monkeypatch):
         put_designs()
         recipes = get_doc('_design/recipes')
         assert 'by-date' in recipes['views']
+
+
+def test_get_view(app):
+    with app.app_context():
+        view = get_view('_design/recipes', 'by-date')
+    assert view()['total_rows'] > 0