From 083279b4d3776512fa57a51c9a41e5557cef5776 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C4=B0nan=C3=A7=20G=C3=BCm=C3=BC=C5=9F?=
 <inanc.gumus@grafana.com>
Date: Wed, 5 Jun 2024 17:50:49 +0300
Subject: [PATCH] Add Page.GetAttribute tests

---
 tests/page_test.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/tests/page_test.go b/tests/page_test.go
index 50a97f904..1fae03428 100644
--- a/tests/page_test.go
+++ b/tests/page_test.go
@@ -1806,3 +1806,42 @@ func TestPageTargetBlank(t *testing.T) {
 	require.NoError(t, err)
 	assert.Equal(t, "you clicked!", got)
 }
+
+func TestPageGetAttribute(t *testing.T) {
+	t.Parallel()
+
+	p := newTestBrowser(t).NewPage(nil)
+	err := p.SetContent(`<a id="el" href="null">Something</a>`, nil)
+	require.NoError(t, err)
+
+	got, ok, err := p.GetAttribute("#el", "href", nil)
+	require.NoError(t, err)
+	require.True(t, ok)
+	assert.Equal(t, "null", got)
+}
+
+func TestPageGetAttributeMissing(t *testing.T) {
+	t.Parallel()
+
+	p := newTestBrowser(t).NewPage(nil)
+	err := p.SetContent(`<a id="el">Something</a>`, nil)
+	require.NoError(t, err)
+
+	got, ok, err := p.GetAttribute("#el", "missing", nil)
+	require.NoError(t, err)
+	require.False(t, ok)
+	assert.Equal(t, "", got)
+}
+
+func TestPageGetAttributeEmpty(t *testing.T) {
+	t.Parallel()
+
+	p := newTestBrowser(t).NewPage(nil)
+	err := p.SetContent(`<a id="el" empty>Something</a>`, nil)
+	require.NoError(t, err)
+
+	got, ok, err := p.GetAttribute("#el", "empty", nil)
+	require.NoError(t, err)
+	require.True(t, ok)
+	assert.Equal(t, "", got)
+}