Make it possible to switch source for some domains

1 files changed, 8 insertions, 0 deletions

diff --git a/internal/source/domains.go b/internal/source/domains.go
index eadf25c4..df375883 100644
--- a/internal/source/domains.go
+++ b/internal/source/domains.go
@@ -1,6 +1,7 @@
 package source
 
 import (
+	"gitlab.com/gitlab-org/gitlab-pages/internal/domain"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/source/disk"
 )
 
@@ -18,3 +19,10 @@ func NewDomains() *Domains {
 		Source: disk.New(),
 	}
 }
+
+// GetDomain overridden here allows us to switch behavior and the domains
+// source for some subset of domains, to test / PoC the new GitLab Domains
+// Source that we plan to introduce
+func (d *Domains) GetDomain(name string) *domain.Domain {
+	return d.Source.GetDomain(name)
+}