package service

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNew(t *testing.T) {
	s := New()
	usernameInput1 := `Hello <STYLE>.XSS{background-image:url("javascript:alert('XSS')");}</STYLE><A CLASS=XSS></A>World`
	usernameInput2 := "Hello World"

	assert.Equal(t, "Hello World", s.SanitizerPolicy().Sanitize(usernameInput1))
	assert.Equal(t, "Hello World", s.SanitizerPolicy().Sanitize(usernameInput2))
}

func TestIsValidEmail(t *testing.T) {
	email1 := "test@pingcap.com"
	email2 := "test"
	email3 := "test@c0m"
	email4 := "a@snapmail.cc"

	assert.True(t, IsValidEmail(email1))
	assert.False(t, IsValidEmail(email2))
	assert.False(t, IsValidEmail(email3))
	assert.True(t, IsValidEmail(email4))
}