chore: handle nil steps and session

nndi-oss · Jul 12, 2024 · 547059d · 547059d
1 parent c2cd20c
commit 547059d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/pkg/africastalking/input.go b/pkg/africastalking/input.go
@@ -7,6 +7,10 @@ import (
 )
 
 func ConcatText(session *core.Session) string {
+	if session == nil {
+		return ""
+	}
+
 	inputs := make([]string, 0)
 	for _, step := range session.Steps {
 		inputs = append(inputs, step.Text)

diff --git a/pkg/africastalking/input_test.go b/pkg/africastalking/input_test.go
@@ -8,6 +8,19 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestConcatTextWithNilSession(t *testing.T) {
+	got := africastalking.ConcatText(nil)
+	assert.Equal(t, "", got)
+}
+
+func TestConcatTextWithNilSteps(t *testing.T) {
+	got := africastalking.ConcatText(&core.Session{
+		Steps: nil,
+	})
+
+	assert.Equal(t, "", got)
+}
+
 func TestConcatText(t *testing.T) {
 	got := africastalking.ConcatText(&core.Session{
 		Steps: []*core.Step{