From 1f9eb578be3ee912c04d8aec05295faf8259d64d Mon Sep 17 00:00:00 2001
From: Steve Boyd <emteknetnz@gmail.com>
Date: Mon, 12 Aug 2024 13:27:36 +1200
Subject: [PATCH] MNT Add extension in finally block

---
 .../Controllers/LinkFieldControllerTest.php   | 57 ++++++++++---------
 1 file changed, 30 insertions(+), 27 deletions(-)

diff --git a/tests/php/Controllers/LinkFieldControllerTest.php b/tests/php/Controllers/LinkFieldControllerTest.php
index a4b0691a..24a9b759 100644
--- a/tests/php/Controllers/LinkFieldControllerTest.php
+++ b/tests/php/Controllers/LinkFieldControllerTest.php
@@ -537,38 +537,41 @@ public function testLinkArchive(
         int $expectedCode
     ): void {
         TestPhoneLink::$fail = $fail;
-        if ($fail == 'can-delete') {
+        if ($fail === 'can-delete') {
             // Remove the Versioned extension because there's logic in LinkFieldController
             // to use either canArchive() or canDelete() based on the presence of the Versioned extension
             // Note that you must remove the extension on the base class rather than a TestOnly subclass
             Link::remove_extension(Versioned::class);
         }
-        $owner = $this->getFixtureLinkOwner();
-        $ownerID = $owner->ID;
-        $ownerClass = urlencode($owner->ClassName);
-        $ownerRelation = 'Link';
-        $ownerLinkID = $owner->LinkID;
-        $id = $this->getID($idType);
-        $fixtureID = $this->getFixtureLink()->ID;
-        if ($id === -1) {
-            $url = "/admin/linkfield/delete?ownerID=$ownerID&ownerClass=$ownerClass&ownerRelation=$ownerRelation";
-        } else {
-            $url = "/admin/linkfield/delete/$id?ownerID=$ownerID&ownerClass=$ownerClass&ownerRelation=$ownerRelation";
-        }
-        $headers = [];
-        if ($fail !== 'csrf-token') {
-            $headers = array_merge($headers, $this->csrfTokenheader());
-        }
-        $response = $this->mainSession->sendRequest('DELETE', $url, [], $headers);
-        $this->assertSame('application/json', $response->getHeader('Content-type'));
-        $this->assertSame($expectedCode, $response->getStatusCode());
-        if ($expectedCode >= 400) {
-            $this->assertNotNull(TestPhoneLink::get()->byID($fixtureID));
-        } else {
-            $this->assertNull(TestPhoneLink::get()->byID($fixtureID));
-        }
-        if ($fail == 'can-delete') {
-            Link::add_extension(Versioned::class);
+        try {
+            $owner = $this->getFixtureLinkOwner();
+            $ownerID = $owner->ID;
+            $ownerClass = urlencode($owner->ClassName);
+            $ownerRelation = 'Link';
+            $ownerLinkID = $owner->LinkID;
+            $id = $this->getID($idType);
+            $fixtureID = $this->getFixtureLink()->ID;
+            if ($id === -1) {
+                $url = "/admin/linkfield/delete?ownerID=$ownerID&ownerClass=$ownerClass&ownerRelation=$ownerRelation";
+            } else {
+                $url = "/admin/linkfield/delete/$id?ownerID=$ownerID&ownerClass=$ownerClass&ownerRelation=$ownerRelation";
+            }
+            $headers = [];
+            if ($fail !== 'csrf-token') {
+                $headers = array_merge($headers, $this->csrfTokenheader());
+            }
+            $response = $this->mainSession->sendRequest('DELETE', $url, [], $headers);
+            $this->assertSame('application/json', $response->getHeader('Content-type'));
+            $this->assertSame($expectedCode, $response->getStatusCode());
+            if ($expectedCode >= 400) {
+                $this->assertNotNull(TestPhoneLink::get()->byID($fixtureID));
+            } else {
+                $this->assertNull(TestPhoneLink::get()->byID($fixtureID));
+            }
+        } finally {
+            if ($fail === 'can-delete') {
+                Link::add_extension(Versioned::class);
+            }
         }
     }