From 191b09ce5d3efcb96cc7bf8468ab62626ad62a34 Mon Sep 17 00:00:00 2001
From: Ruslan Kashapov <ruslan.kashapov@pantheon.tech>
Date: Mon, 28 Dec 2020 10:14:41 +0200
Subject: 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>
---
 cps-ri/src/test/java/org/onap/cps/DatabaseTestContainer.java | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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();
+    }
 }
-- 
cgit 1.2.3-korg