wip

Author: CalamarBicefalo

internal/fileupload/client_test.go

@@ -0,0 +1,44 @@
+//go:generate go run github.com/golang/mock/mockgen -package=mocks -destination=./mocks/mock_http_client.go github.com/snyk/cli-extension-os-flows/internal/fileupload/lowlevel SealableClient
+package fileupload
+
+import (
+	"github.com/golang/mock/gomock"
+	"github.com/snyk/cli-extension-os-flows/internal/fileupload/lowlevel"
+	"github.com/snyk/cli-extension-os-flows/internal/fileupload/mocks"
+	"os"
+	"testing"
+)
+
+type File struct {
+	filename string
+	content  string
+}
+
+type FileUploadClient struct {
+}
+
+func (c FileUploadClient) UploadDir(dir os.File) (RevisionID, error) {
+
+}
+
+func Test_uploadingADirectory(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+	sealableClientMock := mocks.NewMockSealableClient(ctrl)
+	client := NewFileUploadClient(sealableClientMock)
+
+	dir := createDirWithFiles(File{filename: "file1", content: "foo"})
+
+	revisionID, err := client.UploadDir(dir)
+
+	sealableClientFake.uploadedAndSealed(File{filename: "file1", content: "foo"})
+
+}
+
+func NewFileUploadClient(client lowlevel.SealableClient) FileUploadClient {
+
+}
+
+func createDirWithFiles(file File) os.File {
+
+}

internal/fileupload/lowlevel/client.go

@@ -1,11 +1,12 @@
-package lowlevel_fileupload //nolint:revive // underscore naming is intentional for this internal package
+package lowlevel
 
 import (
 	"bytes"
 	"context"
 	"encoding/json"
 	"errors"
 	"fmt"
+	"github.com/snyk/cli-extension-os-flows/internal/fileupload"
 	"io"
 	"mime/multipart"
 	"net/http"
@@ -14,23 +15,23 @@ import (
 	"github.com/snyk/error-catalog-golang-public/snyk_errors"
 )
 
-// Client defines the interface for file upload API operations.
-type Client interface {
+// SealableClient defines the interface for file upload API operations.
+type SealableClient interface {
 	CreateRevision(ctx context.Context, orgID OrgID) (*UploadRevisionResponseBody, error)
-	UploadFiles(ctx context.Context, orgID OrgID, revisionID RevisionID, files []UploadFile) error
-	SealRevision(ctx context.Context, orgID OrgID, revisionID RevisionID) (*SealUploadRevisionResponseBody, error)
+	UploadFiles(ctx context.Context, orgID OrgID, revisionID fileupload.RevisionID, files []UploadFile) error
+	SealRevision(ctx context.Context, orgID OrgID, revisionID fileupload.RevisionID) (*SealUploadRevisionResponseBody, error)
 }
 
 // This will force go to complain if the type doesn't satisfy the interface.
-var _ Client = (*HTTPClient)(nil)
+var _ SealableClient = (*HTTPSealableClient)(nil)
 
 // Config contains configuration for the file upload client.
 type Config struct {
 	BaseURL string
 }
 
-// HTTPClient implements the Client interface for file upload operations via HTTP API.
-type HTTPClient struct {
+// HTTPSealableClient implements the SealableClient interface for file upload operations via HTTP API.
+type HTTPSealableClient struct {
 	cfg        Config
 	httpClient *http.Client
 }
@@ -48,8 +49,8 @@ const FileCountLimit = 100 // arbitrary number, will need to be re-evaluated
 const ContentType = "Content-Type"
 
 // NewClient creates a new file upload client with the given configuration and options.
-func NewClient(cfg Config, opts ...Opt) *HTTPClient {
-	c := HTTPClient{cfg, http.DefaultClient}
+func NewClient(cfg Config, opts ...Opt) *HTTPSealableClient {
+	c := HTTPSealableClient{cfg, http.DefaultClient}
 
 	for _, opt := range opts {
 		opt(&c)
@@ -59,7 +60,7 @@ func NewClient(cfg Config, opts ...Opt) *HTTPClient {
 }
 
 // CreateRevision creates a new upload revision for the specified organization.
-func (c *HTTPClient) CreateRevision(ctx context.Context, orgID OrgID) (*UploadRevisionResponseBody, error) {
+func (c *HTTPSealableClient) CreateRevision(ctx context.Context, orgID OrgID) (*UploadRevisionResponseBody, error) {
 	if orgID == uuid.Nil {
 		return nil, ErrEmptyOrgID
 	}
@@ -103,7 +104,7 @@ func (c *HTTPClient) CreateRevision(ctx context.Context, orgID OrgID) (*UploadRe
 }
 
 // UploadFiles uploads the provided files to the specified revision. It will not close the file descriptors.
-func (c *HTTPClient) UploadFiles(ctx context.Context, orgID OrgID, revisionID RevisionID, files []UploadFile) error {
+func (c *HTTPSealableClient) UploadFiles(ctx context.Context, orgID OrgID, revisionID fileupload.RevisionID, files []UploadFile) error {
 	if orgID == uuid.Nil {
 		return ErrEmptyOrgID
 	}
@@ -165,7 +166,7 @@ func (c *HTTPClient) UploadFiles(ctx context.Context, orgID OrgID, revisionID Re
 	return nil
 }
 
-func (c *HTTPClient) streamFilesToPipe(pWriter *io.PipeWriter, mpartWriter *multipart.Writer, files []UploadFile) {
+func (c *HTTPSealableClient) streamFilesToPipe(pWriter *io.PipeWriter, mpartWriter *multipart.Writer, files []UploadFile) {
 	var streamError error
 	defer func() {
 		pWriter.CloseWithError(streamError)
@@ -189,7 +190,7 @@ func (c *HTTPClient) streamFilesToPipe(pWriter *io.PipeWriter, mpartWriter *mult
 }
 
 // SealRevision seals the specified upload revision, marking it as complete.
-func (c *HTTPClient) SealRevision(ctx context.Context, orgID OrgID, revisionID RevisionID) (*SealUploadRevisionResponseBody, error) {
+func (c *HTTPSealableClient) SealRevision(ctx context.Context, orgID OrgID, revisionID fileupload.RevisionID) (*SealUploadRevisionResponseBody, error) {
 	if orgID == uuid.Nil {
 		return nil, ErrEmptyOrgID
 	}
@@ -237,7 +238,7 @@ func (c *HTTPClient) SealRevision(ctx context.Context, orgID OrgID, revisionID R
 	return &respBody, nil
 }
 
-func (c *HTTPClient) handleUnexpectedStatusCodes(body io.ReadCloser, statusCode int, status, operation string) error {
+func (c *HTTPSealableClient) handleUnexpectedStatusCodes(body io.ReadCloser, statusCode int, status, operation string) error {
 	bts, err := io.ReadAll(body)
 	if err != nil {
 		return fmt.Errorf("failed to read response body: %w", err)