Add test for getHxDepth

ekalinin · Nov 8, 2022 · 563f232 · 563f232
1 parent d29c184
commit 563f232
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/headerfinder_test.go b/headerfinder_test.go
@@ -5,6 +5,7 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	"golang.org/x/net/html"
+	"golang.org/x/net/html/atom"
 )
 
 const singleH1 = `
@@ -98,3 +99,13 @@ func TestFindAttribute(t *testing.T) {
 		assert.False(t, ok)
 	})
 }
+
+func TestGetHxDepth(t *testing.T) {
+	assert.Equal(t, HxDepth(0), getHxDepth(atom.H1))
+	assert.Equal(t, HxDepth(1), getHxDepth(atom.H2))
+	assert.Equal(t, HxDepth(2), getHxDepth(atom.H3))
+	assert.Equal(t, HxDepth(3), getHxDepth(atom.H4))
+	assert.Equal(t, HxDepth(4), getHxDepth(atom.H5))
+	assert.Equal(t, HxDepth(5), getHxDepth(atom.H6))
+	assert.Equal(t, InvalidDepth, getHxDepth(atom.A))
+}