Merge pull request #10 from reu98/fix/add-test

fix: remove recover and add testing

captcha_test.go

@@ -1,6 +1,7 @@
 package captcha
 
 import (
+	"image/color"
 	"math/rand"
 	"strconv"
 	"testing"
@@ -60,3 +61,14 @@ func TestCreateByMathByOption(t *testing.T) {
 	require.GreaterOrEqual(t, result, int(minMath)-int(maxMath))
 	require.LessOrEqual(t, result, int(maxMath)*2)
 }
+
+func TestCreateCaptcha(t *testing.T) {
+	opt := getOptionByText(OptionText{
+		BackgroundColor: color.Black,
+	})
+	text := opt.randomText()
+	capt, err := opt.createCaptcha(text)
+
+	require.NotEmpty(t, capt)
+	require.NoError(t, err)
+}

draw.go

@@ -2,19 +2,13 @@ package captcha
 
 import (
 	"fmt"
-	"log"
 	"math/rand"
 
 	"github.com/reu98/go-svg-captcha/fonts"
 	"github.com/tdewolff/canvas"
 )
 
 func (opt *option) drawText(text string) (string, error) {
-	defer func() {
-		if err := recover(); err != nil {
-			log.Fatalln(err)
-		}
-	}()
 
 	fillColorMin := fillColorMinDefault
 	fillColorMax := fillColorMaxDefault

draw_test.go

@@ -45,3 +45,22 @@ func TestDrawTextWithoutDefault(t *testing.T) {
 	require.NoError(t, err)
 	require.NotEmpty(t, data)
 }
+
+func TestDrawTextWithError(t *testing.T) {
+	opt := getOptionByText(OptionText{
+		FontPath: "font.ttf",
+	})
+	text := opt.randomText()
+	data, err := opt.drawText(text)
+
+	require.Empty(t, data)
+	require.Error(t, err)
+}
+
+func TestLoadFontWithFontPathInvalid(t *testing.T) {
+	fontPath := "font.ttf"
+	font, err := loadFont(fontPath)
+
+	require.Nil(t, font)
+	require.Error(t, err)
+}

option_test.go

@@ -40,3 +40,50 @@ func TestGetOptionByMath(t *testing.T) {
 	require.Equal(t, data.mathMax, mathMaxDefault)
 	require.Contains(t, []mathOperator{MathOperatorMinus, MathOperatorPlus}, data.mathOperator)
 }
+
+func TestGetOptionByTextWithOption(t *testing.T) {
+	option := OptionText{
+		Size:             6,
+		Width:            200,
+		Height:           100,
+		FontSize:         18,
+		Curve:            3,
+		IgnoreCharacters: "90",
+		CharactersPreset: "1234567890",
+		BackgroundColor:  color.Black,
+	}
+	opt := getOptionByText(option)
+
+	require.Equal(t, opt.size, uint8(6))
+	require.Equal(t, opt.width, uint16(200))
+	require.Equal(t, opt.height, uint16(100))
+	require.Equal(t, opt.fontSize, uint(18)*uint(ratioFontSize))
+	require.Equal(t, opt.curve, uint8(3))
+	require.Equal(t, opt.ignoreCharacters, "90")
+	require.Equal(t, opt.charactersPreset, "1234567890")
+	require.True(t, opt.isColor)
+	require.Equal(t, opt.backgroundColor, color.Black)
+}
+
+func TestGetOptionByMathWithOption(t *testing.T) {
+	option := OptionMath{
+		Width:           200,
+		Height:          100,
+		FontSize:        18,
+		Curve:           3,
+		BackgroundColor: color.Black,
+		MathOperator:    MathOperatorPlus,
+		MathMin:         10,
+		MathMax:         99,
+	}
+	opt := getOptionByMath(option)
+
+	require.Equal(t, opt.width, uint16(200))
+	require.Equal(t, opt.height, uint16(100))
+	require.Equal(t, opt.fontSize, uint(18)*uint(ratioFontSize))
+	require.True(t, opt.isColor)
+	require.Equal(t, opt.backgroundColor, color.Black)
+	require.Equal(t, opt.mathMin, uint8(10))
+	require.Equal(t, opt.mathMax, uint16(99))
+	require.Equal(t, opt.mathOperator, MathOperatorPlus)
+}