captcha_test.go

// Copyright 2011 Dmitry Chestnykh. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package captcha

import (
	"bytes"
	"testing"
)

func TestNew(t *testing.T) {
	c := New()
	if c == "" {
		t.Errorf("expected id, got empty string")
	}
}

func TestVerify(t *testing.T) {
	id := New()
	if Verify(id, []byte{0, 0}) {
		t.Errorf("verified wrong captcha")
	}
	id = New()
	d, max, _ := globalStore.Get(id, false) // cheating
	if !Verify(id, d) {
		t.Errorf("proper captcha not verified")
	}
	if max != DefaultMaxCheckCnt {
		t.Errorf("最大验证次数不等于 %d", DefaultMaxCheckCnt)
	}
}
func TestVerifyBind(t *testing.T) {
	id := NewLenCheckCntBind(6, 5, "123")
	if Verify(id, []byte{0, 0}) {
		t.Errorf("verified wrong captcha")
	}
	id = New()
	d, max, bind := globalStore.Get(id, false) // cheating
	if !Verify(id, d, bind) {
		t.Errorf("proper captcha not verified")
	}
	if max != DefaultMaxCheckCnt {
		t.Errorf("最大验证次数不等于 %d", DefaultMaxCheckCnt)
	}
}
func TestReload(t *testing.T) {
	id := New()
	d1, _, _ := globalStore.Get(id, false) // cheating
	Reload(id)
	d2, _, _ := globalStore.Get(id, false) // cheating again
	if bytes.Equal(d1, d2) {
		t.Errorf("reload didn't work: %v = %v", d1, d2)
	}
}

func TestTryToReload(t *testing.T) {
	id := NewLenCheckCnt(4, 1)
	d1, _, _ := globalStore.Get(id, false) // cheating
	TryToReload(id)
	d2, _, _ := globalStore.Get(id, false) // cheating again
	if !bytes.Equal(d1, d2) {
		t.Errorf("reload didn't work: %v = %v", d1, d2)
	}
	ok := Verify(id, d1)
	if !ok {
		t.Error("Verify want: true act: false")
	}
	d2, _, _ = globalStore.Get(id, false)
	if !bytes.Equal(d1, d2) {
		t.Errorf("reload didn't work: %v = %v", d1, d2)
	}
}

func TestRandomDigits(t *testing.T) {
	d1 := RandomDigits(10)
	for _, v := range d1 {
		if v > 9 {
			t.Errorf("digits not in range 0-9: %v", d1)
		}
	}
	d2 := RandomDigits(10)
	if bytes.Equal(d1, d2) {
		t.Errorf("digits seem to be not random")
	}
}

func Test_string2bytes(t *testing.T) {
	dig := string2bytes("12345")
	if bytes2string(dig) != "12345" {
		t.Errorf("转换失败")
	}
}