29 lines
649 B
Go
29 lines
649 B
Go
|
// 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))
|
||
|
}
|