From 34fe96a4202745d1edfc63b0124785a43bb4efed Mon Sep 17 00:00:00 2001
From: James Rasell <jrasell@users.noreply.github.com>
Date: Fri, 26 Jan 2024 09:09:16 +0100
Subject: [PATCH] driver/java: populate OOM killed exit result. (#19818)

---
 .changelog/19818.txt   | 3 +++
 drivers/java/driver.go | 5 +++--
 2 files changed, 6 insertions(+), 2 deletions(-)
 create mode 100644 .changelog/19818.txt

diff --git a/.changelog/19818.txt b/.changelog/19818.txt
new file mode 100644
index 000000000..28d447a1e
--- /dev/null
+++ b/.changelog/19818.txt
@@ -0,0 +1,3 @@
+```release-note:bug
+driver/java: Ensure the OOM killed response is populated when the task exits
+```
diff --git a/drivers/java/driver.go b/drivers/java/driver.go
index f5be46ff1..acaeae651 100644
--- a/drivers/java/driver.go
+++ b/drivers/java/driver.go
@@ -593,8 +593,9 @@ func (d *Driver) handleWait(ctx context.Context, handle *taskHandle, ch chan *dr
 		}
 	} else {
 		result = &drivers.ExitResult{
-			ExitCode: ps.ExitCode,
-			Signal:   ps.Signal,
+			ExitCode:  ps.ExitCode,
+			Signal:    ps.Signal,
+			OOMKilled: ps.OOMKilled,
 		}
 	}