Fix test container remain in docker after tests completion. IMPORTANT.

Issue-ID: CPS-124
Change-Id: I6e423792c1a63d0f828577ca01d32366d2e693bb
Signed-off-by: Ruslan Kashapov <ruslan.kashapov@pantheon.tech>

1 files changed, 5 insertions, 1 deletions

diff --git a/cps-ri/src/test/java/org/onap/cps/DatabaseTestContainer.java b/cps-ri/src/test/java/org/onap/cps/DatabaseTestContainer.java
index 759cc9531..a621b2ca4 100644
--- a/cps-ri/src/test/java/org/onap/cps/DatabaseTestContainer.java
+++ b/cps-ri/src/test/java/org/onap/cps/DatabaseTestContainer.java
@@ -44,6 +44,7 @@ public class DatabaseTestContainer extends PostgreSQLContainer<DatabaseTestConta
     public static DatabaseTestContainer getInstance() {
         if (databaseTestContainer == null) {
             databaseTestContainer = new DatabaseTestContainer();
+            Runtime.getRuntime().addShutdownHook(new Thread(databaseTestContainer::terminate));
         }
         return databaseTestContainer;
     }
@@ -58,7 +59,10 @@ public class DatabaseTestContainer extends PostgreSQLContainer<DatabaseTestConta
 
     @Override
     public void stop() {
-        //do nothing, JVM handles shut down
+        // do nothing on test completion, image removal will be performed via terminate() on JVM shutdown
     }
 
+    private void terminate() {
+        super.stop();
+    }
 }