Add Unit test to daemon.SearchRegistryForImages…
… and refactor a little bit some daemon on the way. - Move `SearchRegistryForImages` to a new file (`daemon/search.go`) as `daemon.go` is getting pretty big. - `registry.Service` is now an interface (allowing us to decouple it a little bit and thus unit test easily). - Add some unit test for `SearchRegistryForImages`. - Use UniqueExactMatch for search filters - And use empty restore id for now in client.ContainerStart. Signed-off-by: Vincent Demeester <vincent@sbr.pm>master
							parent
							
								
									c4778ea1be
								
							
						
					
					
						commit
						04476ff5a9
					
				|  | @ -661,7 +661,7 @@ func TestMirrorEndpointLookup(t *testing.T) { | ||||||
| 		} | 		} | ||||||
| 		return false | 		return false | ||||||
| 	} | 	} | ||||||
| 	s := Service{config: makeServiceConfig([]string{"my.mirror"}, nil)} | 	s := DefaultService{config: makeServiceConfig([]string{"my.mirror"}, nil)} | ||||||
| 
 | 
 | ||||||
| 	imageName, err := reference.WithName(IndexName + "/test/image") | 	imageName, err := reference.WithName(IndexName + "/test/image") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
|  |  | ||||||
|  | @ -7,35 +7,50 @@ import ( | ||||||
| 	"net/url" | 	"net/url" | ||||||
| 	"strings" | 	"strings" | ||||||
| 
 | 
 | ||||||
|  | 	"golang.org/x/net/context" | ||||||
|  | 
 | ||||||
| 	"github.com/Sirupsen/logrus" | 	"github.com/Sirupsen/logrus" | ||||||
| 	"github.com/docker/docker/reference" | 	"github.com/docker/docker/reference" | ||||||
| 	"github.com/docker/engine-api/types" | 	"github.com/docker/engine-api/types" | ||||||
| 	registrytypes "github.com/docker/engine-api/types/registry" | 	registrytypes "github.com/docker/engine-api/types/registry" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| // Service is a registry service. It tracks configuration data such as a list
 | // Service is the interface defining what a registry service should implement.
 | ||||||
|  | type Service interface { | ||||||
|  | 	Auth(ctx context.Context, authConfig *types.AuthConfig, userAgent string) (status, token string, err error) | ||||||
|  | 	LookupPullEndpoints(hostname string) (endpoints []APIEndpoint, err error) | ||||||
|  | 	LookupPushEndpoints(hostname string) (endpoints []APIEndpoint, err error) | ||||||
|  | 	ResolveRepository(name reference.Named) (*RepositoryInfo, error) | ||||||
|  | 	ResolveIndex(name string) (*registrytypes.IndexInfo, error) | ||||||
|  | 	Search(ctx context.Context, term string, authConfig *types.AuthConfig, userAgent string, headers map[string][]string) (*registrytypes.SearchResults, error) | ||||||
|  | 	ServiceConfig() *registrytypes.ServiceConfig | ||||||
|  | 	TLSConfig(hostname string) (*tls.Config, error) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // DefaultService is a registry service. It tracks configuration data such as a list
 | ||||||
| // of mirrors.
 | // of mirrors.
 | ||||||
| type Service struct { | type DefaultService struct { | ||||||
| 	config *serviceConfig | 	config *serviceConfig | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // NewService returns a new instance of Service ready to be
 | // NewService returns a new instance of DefaultService ready to be
 | ||||||
| // installed into an engine.
 | // installed into an engine.
 | ||||||
| func NewService(options ServiceOptions) *Service { | func NewService(options ServiceOptions) *DefaultService { | ||||||
| 	return &Service{ | 	return &DefaultService{ | ||||||
| 		config: newServiceConfig(options), | 		config: newServiceConfig(options), | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // ServiceConfig returns the public registry service configuration.
 | // ServiceConfig returns the public registry service configuration.
 | ||||||
| func (s *Service) ServiceConfig() *registrytypes.ServiceConfig { | func (s *DefaultService) ServiceConfig() *registrytypes.ServiceConfig { | ||||||
| 	return &s.config.ServiceConfig | 	return &s.config.ServiceConfig | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // Auth contacts the public registry with the provided credentials,
 | // Auth contacts the public registry with the provided credentials,
 | ||||||
| // and returns OK if authentication was successful.
 | // and returns OK if authentication was successful.
 | ||||||
| // It can be used to verify the validity of a client's credentials.
 | // It can be used to verify the validity of a client's credentials.
 | ||||||
| func (s *Service) Auth(authConfig *types.AuthConfig, userAgent string) (status, token string, err error) { | func (s *DefaultService) Auth(ctx context.Context, authConfig *types.AuthConfig, userAgent string) (status, token string, err error) { | ||||||
|  | 	// TODO Use ctx when searching for repositories
 | ||||||
| 	serverAddress := authConfig.ServerAddress | 	serverAddress := authConfig.ServerAddress | ||||||
| 	if serverAddress == "" { | 	if serverAddress == "" { | ||||||
| 		serverAddress = IndexServer | 		serverAddress = IndexServer | ||||||
|  | @ -93,7 +108,8 @@ func splitReposSearchTerm(reposName string) (string, string) { | ||||||
| 
 | 
 | ||||||
| // Search queries the public registry for images matching the specified
 | // Search queries the public registry for images matching the specified
 | ||||||
| // search terms, and returns the results.
 | // search terms, and returns the results.
 | ||||||
| func (s *Service) Search(term string, authConfig *types.AuthConfig, userAgent string, headers map[string][]string) (*registrytypes.SearchResults, error) { | func (s *DefaultService) Search(ctx context.Context, term string, authConfig *types.AuthConfig, userAgent string, headers map[string][]string) (*registrytypes.SearchResults, error) { | ||||||
|  | 	// TODO Use ctx when searching for repositories
 | ||||||
| 	if err := validateNoScheme(term); err != nil { | 	if err := validateNoScheme(term); err != nil { | ||||||
| 		return nil, err | 		return nil, err | ||||||
| 	} | 	} | ||||||
|  | @ -130,12 +146,12 @@ func (s *Service) Search(term string, authConfig *types.AuthConfig, userAgent st | ||||||
| 
 | 
 | ||||||
| // ResolveRepository splits a repository name into its components
 | // ResolveRepository splits a repository name into its components
 | ||||||
| // and configuration of the associated registry.
 | // and configuration of the associated registry.
 | ||||||
| func (s *Service) ResolveRepository(name reference.Named) (*RepositoryInfo, error) { | func (s *DefaultService) ResolveRepository(name reference.Named) (*RepositoryInfo, error) { | ||||||
| 	return newRepositoryInfo(s.config, name) | 	return newRepositoryInfo(s.config, name) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // ResolveIndex takes indexName and returns index info
 | // ResolveIndex takes indexName and returns index info
 | ||||||
| func (s *Service) ResolveIndex(name string) (*registrytypes.IndexInfo, error) { | func (s *DefaultService) ResolveIndex(name string) (*registrytypes.IndexInfo, error) { | ||||||
| 	return newIndexInfo(s.config, name) | 	return newIndexInfo(s.config, name) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | @ -155,25 +171,25 @@ func (e APIEndpoint) ToV1Endpoint(userAgent string, metaHeaders http.Header) (*V | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // TLSConfig constructs a client TLS configuration based on server defaults
 | // TLSConfig constructs a client TLS configuration based on server defaults
 | ||||||
| func (s *Service) TLSConfig(hostname string) (*tls.Config, error) { | func (s *DefaultService) TLSConfig(hostname string) (*tls.Config, error) { | ||||||
| 	return newTLSConfig(hostname, isSecureIndex(s.config, hostname)) | 	return newTLSConfig(hostname, isSecureIndex(s.config, hostname)) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (s *Service) tlsConfigForMirror(mirrorURL *url.URL) (*tls.Config, error) { | func (s *DefaultService) tlsConfigForMirror(mirrorURL *url.URL) (*tls.Config, error) { | ||||||
| 	return s.TLSConfig(mirrorURL.Host) | 	return s.TLSConfig(mirrorURL.Host) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // LookupPullEndpoints creates a list of endpoints to try to pull from, in order of preference.
 | // LookupPullEndpoints creates a list of endpoints to try to pull from, in order of preference.
 | ||||||
| // It gives preference to v2 endpoints over v1, mirrors over the actual
 | // It gives preference to v2 endpoints over v1, mirrors over the actual
 | ||||||
| // registry, and HTTPS over plain HTTP.
 | // registry, and HTTPS over plain HTTP.
 | ||||||
| func (s *Service) LookupPullEndpoints(hostname string) (endpoints []APIEndpoint, err error) { | func (s *DefaultService) LookupPullEndpoints(hostname string) (endpoints []APIEndpoint, err error) { | ||||||
| 	return s.lookupEndpoints(hostname) | 	return s.lookupEndpoints(hostname) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // LookupPushEndpoints creates a list of endpoints to try to push to, in order of preference.
 | // LookupPushEndpoints creates a list of endpoints to try to push to, in order of preference.
 | ||||||
| // It gives preference to v2 endpoints over v1, and HTTPS over plain HTTP.
 | // It gives preference to v2 endpoints over v1, and HTTPS over plain HTTP.
 | ||||||
| // Mirrors are not included.
 | // Mirrors are not included.
 | ||||||
| func (s *Service) LookupPushEndpoints(hostname string) (endpoints []APIEndpoint, err error) { | func (s *DefaultService) LookupPushEndpoints(hostname string) (endpoints []APIEndpoint, err error) { | ||||||
| 	allEndpoints, err := s.lookupEndpoints(hostname) | 	allEndpoints, err := s.lookupEndpoints(hostname) | ||||||
| 	if err == nil { | 	if err == nil { | ||||||
| 		for _, endpoint := range allEndpoints { | 		for _, endpoint := range allEndpoints { | ||||||
|  | @ -185,7 +201,7 @@ func (s *Service) LookupPushEndpoints(hostname string) (endpoints []APIEndpoint, | ||||||
| 	return endpoints, err | 	return endpoints, err | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (s *Service) lookupEndpoints(hostname string) (endpoints []APIEndpoint, err error) { | func (s *DefaultService) lookupEndpoints(hostname string) (endpoints []APIEndpoint, err error) { | ||||||
| 	endpoints, err = s.lookupV2Endpoints(hostname) | 	endpoints, err = s.lookupV2Endpoints(hostname) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, err | 		return nil, err | ||||||
|  |  | ||||||
|  | @ -6,7 +6,7 @@ import ( | ||||||
| 	"github.com/docker/go-connections/tlsconfig" | 	"github.com/docker/go-connections/tlsconfig" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| func (s *Service) lookupV1Endpoints(hostname string) (endpoints []APIEndpoint, err error) { | func (s *DefaultService) lookupV1Endpoints(hostname string) (endpoints []APIEndpoint, err error) { | ||||||
| 	var cfg = tlsconfig.ServerDefault | 	var cfg = tlsconfig.ServerDefault | ||||||
| 	tlsConfig := &cfg | 	tlsConfig := &cfg | ||||||
| 	if hostname == DefaultNamespace { | 	if hostname == DefaultNamespace { | ||||||
|  |  | ||||||
|  | @ -7,7 +7,7 @@ import ( | ||||||
| 	"github.com/docker/go-connections/tlsconfig" | 	"github.com/docker/go-connections/tlsconfig" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| func (s *Service) lookupV2Endpoints(hostname string) (endpoints []APIEndpoint, err error) { | func (s *DefaultService) lookupV2Endpoints(hostname string) (endpoints []APIEndpoint, err error) { | ||||||
| 	var cfg = tlsconfig.ServerDefault | 	var cfg = tlsconfig.ServerDefault | ||||||
| 	tlsConfig := &cfg | 	tlsConfig := &cfg | ||||||
| 	if hostname == DefaultNamespace || hostname == DefaultV1Registry.Host { | 	if hostname == DefaultNamespace || hostname == DefaultV1Registry.Host { | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue