storj/satellite/mailservice/simulate/linkclicker_test.go

29 lines
649 B
Go
Raw Normal View History

// Copyright (C) 2021 Storj Labs, Inc.
// See LICENSE for copying information.
package simulate_test
import (
"testing"
"github.com/stretchr/testify/require"
"storj.io/storj/satellite/mailservice/simulate"
)
func TestFindLinks(t *testing.T) {
data := `
<a href="link1" data-simulate>
<A HREF="link2" data-simulate>
<a href="link3">
<a href>
<a data-simulate>
`
clicker := simulate.LinkClicker{MarkerAttribute: "data-simulate"}
require.ElementsMatch(t, []string{"link1", "link2"}, clicker.FindLinks(data))
clicker.MarkerAttribute = ""
require.ElementsMatch(t, []string{"link1", "link2", "link3"}, clicker.FindLinks(data))
}