captcha_test.go

package base64Captcha

import (
	"io/ioutil"
	"strings"
	"testing"
)

const ExampleFontDirPath = "/Users/ericzhou/go/src/github.com/mojocn/base64Captcha/examples/fonts"
const ExampleFontExtension = "ttf"

var configD = ConfigDigit{
	Height:     80,
	Width:      240,
	MaxSkew:    0.7,
	DotCount:   80,
	CaptchaLen: 5,
}

var configA = ConfigAudio{
	CaptchaLen: 6,
	Language:   "zh",
}

var configC = ConfigCharacter{
	Height:             60,
	Width:              240,
	Mode:               0,
	ComplexOfNoiseText: 0,
	ComplexOfNoiseDot:  0,
	IsUseSimpleFont:    false,
	IsShowHollowLine:   false,
	IsShowNoiseDot:     false,
	IsShowNoiseText:    false,
	IsShowSlimeLine:    false,
	IsShowSineLine:     false,
	CaptchaLen:         6,
}

func TestGenerateCaptcha(t *testing.T) {
	GenerateCaptcha("", CaptchaItem{})
	for idx, vv := range []interface{}{configA, configD} {

		idkey, cap := GenerateCaptcha("", vv)
		ext := "png"
		if idx == 0 {
			ext = "wav"
		}
		GoTestOutputDir, _ := ioutil.TempDir("", "")

		CaptchaWriteToFile(cap, GoTestOutputDir, idkey, ext)
		CaptchaWriteToFile(cap, GoTestOutputDir, idkey, ext)

		tempDir, _ := ioutil.TempDir("", "")
		CaptchaWriteToFile(cap, tempDir, idkey, ext)

		//t.Log(idkey, globalStore.Get(idkey, false))

	}

	for i := 0; i < 16; i++ {
		configC.Mode = i % 4
		idkey, cap := GenerateCaptcha("", configC)
		ext := "png"
		err := CaptchaWriteToFile(cap, GoTestOutputDir+"/all", "char_"+idkey, ext)
		if err == nil {
			t.Log(idkey, globalStore.Get(idkey, false))
		} else {
			t.Error(idkey)
		}
	}
}

func TestCaptchaWriteToBase64Encoding(t *testing.T) {
	idkey, cap := GenerateCaptcha("", configD)
	base64string := CaptchaWriteToBase64Encoding(cap)
	if strings.Contains(base64string, MimeTypeCaptchaImage) {
		t.Log(base64string, idkey)
	} else {
		t.Error("encodeing base64 string failed.")
	}
	idkeyA, capA := GenerateCaptcha("", configA)
	base64stringA := CaptchaWriteToBase64Encoding(capA)
	if strings.Contains(base64stringA, MimeTypeCaptchaAudio) {
		t.Log(base64string, idkeyA)
	} else {
		t.Error("encodeing base64 string failed.")
	}

}

func TestVerifyCaptcha(t *testing.T) {
	idkey, _ := GenerateCaptcha("", configD)
	verifyValue := globalStore.Get(idkey, false)
	if VerifyCaptcha(idkey, verifyValue) {
		t.Log(idkey, verifyValue)
	} else {
		t.Error("verify captcha content is failed.")
	}

	VerifyCaptcha("", "")
	VerifyCaptcha("dsafasf", "ddd")

}