mark mock engine always up

projecteru2 · Feb 3, 2023 · 55e504a · 55e504a
1 parent ef94eb4
commit 55e504a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/store/etcdv3/node.go b/store/etcdv3/node.go
@@ -15,6 +15,7 @@ import (
 	"github.com/projecteru2/core/engine"
 	enginefactory "github.com/projecteru2/core/engine/factory"
 	"github.com/projecteru2/core/engine/fake"
+	"github.com/projecteru2/core/engine/mocks/fakeengine"
 	"github.com/projecteru2/core/log"
 	"github.com/projecteru2/core/types"
 	"github.com/projecteru2/core/utils"
@@ -319,7 +320,9 @@ func (m *Mercury) doGetNodes(ctx context.Context, kvs []*mvccpb.KeyValue, labels
 		node := node
 		_ = m.pool.Invoke(func() {
 			defer wg.Done()
-			if _, err := m.GetNodeStatus(ctx, node.Name); err != nil && !errors.Is(err, types.ErrInvaildCount) {
+			if strings.HasPrefix(node.Endpoint, fakeengine.PrefixKey) {
+				node.Available = true
+			} else if _, err := m.GetNodeStatus(ctx, node.Name); err != nil && !errors.Is(err, types.ErrInvaildCount) {
 				logger.Errorf(ctx, err, "failed to get node status of %+v", node.Name)
 			} else {
 				node.Available = err == nil

diff --git a/store/redis/node.go b/store/redis/node.go
@@ -12,6 +12,7 @@ import (
 	"github.com/projecteru2/core/engine"
 	enginefactory "github.com/projecteru2/core/engine/factory"
 	"github.com/projecteru2/core/engine/fake"
+	"github.com/projecteru2/core/engine/mocks/fakeengine"
 	"github.com/projecteru2/core/log"
 	"github.com/projecteru2/core/types"
 	"github.com/projecteru2/core/utils"
@@ -302,7 +303,9 @@ func (r *Rediaron) doGetNodes(ctx context.Context, kvs map[string]string, labels
 		node := node
 		_ = r.pool.Invoke(func() {
 			defer wg.Done()
-			if _, err := r.GetNodeStatus(ctx, node.Name); err != nil && !errors.Is(err, types.ErrInvaildCount) {
+			if strings.HasPrefix(node.Endpoint, fakeengine.PrefixKey) {
+				node.Available = true
+			} else if _, err := r.GetNodeStatus(ctx, node.Name); err != nil && !errors.Is(err, types.ErrInvaildCount) {
 				logger.Errorf(ctx, err, "failed to get node status of %+v", node.Name)
 			} else {
 				node.Available = err == nil