kit/pkg/base64Captcha/driver_digit_test.go

// Copyright 2017 Eric Zhou. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package base64Captcha

import (
	"reflect"
	"testing"
)

func TestDriverDigit_DrawCaptcha(t *testing.T) {
	type fields struct {
		Height     int
		Width      int
		CaptchaLen int
		MaxSkew    float64
		DotCount   int
	}
	type args struct {
		content string
	}
	tests := []struct {
		name     string
		fields   fields
		args     args
		wantItem Item
		wantErr  bool
	}{
		{"Digit", fields{80, 240, 6, 0.6, 8}, args{RandText(6, TxtNumbers)}, nil, false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			d := &DriverDigit{
				Height:   tt.fields.Height,
				Width:    tt.fields.Width,
				Length:   tt.fields.CaptchaLen,
				MaxSkew:  tt.fields.MaxSkew,
				DotCount: tt.fields.DotCount,
			}
			gotItem, err := d.DrawCaptcha(tt.args.content)
			if (err != nil) != tt.wantErr {
				t.Errorf("DriverDigit.DrawCaptcha() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			itemWriteFile(gotItem, "_builds", tt.args.content, "png")

		})
	}
}

func TestNewDriverDigit(t *testing.T) {
	type args struct {
		height   int
		width    int
		length   int
		maxSkew  float64
		dotCount int
	}
	tests := []struct {
		name string
		args args
		want *DriverDigit
	}{
		// TODO: Add test cases.
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := NewDriverDigit(tt.args.height, tt.args.width, tt.args.length, tt.args.maxSkew, tt.args.dotCount); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("NewDriverDigit() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestDriverDigit_GenerateIdQuestionAnswer(t *testing.T) {
	tests := []struct {
		name   string
		d      *DriverDigit
		wantId string
		wantQ  string
		wantA  string
	}{
		// TODO: Add test cases.
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotId, gotQ, gotA := tt.d.GenerateIdQuestionAnswer()
			if gotId != tt.wantId {
				t.Errorf("DriverDigit.GenerateIdQuestionAnswer() gotId = %v, want %v", gotId, tt.wantId)
			}
			if gotQ != tt.wantQ {
				t.Errorf("DriverDigit.GenerateIdQuestionAnswer() gotQ = %v, want %v", gotQ, tt.wantQ)
			}
			if gotA != tt.wantA {
				t.Errorf("DriverDigit.GenerateIdQuestionAnswer() gotA = %v, want %v", gotA, tt.wantA)
			}
		})
	}
}