78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
|
// 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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|