From d5c966ea6c295d7be642d575ef6b6b3f068c5905 Mon Sep 17 00:00:00 2001
From: heck <heck@pep.foundation>
Date: Sat, 10 Jul 2021 13:37:16 +0200
Subject: [PATCH] Test: PityTest - PityUnit add setContext() twice, pointer
 takes reference, Value copies and own the thing....

---
 test/pitytest11/src/PityUnit.hh  |  7 +++++--
 test/pitytest11/src/PityUnit.hxx | 19 +++++++++++++++++++
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/test/pitytest11/src/PityUnit.hh b/test/pitytest11/src/PityUnit.hh
index b8da15c..2b91434 100644
--- a/test/pitytest11/src/PityUnit.hh
+++ b/test/pitytest11/src/PityUnit.hh
@@ -47,10 +47,12 @@ namespace pEp {
             // copy-assign
             PityUnit<TestContext>& operator=(const PityUnit<TestContext>& rhs);
 
+            PityUnit<TestContext>& getSelf() override;
             // clone
             PityUnit<TestContext>* clone() override;
 
-            // Read-Only
+            void setContext(TestContext* ctx);
+            void setContext(TestContext ctx);
             TestContext* getContext() const;
 
         protected:
@@ -60,7 +62,8 @@ namespace pEp {
             void _copyContext(const PityUnit<TestContext>& rhs);
 
             // Fields
-            TestContext* _ctx;                       // nullptr if inherited
+            // nullptr if inherited
+            TestContext* _ctx;
             std::shared_ptr<TestContext> _owned_ctx; // if you copy
             TestFunction _test_func;
         };
diff --git a/test/pitytest11/src/PityUnit.hxx b/test/pitytest11/src/PityUnit.hxx
index 19a3aaf..1ef51b2 100644
--- a/test/pitytest11/src/PityUnit.hxx
+++ b/test/pitytest11/src/PityUnit.hxx
@@ -60,6 +60,12 @@ namespace pEp {
             return *this;
         }
 
+        template<class TestContext>
+        PityUnit<TestContext> &PityUnit<TestContext>::getSelf()
+        {
+            return *this;
+        }
+
         template<class TestContext>
         PityUnit<TestContext> *PityUnit<TestContext>::clone()
         {
@@ -110,6 +116,19 @@ namespace pEp {
                 _ctx = nullptr;
             }
         }
+
+        template<class TestContext>
+        void PityUnit<TestContext>::setContext(TestContext *ctx)
+        {
+            _ctx = ctx;
+        }
+
+        template<class TestContext>
+        void PityUnit<TestContext>::setContext(TestContext ctx)
+        {
+            _owned_ctx = std::shared_ptr<TestContext>(new TestContext(ctx));
+            _ctx = _owned_ctx.get();
+        }
     } // namespace PityTest11
 } // namespace pEp