From 78b6268f9b4acee98e79be4269117858a10cadcf Mon Sep 17 00:00:00 2001
From: Elias Naur <mail@eliasnaur.com>
Date: Mon, 20 Sep 2021 13:03:29 +0200
Subject: [PATCH] gpu/headless: plug resource leaks in tests

Signed-off-by: Elias Naur <mail@eliasnaur.com>
---
 gpu/headless/driver_test.go | 1 +
 gpu/headless/headless.go    | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/gpu/headless/driver_test.go b/gpu/headless/driver_test.go
index a7b1f94b..eb403f77 100644
--- a/gpu/headless/driver_test.go
+++ b/gpu/headless/driver_test.go
@@ -212,6 +212,7 @@ func newDriver(t *testing.T) driver.Device {
 	b.BeginFrame(nil, true, image.Pt(1, 1))
 	t.Cleanup(func() {
 		b.EndFrame()
+		b.Release()
 		ctx.ReleaseCurrent()
 		runtime.UnlockOSThread()
 		ctx.Release()
diff --git a/gpu/headless/headless.go b/gpu/headless/headless.go
index 5870bf06..d93c4d20 100644
--- a/gpu/headless/headless.go
+++ b/gpu/headless/headless.go
@@ -66,6 +66,7 @@ func NewWindow(width, height int) (*Window, error) {
 		if err != nil {
 			fbo.Release()
 			fboTex.Release()
+			dev.Release()
 			return err
 		}
 		w.fboTex = fboTex
@@ -96,6 +97,10 @@ func (w *Window) Release() {
 			w.gpu.Release()
 			w.gpu = nil
 		}
+		if w.dev != nil {
+			w.dev.Release()
+			w.dev = nil
+		}
 		return nil
 	})
 	if w.ctx != nil {