go-spectre/spectre_test.go

111 lines
2.4 KiB
Go

package spectre_test
import (
_ "embed"
"encoding/xml"
"fmt"
"strconv"
"testing"
"go.jolheiser.com/go-spectre"
)
func TestSpectre(t *testing.T) {
var tests TestCases
if err := xml.Unmarshal(testsXML, &tests); err != nil {
t.Log("could not load test data")
t.FailNow()
}
dc := tests.Cases[0]
for _, tc := range tests.Cases[1:] {
t.Run(tc.ID, func(t *testing.T) {
user := def(dc.UserName, tc.UserName)
secret := def(dc.UserSecret, tc.UserSecret)
s, err := spectre.New(user, secret)
if err != nil {
t.Logf("could not initialize spectre: %v", err)
t.Fail()
}
siteName := def(dc.SiteName, tc.SiteName)
template := def(dc.ResultType, tc.ResultType)
counterStr := def(dc.KeyCounter, tc.KeyCounter)
counter, err := strconv.Atoi(counterStr)
if err != nil {
t.Log("could not convert counter")
t.Fail()
}
scope := def(dc.KeyPurpose, tc.KeyPurpose)
pass := s.Site(siteName,
spectre.WithTemplate(spectre.Template(template)),
spectre.WithCounter(counter),
spectre.WithScope(spectre.Scope(scope)),
)
if pass != tc.Result {
t.Log("passwords did not match")
t.Fail()
}
})
}
}
// From the website sanity check
func Example() {
s, err := spectre.New("Robert Lee Mitchell", "banana colored duckling")
if err != nil {
panic(err)
}
pw := s.Site("masterpasswordapp.com")
fmt.Println(pw)
// Output: Jejr5[RepuSosp
}
// Example with options
func Example_second() {
scoper := spectre.SimpleScoper{
Key: "com.jojodev.jolheiser",
}
s, err := spectre.New("Robert Lee Mitchell", "banana colored duckling", spectre.WithScoper(scoper))
if err != nil {
panic(err)
}
pw := s.Site("jojodev.com",
spectre.WithScope(spectre.Identification),
spectre.WithTemplate(spectre.Maximum),
spectre.WithCounter(2), // Password was leaked, so increment counter (example)
)
fmt.Println(pw)
// Output: Ig^JIcxD!*)TbefJBi6-
}
//go:embed spectre_tests.xml
var testsXML []byte
type TestCases struct {
Cases []TestCase `xml:"case"`
}
type TestCase struct {
ID string `xml:"id,attr"`
UserName string `xml:"userName"`
UserSecret string `xml:"userSecret"`
SiteName string `xml:"siteName"`
ResultType string `xml:"resultType"`
KeyCounter string `xml:"keyCounter"`
KeyPurpose string `xml:"keyPurpose"`
Result string `xml:"result"`
}
func def(def, alt string) string {
if alt != "" {
return alt
}
return def
}