mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package tests
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/url"
|
|
//"os"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
|
|
apiutils "github.com/fnproject/fn/test/fn-api-tests"
|
|
"github.com/fnproject/fn_go/models"
|
|
)
|
|
|
|
func LB() (string, error) {
|
|
lbURL := "http://127.0.0.1:8081"
|
|
|
|
u, err := url.Parse(lbURL)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return u.Host, nil
|
|
}
|
|
|
|
func TestCanExecuteFunction(t *testing.T) {
|
|
s := apiutils.SetupHarness()
|
|
s.GivenAppExists(t, &models.App{Name: s.AppName})
|
|
defer s.Cleanup()
|
|
|
|
rt := s.BasicRoute()
|
|
rt.Type = "sync"
|
|
|
|
s.GivenRouteExists(t, s.AppName, rt)
|
|
|
|
lb, err := LB()
|
|
if err != nil {
|
|
t.Fatalf("Got unexpected error: %v", err)
|
|
}
|
|
u := url.URL{
|
|
Scheme: "http",
|
|
Host: lb,
|
|
}
|
|
u.Path = path.Join(u.Path, "r", s.AppName, s.RoutePath)
|
|
|
|
content := &bytes.Buffer{}
|
|
output := &bytes.Buffer{}
|
|
_, err = apiutils.CallFN(u.String(), content, output, "POST", []string{})
|
|
if err != nil {
|
|
t.Errorf("Got unexpected error: %v", err)
|
|
}
|
|
expectedOutput := "Hello World!\n"
|
|
if !strings.Contains(expectedOutput, output.String()) {
|
|
t.Errorf("Assertion error.\n\tExpected: %v\n\tActual: %v", expectedOutput, output.String())
|
|
}
|
|
}
|
|
|
|
func TestBasicConcurrentExecution(t *testing.T) {
|
|
s := apiutils.SetupHarness()
|
|
|
|
s.GivenAppExists(t, &models.App{Name: s.AppName})
|
|
defer s.Cleanup()
|
|
|
|
rt := s.BasicRoute()
|
|
rt.Type = "sync"
|
|
|
|
s.GivenRouteExists(t, s.AppName, rt)
|
|
|
|
lb, err := LB()
|
|
if err != nil {
|
|
t.Fatalf("Got unexpected error: %v", err)
|
|
}
|
|
u := url.URL{
|
|
Scheme: "http",
|
|
Host: lb,
|
|
}
|
|
u.Path = path.Join(u.Path, "r", s.AppName, s.RoutePath)
|
|
|
|
results := make(chan error)
|
|
concurrentFuncs := 10
|
|
for i := 0; i < concurrentFuncs; i++ {
|
|
go func() {
|
|
content := &bytes.Buffer{}
|
|
output := &bytes.Buffer{}
|
|
_, err = apiutils.CallFN(u.String(), content, output, "POST", []string{})
|
|
if err != nil {
|
|
results <- fmt.Errorf("Got unexpected error: %v", err)
|
|
return
|
|
}
|
|
expectedOutput := "Hello World!\n"
|
|
if !strings.Contains(expectedOutput, output.String()) {
|
|
results <- fmt.Errorf("Assertion error.\n\tExpected: %v\n\tActual: %v", expectedOutput, output.String())
|
|
return
|
|
}
|
|
results <- nil
|
|
}()
|
|
}
|
|
for i := 0; i < concurrentFuncs; i++ {
|
|
err := <-results
|
|
if err != nil {
|
|
t.Errorf("Error in basic concurrency execution test: %v", err)
|
|
}
|
|
}
|
|
|
|
}
|