31 lines
519 B
Go
31 lines
519 B
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type TestMail struct{}
|
|
|
|
func (m TestMail) SendCode(to string, code string) error {
|
|
return nil
|
|
}
|
|
|
|
var testMailer = TestMail{}
|
|
|
|
func TestCreateCode(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
auth := CreateAuth(testMailer)
|
|
|
|
err := auth.CreateCode("test")
|
|
require.NoError(err)
|
|
|
|
code, exists := auth.codes["test"]
|
|
require.True(exists)
|
|
require.True(code.Valid.After(time.Now()))
|
|
require.True(auth.IsCodeValid("test", code.Code))
|
|
}
|