rename mock things to memory
This commit is contained in:
		
							
								
								
									
										89
									
								
								transport/memory/memory_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										89
									
								
								transport/memory/memory_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,89 @@ | ||||
| package memory | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/micro/go-micro/transport" | ||||
| ) | ||||
|  | ||||
| func TestMemoryTransport(t *testing.T) { | ||||
| 	tr := NewTransport() | ||||
|  | ||||
| 	// bind / listen | ||||
| 	l, err := tr.Listen("localhost:8080") | ||||
| 	if err != nil { | ||||
| 		t.Fatalf("Unexpected error listening %v", err) | ||||
| 	} | ||||
| 	defer l.Close() | ||||
|  | ||||
| 	// accept | ||||
| 	go func() { | ||||
| 		if err := l.Accept(func(sock transport.Socket) { | ||||
| 			for { | ||||
| 				var m transport.Message | ||||
| 				if err := sock.Recv(&m); err != nil { | ||||
| 					return | ||||
| 				} | ||||
| 				t.Logf("Server Received %s", string(m.Body)) | ||||
| 				if err := sock.Send(&transport.Message{ | ||||
| 					Body: []byte(`pong`), | ||||
| 				}); err != nil { | ||||
| 					return | ||||
| 				} | ||||
| 			} | ||||
| 		}); err != nil { | ||||
| 			t.Fatalf("Unexpected error accepting %v", err) | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	// dial | ||||
| 	c, err := tr.Dial("localhost:8080") | ||||
| 	if err != nil { | ||||
| 		t.Fatalf("Unexpected error dialing %v", err) | ||||
| 	} | ||||
| 	defer c.Close() | ||||
|  | ||||
| 	// send <=> receive | ||||
| 	for i := 0; i < 3; i++ { | ||||
| 		if err := c.Send(&transport.Message{ | ||||
| 			Body: []byte(`ping`), | ||||
| 		}); err != nil { | ||||
| 			return | ||||
| 		} | ||||
| 		var m transport.Message | ||||
| 		if err := c.Recv(&m); err != nil { | ||||
| 			return | ||||
| 		} | ||||
| 		t.Logf("Client Received %s", string(m.Body)) | ||||
| 	} | ||||
|  | ||||
| } | ||||
|  | ||||
| func TestListener(t *testing.T) { | ||||
| 	tr := NewTransport() | ||||
|  | ||||
| 	// bind / listen on random port | ||||
| 	l, err := tr.Listen(":0") | ||||
| 	if err != nil { | ||||
| 		t.Fatalf("Unexpected error listening %v", err) | ||||
| 	} | ||||
| 	defer l.Close() | ||||
|  | ||||
| 	// try again | ||||
| 	l2, err := tr.Listen(":0") | ||||
| 	if err != nil { | ||||
| 		t.Fatalf("Unexpected error listening %v", err) | ||||
| 	} | ||||
| 	defer l2.Close() | ||||
|  | ||||
| 	// now make sure it still fails | ||||
| 	l3, err := tr.Listen(":8080") | ||||
| 	if err != nil { | ||||
| 		t.Fatalf("Unexpected error listening %v", err) | ||||
| 	} | ||||
| 	defer l3.Close() | ||||
|  | ||||
| 	if _, err := tr.Listen(":8080"); err == nil { | ||||
| 		t.Fatal("Expected error binding to :8080 got nil") | ||||
| 	} | ||||
| } | ||||
		Reference in New Issue
	
	Block a user