alihan/fn-serverless
1
0
Fork 0
mirror of https://github.com/fnproject/fn.git synced 2022-10-28 21:29:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

mock datastore tests (#562)

Browse Source
This commit is contained in:
Jordan Krage
2017-03-02 00:04:04 -06:00
committed by Travis Reeder
parent 947fedb194
commit 17e18d872b
5 changed files with 97 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
api/datastore/mock.go
View File

@@ -9,6 +9,7 @@ import (
type Mock struct {
Apps []*models.App
Routes []*models.Route
data map[string][]byte
}
func NewMock(apps []*models.App, routes []*models.Route) *Mock {
@@ -18,10 +19,13 @@ func NewMock(apps []*models.App, routes []*models.Route) *Mock {
if routes == nil {
routes = []*models.Route{}
}
return &Mock{apps, routes}
return &Mock{apps, routes, make(map[string][]byte)}
}
func (m *Mock) GetApp(ctx context.Context, appName string) (app *models.App, err error) {
if appName == "" {
return nil, models.ErrDatastoreEmptyAppName
}
for _, a := range m.Apps {
if a.Name == appName {
return a, nil
@@ -36,6 +40,13 @@ func (m *Mock) GetApps(ctx context.Context, appFilter *models.AppFilter) ([]*mod
}
func (m *Mock) InsertApp(ctx context.Context, app *models.App) (*models.App, error) {
if app == nil {
return nil, models.ErrDatastoreEmptyApp
}
if app.Name == "" {
return nil, models.ErrDatastoreEmptyAppName
}
if a, _ := m.GetApp(ctx, app.Name); a != nil {
return nil, models.ErrAppsAlreadyExists
}
@@ -48,18 +59,15 @@ func (m *Mock) UpdateApp(ctx context.Context, app *models.App) (*models.App, err
if err != nil {
return nil, err
}
if app.Config != nil {
if a.Config == nil {
a.Config = map[string]string{}
}
for k, v := range app.Config {
a.Config[k] = v
}
}
return a, nil
a.UpdateConfig(app.Config)
return a.Clone(), nil
}
func (m *Mock) RemoveApp(ctx context.Context, appName string) error {
if appName == "" {
return models.ErrDatastoreEmptyAppName
}
for i, a := range m.Apps {
if a.Name == appName {
m.Apps = append(m.Apps[:i], m.Apps[i+1:]...)
@@ -70,6 +78,12 @@ func (m *Mock) RemoveApp(ctx context.Context, appName string) error {
}
func (m *Mock) GetRoute(ctx context.Context, appName, routePath string) (*models.Route, error) {
if appName == "" {
return nil, models.ErrDatastoreEmptyAppName
}
if routePath == "" {
return nil, models.ErrDatastoreEmptyRoutePath
}
for _, r := range m.Routes {
if r.AppName == appName && r.Path == routePath {
return r, nil
@@ -95,8 +109,24 @@ func (m *Mock) GetRoutesByApp(ctx context.Context, appName string, routeFilter *
}
func (m *Mock) InsertRoute(ctx context.Context, route *models.Route) (*models.Route, error) {
if route == nil {
return nil, models.ErrDatastoreEmptyRoute
}
if route.AppName == "" {
return nil, models.ErrDatastoreEmptyAppName
}
if route.Path == "" {
return nil, models.ErrDatastoreEmptyRoutePath
}
if _, err := m.GetApp(ctx, route.AppName); err != nil {
return nil, err
}
if r, _ := m.GetRoute(ctx, route.AppName, route.Path); r != nil {
return nil, models.ErrAppsAlreadyExists
return nil, models.ErrRoutesAlreadyExists
}
m.Routes = append(m.Routes, route)
return route, nil
@@ -107,18 +137,17 @@ func (m *Mock) UpdateRoute(ctx context.Context, route *models.Route) (*models.Ro
if err != nil {
return nil, err
}
if route.Config != nil {
if route.Config == nil {
r.Config = map[string]string{}
}
for k, v := range route.Config {
r.Config[k] = v
}
}
return r, nil
r.Update(route)
return r.Clone(), nil
}
func (m *Mock) RemoveRoute(ctx context.Context, appName, routePath string) error {
if appName == "" {
return models.ErrDatastoreEmptyAppName
}
if routePath == "" {
return models.ErrDatastoreEmptyRoutePath
}
for i, r := range m.Routes {
if r.AppName == appName && r.Path == routePath {
m.Routes = append(m.Routes[:i], m.Routes[i+1:]...)
@@ -129,11 +158,20 @@ func (m *Mock) RemoveRoute(ctx context.Context, appName, routePath string) error
}
func (m *Mock) Put(ctx context.Context, key, value []byte) error {
// TODO: improve this mock method
if len(key) == 0 {
return models.ErrDatastoreEmptyKey
}
if len(value) == 0 {
delete(m.data, string(key))
} else {
m.data[string(key)] = value
}
return nil
}
func (m *Mock) Get(ctx context.Context, key []byte) ([]byte, error) {
// TODO: improve this mock method
return []byte{}, nil
if len(key) == 0 {
return nil, models.ErrDatastoreEmptyKey
}
return m.data[string(key)], nil
}