Make new gitlab domains source more testable

1 files changed, 31 insertions, 0 deletions

diff --git a/internal/source/gitlab/client/client_stub.go b/internal/source/gitlab/client/client_stub.go
new file mode 100644
index 00000000..48dbf549
--- /dev/null
+++ b/internal/source/gitlab/client/client_stub.go
@@ -0,0 +1,31 @@
+package client
+
+import (
+	"encoding/json"
+	"os"
+
+	"gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/domain"
+)
+
+// StubClient is a stubbed client used for testing
+type StubClient struct {
+	file string
+}
+
+// GetVirtualDomain reads a test fixture and unmarshalls it
+func (m *StubClient) GetVirtualDomain(host string) (domain *domain.VirtualDomain, err error) {
+	f, err := os.Open(m.file)
+	defer f.Close()
+	if err != nil {
+		return nil, err
+	}
+
+	err = json.NewDecoder(f).Decode(&domain)
+
+	return domain, err
+}
+
+// NewStubClient return a stubbed client
+func NewStubClient(fixture string) *StubClient {
+	return &StubClient{file: fixture}
+}