DeforaNetworks/corteza
3
0
Fork 0
Code Activity

Fix integration tests

Browse Source 
- Extends helper for integration test with token generation
This commit is contained in:
Vivek Patel
2021-09-24 19:24:00 +05:30
parent ac587db589
commit a9c762aacc
5 changed files with 40 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/automation/main_test.go
+8 -6
View File
@@ -35,6 +35,7 @@ type (
cUser *sysTypes.User
roleID uint64
token string
}
)
@@ -97,6 +98,12 @@ func newHelper(t *testing.T) helper {
},
}
var err error
h.token, err = auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
h.cUser.SetRoles(h.roleID)
helpers.UpdateRBAC(h.roleID)
@@ -116,15 +123,10 @@ func (h helper) secCtx() context.Context {
func (h helper) apiInit() *apitest.APITest {
InitTestApp()
tkn, err := auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
return apitest.
New().
Handler(r).
Intercept(helpers.ReqHeaderRawAuthBearer(tkn))
Intercept(helpers.ReqHeaderRawAuthBearer(h.token))
}
tests/compose/main_test.go
+8 -6
View File
@@ -42,6 +42,7 @@ type (
cUser *sysTypes.User
roleID uint64
token string
}
)
@@ -108,6 +109,12 @@ func newHelper(t *testing.T) helper {
},
}
var err error
h.token, err = auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
h.cUser.SetRoles(h.roleID)
helpers.UpdateRBAC(h.roleID)
@@ -127,15 +134,10 @@ func (h helper) secCtx() context.Context {
func (h helper) apiInit() *apitest.APITest {
InitTestApp()
tkn, err := auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
return apitest.
New().
Handler(r).
Intercept(helpers.ReqHeaderRawAuthBearer(tkn))
Intercept(helpers.ReqHeaderRawAuthBearer(h.token))
}
func (h helper) mockPermissions(rules ...*rbac.Rule) {
tests/federation/main_test.go
+8 -6
View File
@@ -30,6 +30,7 @@ type (
cUser *sysTypes.User
roleID uint64
token string
}
)
@@ -82,6 +83,12 @@ func newHelper(t *testing.T) helper {
},
}
var err error
h.token, err = auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
h.cUser.SetRoles(h.roleID)
helpers.UpdateRBAC(h.roleID)
@@ -101,15 +108,10 @@ func (h helper) secCtx() context.Context {
func (h helper) apiInit() *apitest.APITest {
InitTestApp()
tkn, err := auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
return apitest.
New().
Handler(r).
Intercept(helpers.ReqHeaderRawAuthBearer(tkn))
Intercept(helpers.ReqHeaderRawAuthBearer(h.token))
}
// Unwraps error before it passes it to the tester
tests/reporter/main_test.go
+8 -6
View File
@@ -47,6 +47,7 @@ type (
cUser *sysTypes.User
roleID uint64
token string
}
auxReport struct {
@@ -120,6 +121,12 @@ func newHelper(t *testing.T) helper {
},
}
var err error
h.token, err = auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
h.cUser.SetRoles(h.roleID)
helpers.UpdateRBAC(h.roleID)
@@ -139,15 +146,10 @@ func (h helper) secCtx() context.Context {
func (h helper) apiInit() *apitest.APITest {
InitTestApp()
tkn, err := auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
return apitest.
New().
Handler(r).
Intercept(helpers.ReqHeaderRawAuthBearer(tkn))
Intercept(helpers.ReqHeaderRawAuthBearer(h.token))
}
func (h helper) mockPermissions(rules ...*rbac.Rule) {
h.noError(rbac.Global().Grant(
tests/system/main_test.go
+8 -6
View File
@@ -47,6 +47,7 @@ type (
cUser *types.User
roleID uint64
token string
data embed.FS
}
)
@@ -130,6 +131,12 @@ func newHelper(t *testing.T) helper {
data: mockData,
}
var err error
h.token, err = auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
h.cUser.SetRoles(h.roleID)
helpers.UpdateRBAC(h.roleID)
h.mockPermissionsWithAccess()
@@ -150,15 +157,10 @@ func (h helper) secCtx() context.Context {
func (h helper) apiInit() *apitest.APITest {
InitTestApp()
tkn, err := auth.DefaultJwtHandler.Generate(context.Background(), h.cUser)
if err != nil {
panic(err)
}
return apitest.
New().
Handler(r).
Intercept(helpers.ReqHeaderRawAuthBearer(tkn))
Intercept(helpers.ReqHeaderRawAuthBearer(h.token))
}
func (h helper) mockPermissions(rules ...*rbac.Rule) {