mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
d56a49b321
Largely a removal job, however many tests, particularly system level ones relied on Routes. These have been migrated to use Fns. * Add 410 response to swagger * No app names in log tags * Adding constraint in GetCall for FnID * Adding test to check FnID is required on call * Add fn_id to call selector * Fix text in docker mem warning * Correct buildConfig func name * Test fix up * Removing CPU setting from Agent test CPU setting has been deprecated, but the code base is still riddled with it. This just removes it from this layer. Really we need to remove it from Call. * Remove fn id check on calls * Reintroduce fn id required on call * Adding fnID to calls for execute test * Correct setting of app id in middleware * Removes root middlewares ability to redirect fun invocations * Add over sized test check * Removing call fn id check
116 lines
2.7 KiB
Go
116 lines
2.7 KiB
Go
package logs
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/base64"
|
|
"io"
|
|
"io/ioutil"
|
|
"sort"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/fnproject/fn/api/models"
|
|
)
|
|
|
|
type mock struct {
|
|
Logs map[string][]byte
|
|
Calls []*models.Call
|
|
}
|
|
|
|
func NewMock(args ...interface{}) models.LogStore {
|
|
var mocker mock
|
|
for _, a := range args {
|
|
switch x := a.(type) {
|
|
case []*models.Call:
|
|
mocker.Calls = x
|
|
default:
|
|
panic("unknown type handed to mocker. add me")
|
|
}
|
|
}
|
|
mocker.Logs = make(map[string][]byte)
|
|
return &mocker
|
|
}
|
|
|
|
func (m *mock) InsertLog(ctx context.Context, call *models.Call, callLog io.Reader) error {
|
|
bytes, err := ioutil.ReadAll(callLog)
|
|
m.Logs[call.ID] = bytes
|
|
return err
|
|
}
|
|
|
|
func (m *mock) GetLog(ctx context.Context, fnID, callID string) (io.Reader, error) {
|
|
logEntry, ok := m.Logs[callID]
|
|
if !ok {
|
|
return nil, models.ErrCallLogNotFound
|
|
}
|
|
return bytes.NewReader(logEntry), nil
|
|
}
|
|
|
|
func (m *mock) InsertCall(ctx context.Context, call *models.Call) error {
|
|
m.Calls = append(m.Calls, call)
|
|
return nil
|
|
}
|
|
|
|
func (m *mock) GetCall(ctx context.Context, fnID, callID string) (*models.Call, error) {
|
|
for _, t := range m.Calls {
|
|
if t.ID == callID &&
|
|
t.FnID == fnID {
|
|
return t, nil
|
|
}
|
|
}
|
|
|
|
return nil, models.ErrCallNotFound
|
|
}
|
|
|
|
type sortC []*models.Call
|
|
|
|
func (s sortC) Len() int { return len(s) }
|
|
func (s sortC) Less(i, j int) bool { return strings.Compare(s[i].ID, s[j].ID) < 0 }
|
|
func (s sortC) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
|
|
|
|
func (m *mock) GetCalls(ctx context.Context, filter *models.CallFilter) (*models.CallList, error) {
|
|
// sort them all first for cursoring (this is for testing, n is small & mock is not concurrent..)
|
|
// calls are in DESC order so use sort.Reverse
|
|
sort.Sort(sort.Reverse(sortC(m.Calls)))
|
|
|
|
var calls []*models.Call
|
|
|
|
var cursor = ""
|
|
if filter.Cursor != "" {
|
|
s, err := base64.RawURLEncoding.DecodeString(filter.Cursor)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cursor = string(s)
|
|
}
|
|
|
|
for _, c := range m.Calls {
|
|
if filter.PerPage > 0 && len(calls) == filter.PerPage {
|
|
break
|
|
}
|
|
|
|
if (cursor == "" || strings.Compare(cursor, c.ID) > 0) &&
|
|
(filter.FnID == "" || c.FnID == filter.FnID) &&
|
|
(time.Time(filter.FromTime).IsZero() || time.Time(filter.FromTime).Before(time.Time(c.CreatedAt))) &&
|
|
(time.Time(filter.ToTime).IsZero() || time.Time(c.CreatedAt).Before(time.Time(filter.ToTime))) {
|
|
|
|
calls = append(calls, c)
|
|
}
|
|
}
|
|
|
|
var nextCursor string
|
|
if len(calls) > 0 && len(calls) == filter.PerPage {
|
|
last := []byte(calls[len(calls)-1].ID)
|
|
nextCursor = base64.RawURLEncoding.EncodeToString(last)
|
|
}
|
|
|
|
return &models.CallList{
|
|
NextCursor: nextCursor,
|
|
Items: calls,
|
|
}, nil
|
|
}
|
|
|
|
func (m *mock) Close() error {
|
|
return nil
|
|
}
|