@@ -64,7 +64,7 @@ func TestEndToEnd(t *testing.T) {
6464 notificationChans ["progress_server" ] <- 0
6565 },
6666 }
67- s := NewServer (nil , sopts )
67+ s := NewServer (testImpl , sopts )
6868 AddTool (s , & Tool {
6969 Name : "greet" ,
7070 Description : "say hi" ,
@@ -125,7 +125,7 @@ func TestEndToEnd(t *testing.T) {
125125 notificationChans ["progress_client" ] <- 0
126126 },
127127 }
128- c := NewClient (nil , opts )
128+ c := NewClient (testImpl , opts )
129129 rootAbs , err := filepath .Abs (filepath .FromSlash ("testdata/files" ))
130130 if err != nil {
131131 t .Fatal (err )
@@ -510,7 +510,7 @@ func basicConnection(t *testing.T, config func(*Server)) (*ServerSession, *Clien
510510 ctx := context .Background ()
511511 ct , st := NewInMemoryTransports ()
512512
513- s := NewServer (nil , nil )
513+ s := NewServer (testImpl , nil )
514514 if config != nil {
515515 config (s )
516516 }
@@ -519,7 +519,7 @@ func basicConnection(t *testing.T, config func(*Server)) (*ServerSession, *Clien
519519 t .Fatal (err )
520520 }
521521
522- c := NewClient (nil , nil )
522+ c := NewClient (testImpl , nil )
523523 cs , err := c .Connect (ctx , ct )
524524 if err != nil {
525525 t .Fatal (err )
@@ -562,13 +562,13 @@ func TestBatching(t *testing.T) {
562562 ctx := context .Background ()
563563 ct , st := NewInMemoryTransports ()
564564
565- s := NewServer (nil , nil )
565+ s := NewServer (testImpl , nil )
566566 _ , err := s .Connect (ctx , st )
567567 if err != nil {
568568 t .Fatal (err )
569569 }
570570
571- c := NewClient (nil , nil )
571+ c := NewClient (testImpl , nil )
572572 // TODO: this test is broken, because increasing the batch size here causes
573573 // 'initialize' to block. Therefore, we can only test with a size of 1.
574574 // Since batching is being removed, we can probably just delete this.
@@ -632,7 +632,7 @@ func TestMiddleware(t *testing.T) {
632632 ctx := context .Background ()
633633 ct , st := NewInMemoryTransports ()
634634
635- s := NewServer (nil , nil )
635+ s := NewServer (testImpl , nil )
636636 ss , err := s .Connect (ctx , st )
637637 if err != nil {
638638 t .Fatal (err )
@@ -656,7 +656,7 @@ func TestMiddleware(t *testing.T) {
656656 s .AddSendingMiddleware (traceCalls [* ServerSession ](& sbuf , "S1" ), traceCalls [* ServerSession ](& sbuf , "S2" ))
657657 s .AddReceivingMiddleware (traceCalls [* ServerSession ](& sbuf , "R1" ), traceCalls [* ServerSession ](& sbuf , "R2" ))
658658
659- c := NewClient (nil , nil )
659+ c := NewClient (testImpl , nil )
660660 c .AddSendingMiddleware (traceCalls [* ClientSession ](& cbuf , "S1" ), traceCalls [* ClientSession ](& cbuf , "S2" ))
661661 c .AddReceivingMiddleware (traceCalls [* ClientSession ](& cbuf , "R1" ), traceCalls [* ClientSession ](& cbuf , "R2" ))
662662
@@ -741,13 +741,13 @@ func TestNoJSONNull(t *testing.T) {
741741 var logbuf safeBuffer
742742 ct = NewLoggingTransport (ct , & logbuf )
743743
744- s := NewServer (nil , nil )
744+ s := NewServer (testImpl , nil )
745745 ss , err := s .Connect (ctx , st )
746746 if err != nil {
747747 t .Fatal (err )
748748 }
749749
750- c := NewClient (nil , nil )
750+ c := NewClient (testImpl , nil )
751751 cs , err := c .Connect (ctx , ct )
752752 if err != nil {
753753 t .Fatal (err )
@@ -810,7 +810,7 @@ func TestKeepAlive(t *testing.T) {
810810 serverOpts := & ServerOptions {
811811 KeepAlive : 100 * time .Millisecond ,
812812 }
813- s := NewServer (nil , serverOpts )
813+ s := NewServer (testImpl , serverOpts )
814814 AddTool (s , greetTool (), sayHi )
815815
816816 ss , err := s .Connect (ctx , st )
@@ -822,7 +822,7 @@ func TestKeepAlive(t *testing.T) {
822822 clientOpts := & ClientOptions {
823823 KeepAlive : 100 * time .Millisecond ,
824824 }
825- c := NewClient (nil , clientOpts )
825+ c := NewClient (testImpl , clientOpts )
826826 cs , err := c .Connect (ctx , ct )
827827 if err != nil {
828828 t .Fatal (err )
@@ -855,7 +855,7 @@ func TestKeepAliveFailure(t *testing.T) {
855855 ct , st := NewInMemoryTransports ()
856856
857857 // Server without keepalive (to test one-sided keepalive)
858- s := NewServer (nil , nil )
858+ s := NewServer (testImpl , nil )
859859 AddTool (s , greetTool (), sayHi )
860860 ss , err := s .Connect (ctx , st )
861861 if err != nil {
@@ -866,7 +866,7 @@ func TestKeepAliveFailure(t *testing.T) {
866866 clientOpts := & ClientOptions {
867867 KeepAlive : 50 * time .Millisecond ,
868868 }
869- c := NewClient (nil , clientOpts )
869+ c := NewClient (testImpl , clientOpts )
870870 cs , err := c .Connect (ctx , ct )
871871 if err != nil {
872872 t .Fatal (err )
@@ -895,3 +895,5 @@ func TestKeepAliveFailure(t *testing.T) {
895895
896896 t .Errorf ("expected connection to be closed by keepalive, but it wasn't. Last error: %v" , err )
897897}
898+
899+ var testImpl = & Implementation {Name : "test" , Version : "v1.0.0" }
0 commit comments