mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
a0ccc4d7c4
Copies the log endpoints up to the V2 endpoints, in a similar way to the call endpoints. The main change is to when logs are inserted into S3. The signature of the function has been changed to take the whole call object, rather than just the app and call id's. This allows the function to switch between calls for Routes and those for Fns. Obviously this switching can be removed when v1 is removed. In the sql implementation it inserts with both appID and fnID, this allows the two get's to work, and the down grade of the migration. When the v1 logs are removed, the appId can be dropped. The log fetch test and error messages have been changed to be FnID specific.
222 lines
7.0 KiB
Go
222 lines
7.0 KiB
Go
package testing
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/base64"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fnproject/fn/api/common"
|
|
"github.com/fnproject/fn/api/id"
|
|
"github.com/fnproject/fn/api/models"
|
|
)
|
|
|
|
var testApp = &models.App{
|
|
Name: "Test",
|
|
ID: id.New().String(),
|
|
}
|
|
|
|
var testFn = &models.Fn{
|
|
ID: id.New().String(),
|
|
Image: "fnproject/fn-test-utils",
|
|
Format: "http",
|
|
}
|
|
|
|
func SetupTestCall(t *testing.T, ctx context.Context, ls models.LogStore) *models.Call {
|
|
var call models.Call
|
|
call.AppID = testApp.ID
|
|
call.CreatedAt = common.DateTime(time.Now())
|
|
call.Status = "success"
|
|
call.StartedAt = common.DateTime(time.Now())
|
|
call.CompletedAt = common.DateTime(time.Now())
|
|
call.FnID = testFn.ID
|
|
return &call
|
|
}
|
|
|
|
const tmpLogDb = "/tmp/func_test_log.db"
|
|
|
|
func Test(t *testing.T, fnl models.LogStore) {
|
|
ctx := context.Background()
|
|
call := SetupTestCall(t, ctx, fnl)
|
|
|
|
// test list first, the rest are point lookup tests
|
|
t.Run("calls-get", func(t *testing.T) {
|
|
filter := &models.CallFilter{FnID: call.FnID, PerPage: 100}
|
|
now := time.Now()
|
|
call.CreatedAt = common.DateTime(now)
|
|
call.ID = id.New().String()
|
|
err := fnl.InsertCall(ctx, call)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
calls, err := fnl.GetCalls(ctx, filter)
|
|
if err != nil {
|
|
t.Fatalf("Test GetCalls2(ctx, filter): one call, unexpected error `%v`", err)
|
|
}
|
|
if len(calls.Items) != 1 {
|
|
t.Fatalf("Test GetCalls2(ctx, filter): one call, unexpected length 1 != `%v`", len(calls.Items))
|
|
}
|
|
|
|
c2 := *call
|
|
c3 := *call
|
|
now = time.Now().Add(100 * time.Millisecond)
|
|
c2.CreatedAt = common.DateTime(now) // add ms cuz db uses it for sort
|
|
c2.ID = id.New().String()
|
|
|
|
now = time.Now().Add(200 * time.Millisecond)
|
|
c3.CreatedAt = common.DateTime(now)
|
|
c3.ID = id.New().String()
|
|
|
|
err = fnl.InsertCall(ctx, &c2)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = fnl.InsertCall(ctx, &c3)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// test that no filter works too
|
|
calls, err = fnl.GetCalls(ctx, &models.CallFilter{FnID: testFn.ID, PerPage: 100})
|
|
if err != nil {
|
|
t.Fatalf("Test GetCalls2(ctx, filter): three calls, unexpected error `%v`", err)
|
|
}
|
|
if len(calls.Items) != 3 {
|
|
t.Fatalf("Test GetCalls2(ctx, filter): three calls, unexpected length 3 != `%v`", len(calls.Items))
|
|
}
|
|
|
|
// test that pagination stuff works. id, descending
|
|
filter.PerPage = 1
|
|
calls, err = fnl.GetCalls(ctx, filter)
|
|
if err != nil {
|
|
t.Fatalf("Test GetCalls(ctx, filter): per page 1, unexpected error `%v`", err)
|
|
}
|
|
if len(calls.Items) != 1 {
|
|
t.Fatalf("Test GetCalls(ctx, filter): per page 1, unexpected length 1 != `%v`", len(calls.Items))
|
|
} else if calls.Items[0].ID != c3.ID {
|
|
t.Fatalf("Test GetCalls: per page 1, call ids not in expected order: %v %v", calls.Items[0].ID, c3.ID)
|
|
}
|
|
|
|
filter.PerPage = 100
|
|
filter.Cursor = base64.RawURLEncoding.EncodeToString([]byte(calls.Items[0].ID))
|
|
calls, err = fnl.GetCalls(ctx, filter)
|
|
if err != nil {
|
|
t.Fatalf("Test GetCalls(ctx, filter): cursor set, unexpected error `%v`", err)
|
|
}
|
|
if len(calls.Items) != 2 {
|
|
t.Fatalf("Test GetCalls(ctx, filter): cursor set, unexpected length 2 != `%v`", len(calls.Items))
|
|
} else if calls.Items[0].ID != c2.ID {
|
|
t.Fatalf("Test GetCalls: cursor set, call ids not in expected order: %v %v", calls.Items[0].ID, c2.ID)
|
|
} else if calls.Items[1].ID != call.ID {
|
|
t.Fatalf("Test GetCalls: cursor set, call ids not in expected order: %v %v", calls.Items[1].ID, call.ID)
|
|
}
|
|
|
|
// test that filters actually applied
|
|
calls, err = fnl.GetCalls(ctx, &models.CallFilter{FnID: "wrongfnID", PerPage: 100})
|
|
if err != nil {
|
|
t.Fatalf("Test GetCalls(ctx, filter): bad fnID, unexpected error `%v`", err)
|
|
}
|
|
if len(calls.Items) != 0 {
|
|
t.Fatalf("Test GetCalls(ctx, filter): bad fnID, unexpected length `%v`", len(calls.Items))
|
|
}
|
|
|
|
// make sure from_time and to_time work
|
|
filter = &models.CallFilter{
|
|
PerPage: 100,
|
|
FromTime: call.CreatedAt,
|
|
ToTime: c3.CreatedAt,
|
|
FnID: call.FnID,
|
|
}
|
|
calls, err = fnl.GetCalls(ctx, filter)
|
|
if err != nil {
|
|
t.Fatalf("Test GetCalls(ctx, filter): time filter, unexpected error `%v`", err)
|
|
}
|
|
if len(calls.Items) != 1 {
|
|
t.Fatalf("Test GetCalls(ctx, filter): time filter, unexpected length `%v`", len(calls.Items))
|
|
} else if calls.Items[0].ID != c2.ID {
|
|
t.Fatalf("Test GetCalls: time filter, call id not expected %s vs %s", calls.Items[0].ID, c2.ID)
|
|
}
|
|
})
|
|
|
|
t.Run("call-log-insert-get", func(t *testing.T) {
|
|
call.ID = id.New().String()
|
|
logText := "test"
|
|
log := strings.NewReader(logText)
|
|
err := fnl.InsertLog(ctx, call, log)
|
|
if err != nil {
|
|
t.Fatalf("Test InsertLog(ctx, call.ID, logText): unexpected error during inserting log `%v`", err)
|
|
}
|
|
logEntry, err := fnl.GetLog(ctx, call.FnID, call.ID)
|
|
var b bytes.Buffer
|
|
io.Copy(&b, logEntry)
|
|
if !strings.Contains(b.String(), logText) {
|
|
t.Fatalf("Test GetLog(ctx, call.ID, logText): unexpected error, log mismatch. "+
|
|
"Expected: `%v`. Got `%v`.", logText, b.String())
|
|
}
|
|
})
|
|
|
|
t.Run("call-log-not-found", func(t *testing.T) {
|
|
call.ID = id.New().String()
|
|
_, err := fnl.GetLog(ctx, call.FnID, call.ID)
|
|
if err != models.ErrCallLogNotFound {
|
|
t.Fatal("GetLog should return not found, but got:", err)
|
|
}
|
|
})
|
|
|
|
call = new(models.Call)
|
|
call.CreatedAt = common.DateTime(time.Now())
|
|
call.Status = "error"
|
|
call.Error = "ya dun goofed"
|
|
call.StartedAt = common.DateTime(time.Now())
|
|
call.CompletedAt = common.DateTime(time.Now())
|
|
call.AppID = testApp.ID
|
|
call.FnID = testFn.ID
|
|
|
|
t.Run("call-insert", func(t *testing.T) {
|
|
call.ID = id.New().String()
|
|
err := fnl.InsertCall(ctx, call)
|
|
if err != nil {
|
|
t.Fatalf("Test InsertCall(ctx, &call): unexpected error `%v`", err)
|
|
}
|
|
})
|
|
|
|
t.Run("call-get", func(t *testing.T) {
|
|
call.ID = id.New().String()
|
|
err := fnl.InsertCall(ctx, call)
|
|
if err != nil {
|
|
t.Fatalf("Test GetCall: unexpected error `%v`", err)
|
|
}
|
|
newCall, err := fnl.GetCall(ctx, call.FnID, call.ID)
|
|
if err != nil {
|
|
t.Fatalf("Test GetCall: unexpected error `%v`", err)
|
|
}
|
|
if call.ID != newCall.ID {
|
|
t.Fatalf("Test GetCall: id mismatch `%v` `%v`", call.ID, newCall.ID)
|
|
}
|
|
if call.Status != newCall.Status {
|
|
t.Fatalf("Test GetCall: status mismatch `%v` `%v`", call.Status, newCall.Status)
|
|
}
|
|
if call.Error != newCall.Error {
|
|
t.Fatalf("Test GetCall: error mismatch `%v` `%v`", call.Error, newCall.Error)
|
|
}
|
|
if time.Time(call.CreatedAt).Unix() != time.Time(newCall.CreatedAt).Unix() {
|
|
t.Fatalf("Test GetCall: created_at mismatch `%v` `%v`", call.CreatedAt, newCall.CreatedAt)
|
|
}
|
|
if time.Time(call.StartedAt).Unix() != time.Time(newCall.StartedAt).Unix() {
|
|
t.Fatalf("Test GetCall: started_at mismatch `%v` `%v`", call.StartedAt, newCall.StartedAt)
|
|
}
|
|
if time.Time(call.CompletedAt).Unix() != time.Time(newCall.CompletedAt).Unix() {
|
|
t.Fatalf("Test GetCall: completed_at mismatch `%v` `%v`", call.CompletedAt, newCall.CompletedAt)
|
|
}
|
|
if call.AppID != newCall.AppID {
|
|
t.Fatalf("Test GetCall: fn id mismatch `%v` `%v`", call.FnID, newCall.FnID)
|
|
}
|
|
if call.Path != newCall.Path {
|
|
t.Fatalf("Test GetCall: path mismatch `%v` `%v`", call.Path, newCall.Path)
|
|
}
|
|
})
|
|
}
|