Yuri Tseretyan
d15e1ad8d0
* introduce ExtraConfigsCrypto * delete getLastConfiguration and move all code into method Get * update legacy store to encrypt ExtraConfigs same way we do in MultiOrgAlertmanager * update legacy store init * delete PersistConfig * remove unnecessary unmarshalling * add better error * add tests for legacy store * add tests that cover extra config validation * add integration test for conflicting routes --------- Signed-off-by: Yuri Tseretyan <yuriy.tseretyan@grafana.com>
99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
package legacy_storage
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/grafana/grafana/pkg/services/ngalert/api/tooling/definitions"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
)
|
|
|
|
func (m *MockAMConfigStore_Expecter) GetsConfig(ac models.AlertConfiguration) *MockAMConfigStore_Expecter {
|
|
m.GetLatestAlertmanagerConfiguration(mock.Anything, mock.Anything).Return(&ac, nil)
|
|
return m
|
|
}
|
|
|
|
func (m *MockAMConfigStore_Expecter) SaveSucceeds() *MockAMConfigStore_Expecter {
|
|
m.UpdateAlertmanagerConfiguration(mock.Anything, mock.Anything).Return(nil)
|
|
return m
|
|
}
|
|
|
|
func (m *MockAMConfigStore_Expecter) SaveSucceedsIntercept(intercepted *models.SaveAlertmanagerConfigurationCmd) *MockAMConfigStore_Expecter {
|
|
m.UpdateAlertmanagerConfiguration(mock.Anything, mock.Anything).
|
|
Return(nil).
|
|
Run(func(ctx context.Context, cmd *models.SaveAlertmanagerConfigurationCmd) {
|
|
*intercepted = *cmd
|
|
})
|
|
return m
|
|
}
|
|
|
|
type methodCall struct {
|
|
Method string
|
|
Args []interface{}
|
|
}
|
|
|
|
type AlertmanagerConfigStoreFake struct {
|
|
Calls []methodCall
|
|
GetFn func(ctx context.Context, orgID int64) (*ConfigRevision, error)
|
|
SaveFn func(ctx context.Context, revision *ConfigRevision) error
|
|
}
|
|
|
|
func (a *AlertmanagerConfigStoreFake) Get(ctx context.Context, orgID int64) (*ConfigRevision, error) {
|
|
a.Calls = append(a.Calls, methodCall{
|
|
Method: "Get",
|
|
Args: []interface{}{ctx, orgID},
|
|
})
|
|
if a.GetFn != nil {
|
|
return a.GetFn(ctx, orgID)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (a *AlertmanagerConfigStoreFake) Save(ctx context.Context, revision *ConfigRevision, orgID int64) error {
|
|
a.Calls = append(a.Calls, methodCall{
|
|
Method: "Save",
|
|
Args: []interface{}{ctx, revision, orgID},
|
|
})
|
|
if a.SaveFn != nil {
|
|
return a.SaveFn(ctx, revision)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type fakeCrypto struct {
|
|
Calls []methodCall
|
|
EncryptExtraConfigsFunc func(context.Context, *definitions.PostableUserConfig) error
|
|
DecryptExtraConfigsFunc func(context.Context, *definitions.PostableUserConfig) error
|
|
}
|
|
|
|
func newFakeCrypto() *fakeCrypto {
|
|
return &fakeCrypto{
|
|
Calls: []methodCall{},
|
|
}
|
|
}
|
|
|
|
func (f *fakeCrypto) EncryptExtraConfigs(ctx context.Context, config *definitions.PostableUserConfig) error {
|
|
f.Calls = append(f.Calls, methodCall{
|
|
Method: "EncryptExtraConfigs",
|
|
Args: []interface{}{ctx, config},
|
|
})
|
|
|
|
if f.EncryptExtraConfigsFunc != nil {
|
|
return f.EncryptExtraConfigsFunc(ctx, config)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f *fakeCrypto) DecryptExtraConfigs(ctx context.Context, config *definitions.PostableUserConfig) error {
|
|
f.Calls = append(f.Calls, methodCall{
|
|
Method: "DecryptExtraConfigs",
|
|
Args: []interface{}{ctx, config},
|
|
})
|
|
|
|
if f.DecryptExtraConfigsFunc != nil {
|
|
return f.DecryptExtraConfigsFunc(ctx, config)
|
|
}
|
|
return nil
|
|
}
|