test

v2/test/mencacheBackend_test.go

@@ -0,0 +1,38 @@
+package test
+
+import (
+ "fmt"
+ "github.com/gojuukaze/YTask/v2/backends"
+ "github.com/gojuukaze/YTask/v2/message"
+ "github.com/gojuukaze/YTask/v2/yerrors"
+ "testing"
+ "time"
+)
+
+func TestMemcacheBackend(t *testing.T) {
+ b := backends.NewMemCacheBackend("127.0.0.1", "11211", 1)
+ result := message.NewResult("xx123")
+ result.FuncReturn = nil
+ b.Activate()
+ err := b.SetResult(result, 2)
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ r2, err := b.GetResult(result.GetBackendKey())
+ if err != nil {
+ t.Fatal(err)
+ }
+ if fmt.Sprintf("%v",r2) != fmt.Sprintf("%v",result) {
+ t.Fatalf("%v != %v", r2, result)
+ }
+
+ time.Sleep(2 * time.Second)
+
+ _, err = b.GetResult(result.GetBackendKey())
+ if !yerrors.IsEqual(err, yerrors.ErrTypeNilResult) {
+ t.Fatal("err != ErrNilResult")
+
+ }
+
+}

v2/test/mongoBackend_test.go

@@ -0,0 +1,38 @@
+package test
+
+import (
+ "fmt"
+ "github.com/gojuukaze/YTask/v2/backends"
+ "github.com/gojuukaze/YTask/v2/message"
+ "testing"
+)
+
+func TestMongoBackend(t *testing.T) {
+ b := backends.NewMongoBackend("127.0.0.1", "27017", "", "", "task", "task")
+ result := message.NewResult("xx123")
+ result.FuncReturn = nil
+ b.Activate()
+ err := b.SetResult(result, 2)
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ r2, err := b.GetResult(result.GetBackendKey())
+ if err != nil {
+ t.Fatal(err)
+ }
+ if fmt.Sprintf("%v", r2) != fmt.Sprintf("%v", result) {
+ t.Fatalf("%v != %v", r2, result)
+ }
+
+ // mongo不支持过期时间
+
+ //time.Sleep(2 * time.Second)
+ //
+ //_, err = b.GetResult(result.GetBackendKey())
+ //if !yerrors.IsEqual(err, yerrors.ErrTypeNilResult) {
+ // t.Fatal("err != ErrNilResult")
+ //
+ //}
+
+}

v2/test/rabbitmqBroker_test.go

@@ -0,0 +1,43 @@
+package test
+
+import (
+ "fmt"
+ "github.com/gojuukaze/YTask/v2/brokers"
+ "github.com/gojuukaze/YTask/v2/controller"
+ "github.com/gojuukaze/YTask/v2/message"
+ "testing"
+)
+
+func TestRabbitmqBroker(t *testing.T) {
+ b := brokers.NewRabbitMqBroker("127.0.0.1", "5672", "guest", "guest")
+ var broker brokers.BrokerInterface = &b
+ broker.Activate()
+ msg := message.NewMessage(controller.NewTaskCtl())
+ msg2 := message.NewMessage(controller.NewTaskCtl())
+
+ err := broker.Send("test_redis", msg)
+ if err != nil {
+ t.Fatal(err)
+ }
+ err = broker.Send("test_redis", msg2)
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ m, err := broker.Next("test_redis")
+ if err != nil {
+ t.Fatal(err)
+ }
+ if fmt.Sprintf("%v", m) != fmt.Sprintf("%v", msg) {
+ t.Fatalf("%v != %v", m, msg)
+ }
+
+ m2, err := broker.Next("test_redis")
+ if err != nil {
+ t.Fatal(err)
+ }
+ if fmt.Sprintf("%v", m2) != fmt.Sprintf("%v", msg2) {
+ t.Fatalf("%v != %v", m2, msg2)
+
+ }
+}

v2/test/ytask_test.go

@@ -1,5 +1,10 @@
 package test
 
+//
+// cd v2
+// go test -v -count=1 test/*
+//
+
 import (
  "context"
  "errors"