14e29a7bee
This is step one in detaching the Database layer from Headscale (h). The ultimate goal is to have all function that does database operations in its own package, and keep the business logic and writing separate. Signed-off-by: Kristoffer Dalby <kristoffer@tailscale.com>
89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package hscontrol
|
|
|
|
import (
|
|
"time"
|
|
|
|
"gopkg.in/check.v1"
|
|
)
|
|
|
|
func (*Suite) TestCreateAPIKey(c *check.C) {
|
|
apiKeyStr, apiKey, err := app.db.CreateAPIKey(nil)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(apiKey, check.NotNil)
|
|
|
|
// Did we get a valid key?
|
|
c.Assert(apiKey.Prefix, check.NotNil)
|
|
c.Assert(apiKey.Hash, check.NotNil)
|
|
c.Assert(apiKeyStr, check.Not(check.Equals), "")
|
|
|
|
_, err = app.db.ListAPIKeys()
|
|
c.Assert(err, check.IsNil)
|
|
|
|
keys, err := app.db.ListAPIKeys()
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(len(keys), check.Equals, 1)
|
|
}
|
|
|
|
func (*Suite) TestAPIKeyDoesNotExist(c *check.C) {
|
|
key, err := app.db.GetAPIKey("does-not-exist")
|
|
c.Assert(err, check.NotNil)
|
|
c.Assert(key, check.IsNil)
|
|
}
|
|
|
|
func (*Suite) TestValidateAPIKeyOk(c *check.C) {
|
|
nowPlus2 := time.Now().Add(2 * time.Hour)
|
|
apiKeyStr, apiKey, err := app.db.CreateAPIKey(&nowPlus2)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(apiKey, check.NotNil)
|
|
|
|
valid, err := app.db.ValidateAPIKey(apiKeyStr)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(valid, check.Equals, true)
|
|
}
|
|
|
|
func (*Suite) TestValidateAPIKeyNotOk(c *check.C) {
|
|
nowMinus2 := time.Now().Add(time.Duration(-2) * time.Hour)
|
|
apiKeyStr, apiKey, err := app.db.CreateAPIKey(&nowMinus2)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(apiKey, check.NotNil)
|
|
|
|
valid, err := app.db.ValidateAPIKey(apiKeyStr)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(valid, check.Equals, false)
|
|
|
|
now := time.Now()
|
|
apiKeyStrNow, apiKey, err := app.db.CreateAPIKey(&now)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(apiKey, check.NotNil)
|
|
|
|
validNow, err := app.db.ValidateAPIKey(apiKeyStrNow)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(validNow, check.Equals, false)
|
|
|
|
validSilly, err := app.db.ValidateAPIKey("nota.validkey")
|
|
c.Assert(err, check.NotNil)
|
|
c.Assert(validSilly, check.Equals, false)
|
|
|
|
validWithErr, err := app.db.ValidateAPIKey("produceerrorkey")
|
|
c.Assert(err, check.NotNil)
|
|
c.Assert(validWithErr, check.Equals, false)
|
|
}
|
|
|
|
func (*Suite) TestExpireAPIKey(c *check.C) {
|
|
nowPlus2 := time.Now().Add(2 * time.Hour)
|
|
apiKeyStr, apiKey, err := app.db.CreateAPIKey(&nowPlus2)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(apiKey, check.NotNil)
|
|
|
|
valid, err := app.db.ValidateAPIKey(apiKeyStr)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(valid, check.Equals, true)
|
|
|
|
err = app.db.ExpireAPIKey(apiKey)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(apiKey.Expiration, check.NotNil)
|
|
|
|
notValid, err := app.db.ValidateAPIKey(apiKeyStr)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(notValid, check.Equals, false)
|
|
}
|