package smaato
import (
"github.com/stretchr/testify/assert"
"testing"
)
func TestExtractAdmBanner(t *testing.T) {
tests := []struct {
testName string
adMarkup string
curls []string
expectedAdMarkup string
}{
{
testName: "extract_banner_without_curls",
adMarkup: `
`,
expectedAdMarkup: ``,
curls: []string{},
},
{
testName: "extract_banner_with_nil_curls",
adMarkup: ``,
expectedAdMarkup: ``,
curls: nil,
},
{
testName: "extract_banner_with_curls",
adMarkup: ``,
expectedAdMarkup: ``,
curls: []string{"curls.net"},
},
}
for _, tt := range tests {
t.Run(tt.testName, func(t *testing.T) {
adMarkup := extractAdmBanner(tt.adMarkup, tt.curls)
assert.Equal(t, tt.expectedAdMarkup, adMarkup)
})
}
}