Move mock handlers to mock and implement proxy.

Signed-off-by: Will Piers <wpiers0405@gmail.com>
pull/10616/head
Chris Goller 2016-09-12 18:30:03 -05:00
parent eb870302ca
commit 5dbfce783c
3 changed files with 27 additions and 28 deletions

View File

@ -1,4 +1,4 @@
package handlers
package mock
import (
"strconv"
@ -6,25 +6,35 @@ import (
"github.com/go-openapi/runtime/middleware"
"github.com/go-openapi/strfmt"
"github.com/influxdata/mrfusion"
"github.com/influxdata/mrfusion/mock"
"github.com/influxdata/mrfusion/models"
op "github.com/influxdata/mrfusion/restapi/operations"
"golang.org/x/net/context"
)
type MockHandler struct {
type Handler struct {
Store mrfusion.ExplorationStore
Response mrfusion.Response
}
func NewMockHandler() MockHandler {
return MockHandler{
mock.DefaultExplorationStore,
mock.SampleResponse,
func NewHandler() Handler {
return Handler{
DefaultExplorationStore,
SampleResponse,
}
}
func (m *MockHandler) Explorations(ctx context.Context, params op.GetSourcesIDUsersUserIDExplorationsParams) middleware.Responder {
func (m *Handler) Proxy(ctx context.Context, params op.PostSourcesIDProxyParams) middleware.Responder {
results, err := m.Response.Results()
if err != nil {
return op.NewPostSourcesIDProxyDefault(500)
}
res := &models.ProxyResponse{
Results: results,
}
return op.NewPostSourcesIDProxyOK().WithPayload(res)
}
func (m *Handler) Explorations(ctx context.Context, params op.GetSourcesIDUsersUserIDExplorationsParams) middleware.Responder {
id, err := strconv.Atoi(params.UserID)
if err != nil {
return op.NewGetSourcesIDUsersUserIDExplorationsDefault(500)
@ -46,7 +56,7 @@ func (m *MockHandler) Explorations(ctx context.Context, params op.GetSourcesIDUs
return op.NewGetSourcesIDUsersUserIDExplorationsOK().WithPayload(res)
}
func (m *MockHandler) Exploration(ctx context.Context, params op.GetSourcesIDUsersUserIDExplorationsExplorationIDParams) middleware.Responder {
func (m *Handler) Exploration(ctx context.Context, params op.GetSourcesIDUsersUserIDExplorationsExplorationIDParams) middleware.Responder {
eID, err := strconv.Atoi(params.ExplorationID)
if err != nil {
return op.NewGetSourcesIDUsersUserIDExplorationsExplorationIDDefault(500)
@ -65,7 +75,7 @@ func (m *MockHandler) Exploration(ctx context.Context, params op.GetSourcesIDUse
return op.NewGetSourcesIDUsersUserIDExplorationsExplorationIDOK().WithPayload(res)
}
func (m *MockHandler) UpdateExploration(ctx context.Context, params op.PatchSourcesIDUsersUserIDExplorationsExplorationIDParams) middleware.Responder {
func (m *Handler) UpdateExploration(ctx context.Context, params op.PatchSourcesIDUsersUserIDExplorationsExplorationIDParams) middleware.Responder {
eID, err := strconv.Atoi(params.ExplorationID)
if err != nil {
return op.NewPatchSourcesIDUsersUserIDExplorationsExplorationIDDefault(500)
@ -83,7 +93,7 @@ func (m *MockHandler) UpdateExploration(ctx context.Context, params op.PatchSour
return op.NewPatchSourcesIDUsersUserIDExplorationsExplorationIDNoContent()
}
func (m *MockHandler) NewExploration(ctx context.Context, params op.PostSourcesIDUsersUserIDExplorationsParams) middleware.Responder {
func (m *Handler) NewExploration(ctx context.Context, params op.PostSourcesIDUsersUserIDExplorationsParams) middleware.Responder {
id, err := strconv.Atoi(params.UserID)
if err != nil {
return op.NewPostSourcesIDUsersUserIDExplorationsDefault(500)
@ -106,7 +116,7 @@ func (m *MockHandler) NewExploration(ctx context.Context, params op.PostSourcesI
return op.NewPostSourcesIDUsersUserIDExplorationsCreated()
}
func (m *MockHandler) DeleteExploration(ctx context.Context, params op.DeleteSourcesIDUsersUserIDExplorationsExplorationIDParams) middleware.Responder {
func (m *Handler) DeleteExploration(ctx context.Context, params op.DeleteSourcesIDUsersUserIDExplorationsExplorationIDParams) middleware.Responder {
ID, err := strconv.Atoi(params.ExplorationID)
if err != nil {
return op.NewDeleteSourcesIDUsersUserIDExplorationsExplorationIDDefault(500)

View File

@ -9,7 +9,7 @@ import (
middleware "github.com/go-openapi/runtime/middleware"
"golang.org/x/net/context"
"github.com/influxdata/mrfusion/restapi/handlers"
"github.com/influxdata/mrfusion/mock"
"github.com/influxdata/mrfusion/restapi/operations"
)
@ -35,7 +35,7 @@ func configureAPI(api *operations.MrFusionAPI) http.Handler {
api.JSONProducer = runtime.JSONProducer()
mockHandler := handlers.NewMockHandler()
mockHandler := mock.NewHandler()
api.DeleteDashboardsIDHandler = operations.DeleteDashboardsIDHandlerFunc(func(ctx context.Context, params operations.DeleteDashboardsIDParams) middleware.Responder {
return middleware.NotImplemented("operation .DeleteDashboardsID has not yet been implemented")
@ -105,9 +105,9 @@ func configureAPI(api *operations.MrFusionAPI) http.Handler {
api.PostSourcesHandler = operations.PostSourcesHandlerFunc(func(ctx context.Context, params operations.PostSourcesParams) middleware.Responder {
return middleware.NotImplemented("operation .PostSources has not yet been implemented")
})
api.PostSourcesIDProxyHandler = operations.PostSourcesIDProxyHandlerFunc(func(ctx context.Context, params operations.PostSourcesIDProxyParams) middleware.Responder {
return middleware.NotImplemented("operation .PostSourcesIDProxy has not yet been implemented")
})
api.PostSourcesIDProxyHandler = operations.PostSourcesIDProxyHandlerFunc(mockHandler.Proxy)
api.PostSourcesIDRolesHandler = operations.PostSourcesIDRolesHandlerFunc(func(ctx context.Context, params operations.PostSourcesIDRolesParams) middleware.Responder {
return middleware.NotImplemented("operation .PostSourcesIDRoles has not yet been implemented")
})

View File

@ -1,11 +0,0 @@
package handlers
import (
"github.com/go-openapi/runtime/middleware"
op "github.com/influxdata/mrfusion/restapi/operations"
"golang.org/x/net/context"
)
func MockProxy(ctx context.Context, params op.PostSourcesIDProxyParams) middleware.Responder {
return nil
}