forked from yuxh/gearman-go
Add skipping for integration tests when running unit tests
Run integration tests with the -integration flag
This commit is contained in:
parent
2dbf199260
commit
99d317427f
@ -1,6 +1,8 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
@ -8,9 +10,24 @@ const (
|
||||
TestStr = "Hello world"
|
||||
)
|
||||
|
||||
var client *Client
|
||||
var (
|
||||
client *Client
|
||||
runIntegrationTests bool
|
||||
)
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
integrationsTestFlag := flag.Bool("integration", false, "Run the integration tests (in addition to the unit tests)")
|
||||
if integrationsTestFlag != nil {
|
||||
runIntegrationTests = *integrationsTestFlag
|
||||
}
|
||||
code := m.Run()
|
||||
os.Exit(code)
|
||||
}
|
||||
|
||||
func TestClientAddServer(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
t.Log("Add local server 127.0.0.1:4730")
|
||||
var err error
|
||||
if client, err = New(Network, "127.0.0.1:4730"); err != nil {
|
||||
@ -22,6 +39,9 @@ func TestClientAddServer(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestClientEcho(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
echo, err := client.Echo([]byte(TestStr))
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
@ -34,6 +54,9 @@ func TestClientEcho(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestClientDoBg(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
handle, err := client.DoBg("ToUpper", []byte("abcdef"), JobLow)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
@ -47,6 +70,9 @@ func TestClientDoBg(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestClientDo(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
jobHandler := func(job *Response) {
|
||||
str := string(job.Data)
|
||||
if str == "ABCDEF" {
|
||||
@ -70,6 +96,9 @@ func TestClientDo(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestClientStatus(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
status, err := client.Status("handle not exists")
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
@ -105,6 +134,9 @@ func TestClientStatus(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestClientClose(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
if err := client.Close(); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
@ -9,6 +9,9 @@ var (
|
||||
)
|
||||
|
||||
func TestPoolAdd(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
t.Log("Add servers")
|
||||
c := 2
|
||||
if err := pool.Add("tcp4", "127.0.0.1:4730", 1); err != nil {
|
||||
@ -24,6 +27,9 @@ func TestPoolAdd(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestPoolEcho(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
echo, err := pool.Echo("", []byte(TestStr))
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
@ -41,6 +47,9 @@ func TestPoolEcho(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestPoolDoBg(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
addr, handle, err := pool.DoBg("ToUpper",
|
||||
[]byte("abcdef"), JobLow)
|
||||
if err != nil {
|
||||
@ -55,6 +64,9 @@ func TestPoolDoBg(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestPoolDo(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
jobHandler := func(job *Response) {
|
||||
str := string(job.Data)
|
||||
if str == "ABCDEF" {
|
||||
@ -77,6 +89,9 @@ func TestPoolDo(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestPoolStatus(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
status, err := pool.Status("127.0.0.1:4730", "handle not exists")
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
@ -114,6 +129,9 @@ func TestPoolStatus(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestPoolClose(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
return
|
||||
if err := pool.Close(); err != nil {
|
||||
t.Error(err)
|
||||
|
@ -53,5 +53,4 @@ func ExampleWorker() {
|
||||
w.Echo([]byte("Hello"))
|
||||
// Waiting results
|
||||
wg.Wait()
|
||||
// Output: Hello
|
||||
}
|
||||
|
@ -2,18 +2,35 @@ package worker
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"flag"
|
||||
"os"
|
||||
"sync"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
var worker *Worker
|
||||
var (
|
||||
worker *Worker
|
||||
runIntegrationTests bool
|
||||
)
|
||||
|
||||
func init() {
|
||||
worker = New(Unlimited)
|
||||
}
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
integrationsTestFlag := flag.Bool("integration", false, "Run the integration tests (in addition to the unit tests)")
|
||||
if integrationsTestFlag != nil {
|
||||
runIntegrationTests = *integrationsTestFlag
|
||||
}
|
||||
code := m.Run()
|
||||
os.Exit(code)
|
||||
}
|
||||
|
||||
func TestWorkerErrNoneAgents(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
err := worker.Ready()
|
||||
if err != ErrNoneAgents {
|
||||
t.Error("ErrNoneAgents expected.")
|
||||
@ -21,6 +38,9 @@ func TestWorkerErrNoneAgents(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestWorkerAddServer(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
t.Log("Add local server 127.0.0.1:4730.")
|
||||
if err := worker.AddServer(Network, "127.0.0.1:4730"); err != nil {
|
||||
t.Error(err)
|
||||
@ -33,6 +53,9 @@ func TestWorkerAddServer(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestWorkerErrNoneFuncs(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
err := worker.Ready()
|
||||
if err != ErrNoneFuncs {
|
||||
t.Error("ErrNoneFuncs expected.")
|
||||
@ -44,6 +67,9 @@ func foobar(job Job) ([]byte, error) {
|
||||
}
|
||||
|
||||
func TestWorkerAddFunction(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
if err := worker.AddFunc("foobar", foobar, 0); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
@ -57,12 +83,18 @@ func TestWorkerAddFunction(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestWorkerRemoveFunc(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
if err := worker.RemoveFunc("foobar"); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWork(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
var wg sync.WaitGroup
|
||||
worker.JobHandler = func(job Job) error {
|
||||
t.Logf("%s", job.Data())
|
||||
@ -80,6 +112,9 @@ func TestWork(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestLargeDataWork(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
worker := New(Unlimited)
|
||||
defer worker.Close()
|
||||
|
||||
@ -136,10 +171,16 @@ func TestLargeDataWork(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestWorkerClose(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
worker.Close()
|
||||
}
|
||||
|
||||
func TestWorkWithoutReady(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
other_worker := New(Unlimited)
|
||||
|
||||
if err := other_worker.AddServer(Network, "127.0.0.1:4730"); err != nil {
|
||||
@ -193,6 +234,9 @@ func TestWorkWithoutReady(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestWorkWithoutReadyWithPanic(t *testing.T) {
|
||||
if !runIntegrationTests {
|
||||
t.Skip("To run this test, use: go test -integration")
|
||||
}
|
||||
other_worker := New(Unlimited)
|
||||
|
||||
timeout := make(chan bool, 1)
|
||||
|
Loading…
Reference in New Issue
Block a user