From e87ef5a361e924152a00adef7f38302f1f7d34a7 Mon Sep 17 00:00:00 2001
From: gojuukaze <ikaze_email@163.com>
Date: Mon, 16 Mar 2020 15:28:40 +0800
Subject: [PATCH] test

---
 v2/test/mencacheBackend_test.go | 38 +++++++++++++++++++++++++++++
 v2/test/mongoBackend_test.go    | 38 +++++++++++++++++++++++++++++
 v2/test/rabbitmqBroker_test.go  | 43 +++++++++++++++++++++++++++++++++
 v2/test/ytask_test.go           |  5 ++++
 4 files changed, 124 insertions(+)
 create mode 100644 v2/test/mencacheBackend_test.go
 create mode 100644 v2/test/mongoBackend_test.go
 create mode 100644 v2/test/rabbitmqBroker_test.go

diff --git a/v2/test/mencacheBackend_test.go b/v2/test/mencacheBackend_test.go
new file mode 100644
index 0000000..5fc9f0f
--- /dev/null
+++ b/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")
+
+	}
+
+}
diff --git a/v2/test/mongoBackend_test.go b/v2/test/mongoBackend_test.go
new file mode 100644
index 0000000..7726af4
--- /dev/null
+++ b/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")
+	//
+	//}
+
+}
diff --git a/v2/test/rabbitmqBroker_test.go b/v2/test/rabbitmqBroker_test.go
new file mode 100644
index 0000000..9e5872a
--- /dev/null
+++ b/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)
+
+	}
+}
diff --git a/v2/test/ytask_test.go b/v2/test/ytask_test.go
index 9204728..dbb3883 100644
--- a/v2/test/ytask_test.go
+++ b/v2/test/ytask_test.go
@@ -1,5 +1,10 @@
 package test
 
+//
+// cd v2
+// go test -v -count=1 test/*
+//
+
 import (
 	"context"
 	"errors"