route/vendor/github.com/calmh/luhn/luhn_test.go

78 lines
1.5 KiB
Go
Raw Normal View History

2017-10-26 15:46:27 +00:00
// Copyright (C) 2014 Jakob Borg
package luhn_test
import (
"testing"
"github.com/calmh/luhn"
)
func TestGenerate(t *testing.T) {
// Base 6 Luhn
a := luhn.Alphabet("abcdef")
c, err := a.Generate("abcdef")
if err != nil {
t.Fatal(err)
}
if c != 'e' {
t.Errorf("Incorrect check digit %c != e", c)
}
// Base 10 Luhn
a = luhn.Alphabet("0123456789")
c, err = a.Generate("7992739871")
if err != nil {
t.Fatal(err)
}
if c != '3' {
t.Errorf("Incorrect check digit %c != 3", c)
}
}
func TestInvalidString(t *testing.T) {
a := luhn.Alphabet("ABC")
_, err := a.Generate("7992739871")
t.Log(err)
if err == nil {
t.Error("Unexpected nil error")
}
}
func TestBadAlphabet(t *testing.T) {
_, err := luhn.NewAlphabet("01234566789")
t.Log(err)
if err == nil {
t.Error("Unexpected nil error")
}
}
func TestValidate(t *testing.T) {
a := luhn.Alphabet("abcdef")
if !a.Validate("abcdefe") {
t.Errorf("Incorrect validation response for abcdefe")
}
if a.Validate("abcdefd") {
t.Errorf("Incorrect validation response for abcdefd")
}
}
func TestValidateRosetta(t *testing.T) {
// http://rosettacode.org/wiki/Luhn_test_of_credit_card_numbers
a := luhn.Alphabet("0123456789")
cases := []struct {
v string
ok bool
}{
{"49927398716", true},
{"49927398717", false},
{"1234567812345678", false},
{"1234567812345670", true},
}
for _, tc := range cases {
if res := a.Validate(tc.v); res != tc.ok {
t.Errorf("Validate(%q) => %v, expected %v", tc.v, res, tc.ok)
}
}
}