storj/satellite/mailservice/simulate/linkclicker_test.go

// Copyright (C) 2021 Storj Labs, Inc.
// See LICENSE for copying information.

package simulate_test

import (
	"testing"

	"github.com/stretchr/testify/require"
	"go.uber.org/zap/zaptest"

	"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.NewDefaultLinkClicker(zaptest.NewLogger(t))
	require.ElementsMatch(t, []string{"link1", "link2"}, clicker.FindLinks(data))
}