Add Page.GetAttribute tests

grafana · Jun 6, 2024 · 083279b · 083279b
1 parent f7480ec
commit 083279b
Showing 1 changed file with 39 additions and 0 deletions.
diff --git 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)
+}