Function TestGraphServiceV1ReturnsProtobuf

internal/connect/api_test.go:363–418

(t *testing.T)

}

func TestGraphServiceV1ReturnsProtobuf(t *testing.T) {
	p := &mockProvider{
		meta: content.Meta{
			Commit:        "cafe1234",
			DefaultBranch: "main",
		},
		files: []content.File{
			{Path: "graph.proto", Data: []byte("syntax = \"proto3\";"), Hash: shake256.Hash{}},
		},
	}
	mux := testMux(p)
	server := httptest.NewServer(mux)
	defer server.Close()

	// Pre-populate commit cache via CommitService so GraphService can look it up.
	commitResp, err := http.Post(
		server.URL+"/buf.registry.module.v1.CommitService/GetCommits",
		"application/proto",
		bytes.NewReader(buildGetCommitsRequest("owner", "repo")),
	)
	if err != nil {
		t.Fatalf("pre-seed CommitService request failed: %v", err)
	}
	io.ReadAll(commitResp.Body)
	commitResp.Body.Close()

	testPaths := []struct {
			path string
			body []byte
		}{
			{"/buf.registry.module.v1.GraphService/GetGraph", buildV1GetGraphRequest("owner", "repo")},
			{"/buf.registry.module.v1beta1.GraphService/GetGraph", buildGetGraphRequest("owner", "repo")},
		}
		for _, tc := range testPaths {
			t.Run(tc.path, func(t *testing.T) {
				resp, err := http.Post(server.URL+tc.path, "application/proto", bytes.NewReader(tc.body))
			if err != nil {
				t.Fatalf("request failed: %v", err)
			}
			defer resp.Body.Close()

			if ct := resp.Header.Get("Content-Type"); ct != "application/proto" {
				t.Errorf("Content-Type = %q, want %q", ct, "application/proto")
			}
			if resp.StatusCode != http.StatusOK {
				respBody, _ := io.ReadAll(resp.Body)
				t.Fatalf("status = %d, want 200; body: %s", resp.StatusCode, respBody)
			}

			respBody, _ := io.ReadAll(resp.Body)
			if len(respBody) == 0 {
				t.Fatal("empty response body")
			}
		})
	}
}

func TestDownloadServiceV1ReturnsProtobuf(t *testing.T) {