Have obnam.Operation remember the obnam.Application instance it was given.

2 files changed, 11 insertions, 3 deletions

diff --git a/obnam/oper.py b/obnam/oper.py
index f174d631..d610e335 100644
--- a/obnam/oper.py
+++ b/obnam/oper.py
@@ -23,4 +23,8 @@ class Operation:
     name = None
 
     def __init__(self, app, args):
-        pass
+        self._app = app
+
+    def get_application(self):
+        """Return application this operation instance will use."""
+        return self._app
diff --git a/obnam/operTests.py b/obnam/operTests.py
index 272bf23c..d9dfd334 100644
--- a/obnam/operTests.py
+++ b/obnam/operTests.py
@@ -27,8 +27,12 @@ class OperationTests(unittest.TestCase):
 
     def setUp(self):
         context = obnam.context.Context()
-        app = obnam.Application(context)
-        self.op = obnam.Operation(app, ["pink", "pretty"])
+        self.app = obnam.Application(context)
+        self.args = ["pink", "pretty"]
+        self.op = obnam.Operation(self.app, self.args)
 
     def testNameIsNone(self):
         self.failUnlessEqual(self.op.name, None)
+
+    def testHasRightApplication(self):
+        self.failUnlessEqual(self.op.get_application(), self.app)