@@ -12,14 +12,14 @@ import (
1212 "github.com/stacklok/toolhive/pkg/registry/types"
1313)
1414
15- func TestNewServerRegistryFromToolhive (t * testing.T ) {
15+ func TestNewUpstreamRegistryFromToolhiveRegistry (t * testing.T ) {
1616 t .Parallel ()
1717
1818 tests := []struct {
1919 name string
2020 toolhiveReg * types.Registry
2121 expectError bool
22- validate func (* testing.T , * types.ServerRegistry )
22+ validate func (* testing.T , * types.UpstreamRegistry )
2323 }{
2424 {
2525 name : "successful conversion with container servers" ,
@@ -42,7 +42,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
4242 RemoteServers : make (map [string ]* types.RemoteServerMetadata ),
4343 },
4444 expectError : false ,
45- validate : func (t * testing.T , sr * types.ServerRegistry ) {
45+ validate : func (t * testing.T , sr * types.UpstreamRegistry ) {
4646 t .Helper ()
4747 assert .Equal (t , "1.0.0" , sr .Version )
4848 assert .Equal (t , "2024-01-01T00:00:00Z" , sr .LastUpdated )
@@ -72,7 +72,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
7272 },
7373 },
7474 expectError : false ,
75- validate : func (t * testing.T , sr * types.ServerRegistry ) {
75+ validate : func (t * testing.T , sr * types.UpstreamRegistry ) {
7676 t .Helper ()
7777 assert .Len (t , sr .Servers , 1 )
7878 assert .Contains (t , sr .Servers [0 ].Name , "remote-server" )
@@ -87,7 +87,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
8787 RemoteServers : make (map [string ]* types.RemoteServerMetadata ),
8888 },
8989 expectError : false ,
90- validate : func (t * testing.T , sr * types.ServerRegistry ) {
90+ validate : func (t * testing.T , sr * types.UpstreamRegistry ) {
9191 t .Helper ()
9292 assert .Empty (t , sr .Servers )
9393 },
@@ -98,7 +98,7 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
9898 t .Run (tt .name , func (t * testing.T ) {
9999 t .Parallel ()
100100
101- result , err := NewServerRegistryFromToolhive (tt .toolhiveReg )
101+ result , err := NewUpstreamRegistryFromToolhiveRegistry (tt .toolhiveReg )
102102
103103 if tt .expectError {
104104 assert .Error (t , err )
@@ -114,13 +114,13 @@ func TestNewServerRegistryFromToolhive(t *testing.T) {
114114 }
115115}
116116
117- func TestNewServerRegistryFromUpstream (t * testing.T ) {
117+ func TestNewUpstreamRegistryFromUpstreamServers (t * testing.T ) {
118118 t .Parallel ()
119119
120120 tests := []struct {
121121 name string
122122 servers []upstreamv0.ServerJSON
123- validate func (* testing.T , * types.ServerRegistry )
123+ validate func (* testing.T , * types.UpstreamRegistry )
124124 }{
125125 {
126126 name : "create from upstream servers" ,
@@ -139,7 +139,7 @@ func TestNewServerRegistryFromUpstream(t *testing.T) {
139139 },
140140 },
141141 },
142- validate : func (t * testing.T , sr * types.ServerRegistry ) {
142+ validate : func (t * testing.T , sr * types.UpstreamRegistry ) {
143143 t .Helper ()
144144 assert .Equal (t , "1.0.0" , sr .Version )
145145 assert .NotEmpty (t , sr .LastUpdated )
@@ -150,7 +150,7 @@ func TestNewServerRegistryFromUpstream(t *testing.T) {
150150 {
151151 name : "create from empty slice" ,
152152 servers : []upstreamv0.ServerJSON {},
153- validate : func (t * testing.T , sr * types.ServerRegistry ) {
153+ validate : func (t * testing.T , sr * types.UpstreamRegistry ) {
154154 t .Helper ()
155155 assert .Empty (t , sr .Servers )
156156 },
@@ -161,7 +161,7 @@ func TestNewServerRegistryFromUpstream(t *testing.T) {
161161 t .Run (tt .name , func (t * testing.T ) {
162162 t .Parallel ()
163163
164- result := NewServerRegistryFromUpstream (tt .servers )
164+ result := NewUpstreamRegistryFromUpstreamServers (tt .servers )
165165
166166 assert .NotNil (t , result )
167167 if tt .validate != nil {
@@ -183,7 +183,7 @@ func TestNewServerRegistryFromUpstream_DefaultValues(t *testing.T) {
183183 },
184184 }
185185
186- result := NewServerRegistryFromUpstream (servers )
186+ result := NewUpstreamRegistryFromUpstreamServers (servers )
187187
188188 // Verify defaults
189189 assert .Equal (t , "1.0.0" , result .Version )
0 commit comments