diff options
author | John McDonnell <jmcdonnell@gitlab.com> | 2023-02-21 18:30:50 +0300 |
---|---|---|
committer | John McDonnell <jmcdonnell@gitlab.com> | 2023-02-21 18:30:50 +0300 |
commit | 0f1861320a6e31ddb5a6c4ac5bffc2b0f46e4047 (patch) | |
tree | 38f5fe58da9ab2084b254543221772eb5c488c6c | |
parent | 9c1e1d399a8f01d6a41bdd6d540a06fe57c88f97 (diff) |
Add blobService/list_all_blobs smoke test
-rw-r--r-- | internal/qa/blobService/list_all_blobs_test.go | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/internal/qa/blobService/list_all_blobs_test.go b/internal/qa/blobService/list_all_blobs_test.go new file mode 100644 index 000000000..1af09bdc5 --- /dev/null +++ b/internal/qa/blobService/list_all_blobs_test.go @@ -0,0 +1,62 @@ +package blobService + +import ( + "context" + "fmt" + . "github.com/onsi/ginkgo/v2" + . "github.com/onsi/gomega" + "gitlab.com/gitlab-org/gitaly/v15/proto/go/gitalypb" + "google.golang.org/grpc" +) + +var _ = Describe("BlobService", func() { + var ( + client gitalypb.BlobServiceClient + conn *grpc.ClientConn + err error + ) + + BeforeEach(func() { + //socketPath := filepath.Join(cfg.Storages[0].Path, "/Users/john/dev/gdk", "gitaly.socket") + conn, err = grpc.Dial("gdk.test:9999", grpc.WithInsecure()) + Expect(err).NotTo(HaveOccurred()) + + client = gitalypb.NewBlobServiceClient(conn) + }) + + AfterEach(func() { + conn.Close() + }) + + It("should list all blobs for a repository", func() { + repo := &gitalypb.Repository{ + StorageName: "default", + RelativePath: "@hashed/2f/ca/2fca346db656187102ce806ac732e06a62df0dbb2829e511a770556d398e1a6e.git", + } + ctx, cancel := context.WithCancel(context.Background()) + defer cancel() + + request := &gitalypb.ListAllBlobsRequest{ + Repository: repo, + } + stream, err := client.ListAllBlobs(ctx, request) + + var response *gitalypb.ListAllBlobsResponse + var blobs []*gitalypb.ListAllBlobsResponse_Blob + var count int + for { + response, err = stream.Recv() + if err != nil { + break + } + count += len(response.Blobs) + blobs = append(blobs, response.Blobs...) + } + + for i, blob := range blobs { + fmt.Println("BLOB FOUND", i, blob) + } + + Expect(count).To(Equal(3)) + }) +}) |