addAttribute method.

src/WPDesk/Notice/Notice.php

@@ -167,6 +167,9 @@ class Notice
         }
     }
 
+    /**
+     * Remove action.
+     */
     protected function removeAction()
     {
         if ($this->actionAdded) {
@@ -180,6 +183,17 @@ class Notice
         }
     }
 
+    /**
+     * Add attribute.
+     *
+     * @param string $name Name
+     * @param string $value Value.
+     */
+    public function addAttribute($name, $value)
+    {
+        $this->attributes[ $name ] = $value;
+    }
+
     /**
      * Get notice class.
      *

tests/integration/TestNotice.php

@@ -118,4 +118,26 @@ class TestNotice extends WP_UnitTestCase
         $this->assertEquals(20, $notice->getPriority());
     }
 
+    public function testAddAttribute()
+    {
+        $notice = new Notice('test', Notice::NOTICE_TYPE_WARNING);
+
+        $notice->addAttribute('id', 'test_id');
+
+        $this->expectOutputString('<div class="notice notice-warning" id="test_id"><p>test</p></div>');
+
+        $notice->showNotice();
+    }
+
+    public function testAddAttributeClass()
+    {
+        $notice = new Notice('test', Notice::NOTICE_TYPE_WARNING);
+
+        $notice->addAttribute('class', 'test-class');
+
+        $this->expectOutputString('<div class="notice notice-warning test-class"><p>test</p></div>');
+
+        $notice->showNotice();
+    }
+
 }