segmenter_test.go

package segmenter_test

import (
	"testing"

	"github.com/riandyrn/go-segmenter"
	"github.com/stretchr/testify/require"
)

func TestNew(t *testing.T) {
	testCases := []struct {
		Name        string
		Config      segmenter.Config[string]
		ExpHasError bool
	}{
		{
			Name: "Valid Config",
			Config: segmenter.Config[string]{
				Slice:         []string{"1", "2", "3"},
				SegmentLength: 5,
			},
			ExpHasError: false,
		},
		{
			Name: "Empty Items",
			Config: segmenter.Config[string]{
				Slice:         nil,
				SegmentLength: 10,
			},
			ExpHasError: true,
		},
		{
			Name: "No Segment Length",
			Config: segmenter.Config[string]{
				Slice: []string{"1", "2", "3"},
			},
			ExpHasError: true,
		},
		{
			Name: "Negative Segment Length",
			Config: segmenter.Config[string]{
				Slice:         []string{"1", "2", "3"},
				SegmentLength: -5,
			},
			ExpHasError: true,
		},
	}
	for _, testCase := range testCases {
		t.Run(testCase.Name, func(t *testing.T) {
			_, err := segmenter.New(testCase.Config)
			require.Equal(t, testCase.ExpHasError, err != nil)
		})
	}
}

func TestNext(t *testing.T) {
	testCases := []struct {
		Name          string
		Slice         []string
		SegmentLength int
		ExpResults    [][]string
	}{
		{
			Name:          "Segment Length Evenly Divisible With List Length",
			Slice:         []string{"0", "1", "2", "3", "4", "5", "6", "7", "8", "9"},
			SegmentLength: 5,
			ExpResults: [][]string{
				{"0", "1", "2", "3", "4"},
				{"5", "6", "7", "8", "9"},
			},
		},
		{
			Name:          "Segment Length Not Evenly Divisible With List Length",
			Slice:         []string{"0", "1", "2", "3", "4", "5", "6", "7", "8", "9"},
			SegmentLength: 3,
			ExpResults: [][]string{
				{"0", "1", "2"},
				{"3", "4", "5"},
				{"6", "7", "8"},
				{"9"},
			},
		},
		{
			Name:          "Segment Length Larger Than List Length",
			Slice:         []string{"0", "1", "2", "3"},
			SegmentLength: 5,
			ExpResults: [][]string{
				{"0", "1", "2", "3"},
			},
		},
	}
	for _, testCase := range testCases {
		t.Run(testCase.Name, func(t *testing.T) {
			// initialize segmenter
			sgmntr, err := segmenter.New(segmenter.Config[string]{
				Slice:         testCase.Slice,
				SegmentLength: testCase.SegmentLength,
			})
			require.NoError(t, err)
			// collect all results
			var results [][]string
			for sgmntr.HasNext() {
				results = append(results, sgmntr.Next())
			}
			// compare collected results
			require.Equal(t, testCase.ExpResults, results)
		})
	}
}