Add getRawUrlFileContent method to handle raw url & Unit Test

Signed-off-by: Aniruddha Basak <[email protected]>

cmd/clusterctl/client/cluster/template.go

@@ -145,6 +145,8 @@ func (t *templateClient) getURLContent(templateURL string) ([]byte, error) {
 
 	if rURL.Scheme == "https" && rURL.Host == "github.com" {
 		return t.getGitHubFileContent(rURL)
+	} else if rURL.Scheme == "https" {
+		return t.getRawUrlFileContent(templateURL)
 	}
 
 	if rURL.Scheme == "file" || rURL.Scheme == "" {
@@ -210,6 +212,20 @@ func (t *templateClient) getGitHubFileContent(rURL *url.URL) ([]byte, error) {
 	return content, nil
 }
 
+func (t *templateClient) getRawUrlFileContent(rURL string) ([]byte, error) {
+	res, err := http.Get(rURL)
+	if err != nil {
+		return nil, err
+	}
+
+	content, err := io.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	return content, nil
+}
+
 func getGitHubClient(configVariablesClient config.VariablesClient) (*github.Client, error) {
 	var authenticatingHTTPClient *http.Client
 	if token, err := configVariablesClient.Get(config.GitHubTokenVariable); err == nil {

cmd/clusterctl/client/cluster/template_test.go

@@ -42,6 +42,21 @@ kind: Cluster
 apiVersion: cluster.x-k8s.io/v1beta1
 kind: Machine`
 
+var rawURLTemplate = `apiVersion: v1
+kind: Pod
+metadata:
+  name: command-demo
+  labels:
+    purpose: demonstrate-command
+spec:
+  containers:
+  - name: command-demo-container
+    image: debian
+    command: ["printenv"]
+    args: ["HOSTNAME", "KUBERNETES_PORT"]
+  restartPolicy: OnFailure
+`
+
 func Test_templateClient_GetFromConfigMap(t *testing.T) {
 	g := NewWithT(t)
 
@@ -225,6 +240,51 @@ func Test_templateClient_getGitHubFileContent(t *testing.T) {
 	}
 }
 
+func Test_templateClient_getRawUrlFileContent(t *testing.T) {
+	type args struct {
+		rURL string
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    []byte
+		wantErr bool
+	}{
+		{
+			name: "Return custom template",
+			args: args{
+				rURL: "https://raw.githubusercontent.com/kubernetes/website/main/content/en/examples/pods/commands.yaml",
+			},
+			want:    []byte(rawURLTemplate),
+			wantErr: false,
+		},
+		{
+			name: "Wrong url",
+			args: args{
+				rURL: "https://foo.com/path/to/cluster-template.yaml",
+			},
+			want:    nil,
+			wantErr: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			g := NewWithT(t)
+
+			c := &templateClient{}
+			got, err := c.getRawUrlFileContent(tt.args.rURL)
+			if tt.wantErr {
+				g.Expect(err).To(HaveOccurred())
+				return
+			}
+
+			g.Expect(err).NotTo(HaveOccurred())
+
+			g.Expect(got).To(Equal(tt.want))
+		})
+	}
+}
+
 func Test_templateClient_getLocalFileContent(t *testing.T) {
 	g := NewWithT(t)