package documenttext

import (
	"testing"

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

func TestHasCheckboxKeywords(t *testing.T) {
	svc := Service{}
	assert.False(t, svc.hasCheckboxKeywords(""))
	assert.False(t, svc.hasCheckboxKeywords("aaaaaa"))
	assert.False(t, svc.hasCheckboxKeywords("benefit plan"))
	assert.False(t, svc.hasCheckboxKeywords("benefit plans"))
	assert.True(t, svc.hasCheckboxKeywords("exhibit benefit plans"))
}

func TestHasExhibit(t *testing.T) {
	svc := Service{}
	assert.False(t, svc.hasExhibit(""))
	assert.False(t, svc.hasExhibit("aaaaaa"))
	assert.True(t, svc.hasExhibit("exhibit"))
}

func TestHasBenefit(t *testing.T) {
	svc := Service{}
	assert.False(t, svc.hasBenefit(""))
	assert.False(t, svc.hasBenefit("aaaaaa"))
	assert.False(t, svc.hasBenefit("benefit"))
	assert.True(t, svc.hasBenefit("benefit plan"))
	assert.True(t, svc.hasBenefit("benefit plan(s)"))
	assert.True(t, svc.hasBenefit("applicable benefit"))
	assert.True(t, svc.hasBenefit("benefits"))
}

func TestHasPlans(t *testing.T) {
	svc := Service{}
	assert.False(t, svc.hasPlans(""))
	assert.False(t, svc.hasPlans("aaaaaa"))
	assert.True(t, svc.hasPlans("plan(s):"))
	assert.True(t, svc.hasPlans("plan(s)/program(s):"))
	assert.True(t, svc.hasPlans("plans:"))
	assert.True(t, svc.hasPlans("plan/program:"))
	assert.True(t, svc.hasPlans("plan(s)"))
	assert.True(t, svc.hasPlans("plans"))
}

func TestHasServices(t *testing.T) {
	svc := Service{}
	assert.False(t, svc.hasServices(""))
	assert.False(t, svc.hasServices("aaaaaa"))
	assert.True(t, svc.hasServices("services:"))
	assert.True(t, svc.hasServices("service:"))
	assert.True(t, svc.hasServices("service(s):"))
	assert.True(t, svc.hasServices("services"))
	assert.True(t, svc.hasServices("service(s)"))
}