Add enforce nonce to message builder (#89)

dc-Ragnarok · Apr 23, 2024 · 25855f8 · 25855f8
1 parent 46ff7aa
commit 25855f8
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/src/Rest/Helpers/Channel/MessageBuilder.php b/src/Rest/Helpers/Channel/MessageBuilder.php
@@ -48,6 +48,18 @@ public function getNonce(): ?string
         return $this->data['nonce'] ?? null;
     }
 
+    public function setEnforceNonce(bool $enforceNonce)
+    {
+        $this->data['enforce_nonce'] = $enforceNonce;
+
+        return $this;
+    }
+
+    public function getEnforceNonce(): ?bool
+    {
+        return $this->data['enforce_nonce'] ?? null;
+    }
+
     public function setTts(bool $tts): self
     {
         $this->data['tts'] = $tts;

diff --git a/tests/Command/GuildCommandExtensionTest.php b/tests/Command/GuildCommandExtensionTest.php
@@ -27,7 +27,7 @@ protected function setUp(): void
 
     public function testItEmitsEventsForApplicationCommands()
     {
-        $extension = new GuildCommandExtension(null, '::guild id::');
+        $extension = new GuildCommandExtension('::guild id::');
         $extension->initialize($this->discord);
 
         $hasRun = [false, false, false];
@@ -69,7 +69,7 @@ public function testItEmitsEventsForApplicationCommands()
 
     public function testItDoesNotEmitEventsForDifferentGuilds()
     {
-        $extension = new GuildCommandExtension(null, '::guild id::');
+        $extension = new GuildCommandExtension('::guild id::');
         $extension->initialize($this->discord);
 
         $hasRun = false;
@@ -98,7 +98,7 @@ public function testItDoesNotEmitEventsForDifferentGuilds()
      */
     public function testItMapsNamesCorrectly(InteractionCreate $interaction, string $expectedName)
     {
-        $extension = new GuildCommandExtension(null, '::guild id::');
+        $extension = new GuildCommandExtension('::guild id::');
         $extension->initialize($this->discord);
 
         $hasRun = false;

diff --git a/tests/Rest/Helpers/Channel/MessageBuilderTest.php b/tests/Rest/Helpers/Channel/MessageBuilderTest.php
@@ -25,6 +25,18 @@ public function testSetNonce(): void
         $this->assertEquals('::nonce::', $builder->getNonce());
     }
 
+    public function testSetEnforceNonce(): void
+    {
+        $builder = new MessageBuilder();
+
+        $this->assertNull($builder->getEnforceNonce());
+
+        $builder->setEnforceNonce(true);
+
+        $this->assertTrue($builder->get()['enforce_nonce']);
+        $this->assertTrue($builder->getEnforceNonce());
+    }
+
     public function testSetTts(): void
     {
         $builder = new MessageBuilder();