TheArchive/headscale
1
0
Fork 0
mirror of https://github.com/juanfont/headscale.git synced 2025-08-25 09:37:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP Working on authkeys + tests

Browse Source
This commit is contained in:
Juan Font Alonso
2021-05-05 23:00:04 +02:00
parent 03bb32083b
commit 486faa9656
3 changed files with 50 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
preauth_keys_test.go
View File

@@ -5,6 +5,7 @@ import (
"io/ioutil"
"os"
"testing"
"time"
_ "github.com/jinzhu/gorm/dialects/sqlite" // sql driver
@@ -48,6 +49,7 @@ func (s *Suite) TearDownSuite(c *check.C) {
func (*Suite) TestCreatePreAuthKey(c *check.C) {
_, err := h.CreatePreAuthKey("bogus", true, nil)
c.Assert(err, check.NotNil)
n, err := h.CreateNamespace("test")
@@ -73,3 +75,22 @@ func (*Suite) TestCreatePreAuthKey(c *check.C) {
// Make sure the Namespace association is populated
c.Assert((*keys)[0].Namespace.Name, check.Equals, n.Name)
}
func (*Suite) TestExpiredPreAuthKey(c *check.C) {
n, err := h.CreateNamespace("test2")
c.Assert(err, check.IsNil)
now := time.Now()
pak, err := h.CreatePreAuthKey(n.Name, true, &now)
c.Assert(err, check.IsNil)
p, err := h.checkKeyValidity(pak.Key)
c.Assert(err, check.Equals, errorAuthKeyExpired)
c.Assert(p, check.IsNil)
}
func (*Suite) TestPreAuthKeyDoesNotExist(c *check.C) {
p, err := h.checkKeyValidity("potatoKey")
c.Assert(err, check.Equals, errorAuthKeyNotFound)
c.Assert(p, check.IsNil)
}