testing - replace App to allow flexibility of assigned objects

tbone-framework · Dec 31, 2018 · faadfcb · faadfcb
1 parent 6d6df0d
commit faadfcb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tbone/testing/__init__.py b/tbone/testing/__init__.py
@@ -4,10 +4,14 @@
 from collections import namedtuple
 
 
-App = namedtuple('App', 'db')
 Response = namedtuple('Response', 'headers, payload, status')
 
 
+class App(object):
+    def __init__(self, db=None, **kwargs):
+        self.db = db
+
+
 class Request(dict):
     __slots__ = (
         'app', 'headers', 'method', 'body', 'args', 'url', 'user'