mirror of
https://github.com/juanfont/headscale.git
synced 2024-12-02 03:33:05 +00:00
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)
|
|
}
|