Fix zmq endpoints in test cases (#150)

* Using the IP address 127.0.0.1 instead of localhost in test cases for zmq to prevent potential name resolution issues

Signed-off-by: Qifan Deng <[email protected]>

* Ignore vscode devcontainer config

Signed-off-by: Qifan Deng <[email protected]>

* Fix a formatting error introduced by commit 9235047

Signed-off-by: Qifan Deng <[email protected]>

---------

Signed-off-by: Qifan Deng <[email protected]>

Author: pancak3

.gitignore

@@ -2,5 +2,6 @@ bin
 lib
 vendor
 .vscode
+.devcontainer
 # MacOSX
 .DS_Store

pkg/common/publisher_test.go

@@ -29,9 +29,10 @@ import (
 )
 
 const (
-	topic    = "test-topic"
-	endpoint = "tcp://localhost:5557"
-	data     = "Hello"
+	topic       = "test-topic"
+	subEndpoint = "tcp://*:5557"
+	pubEndpoint = "tcp://localhost:5557"
+	data        = "Hello"
 )
 
 var _ = Describe("Publisher", func() {
@@ -40,7 +41,7 @@ var _ = Describe("Publisher", func() {
 		Expect(err).NotTo(HaveOccurred())
 		sub, err := zctx.NewSocket(zmq.SUB)
 		Expect(err).NotTo(HaveOccurred())
-		err = sub.Bind(endpoint)
+		err = sub.Bind(subEndpoint)
 		Expect(err).NotTo(HaveOccurred())
 		err = sub.SetSubscribe(topic)
 		Expect(err).NotTo(HaveOccurred())
@@ -49,7 +50,7 @@ var _ = Describe("Publisher", func() {
 
 		time.Sleep(100 * time.Millisecond)
 
-		pub, err := NewPublisher(endpoint)
+		pub, err := NewPublisher(pubEndpoint)
 		Expect(err).NotTo(HaveOccurred())
 
 		ctx, cancel := context.WithCancel(context.Background())

pkg/kv-cache/kv_cache_test.go

@@ -33,10 +33,11 @@ import (
 )
 
 const (
-	req1ID   = "req1"
-	req2ID   = "req2"
-	req3ID   = "req3"
-	endpoint = "tcp://localhost:5557"
+	req1ID      = "req1"
+	req2ID      = "req2"
+	req3ID      = "req3"
+	subEndpoint = "tcp://*:5557"
+	pubEndpoint = "tcp://localhost:5557"
 )
 
 type ActionType int
@@ -203,7 +204,7 @@ var _ = Describe("KV cache", Ordered, func() {
 					Port:           1234,
 					Model:          "model",
 					KVCacheSize:    test.cacheSize,
-					ZMQEndpoint:    endpoint,
+					ZMQEndpoint:    pubEndpoint,
 					EventBatchSize: 1,
 				}
 
@@ -306,7 +307,7 @@ var _ = Describe("KV cache", Ordered, func() {
 				Port:        1234,
 				Model:       "model",
 				KVCacheSize: 4,
-				ZMQEndpoint: endpoint,
+				ZMQEndpoint: pubEndpoint,
 			}
 
 			sub, topic := createSub(config)
@@ -415,7 +416,7 @@ var _ = Describe("KV cache", Ordered, func() {
 					Port:        1234,
 					Model:       "model",
 					KVCacheSize: testCase.cacheSize,
-					ZMQEndpoint: endpoint,
+					ZMQEndpoint: pubEndpoint,
 				}
 				blockCache, err := newBlockCache(&config, GinkgoLogr)
 				Expect(err).NotTo(HaveOccurred())
@@ -531,7 +532,7 @@ func createSub(config *common.Configuration) (*zmq.Socket, string) {
 	Expect(err).NotTo(HaveOccurred())
 	sub, err := zctx.NewSocket(zmq.SUB)
 	Expect(err).NotTo(HaveOccurred())
-	err = sub.Bind(endpoint)
+	err = sub.Bind(subEndpoint)
 	Expect(err).NotTo(HaveOccurred())
 	topic := createTopic(config)
 	err = sub.SetSubscribe(topic)