Feat/bundle from raw (#101)

### Description

Add an endpoint for creating a Batch from (in-memory) raw content.
This is a pre-requisite for [AIBOM-55](https://snyksec.atlassian.net/browse/AIBOM-55), wherein the objective is to have the [aibom cli extension](https://github.com/snyk/cli-extension-ai-bom/) upload dep graphs from memory without writing them to file (as we currently do).

The unit tests have also been updated to test this new method.

### Checklist

- [✅] Tests added and all succeed
- [✅] Linted
- ~~README.md updated, if user-facing~~ (not user-facing)


[AIBOM-55]: https://snyksec.atlassian.net/browse/AIBOM-55?atlOrigin=eyJpIjoiNWRkNTljNzYxNjVmNDY3MDlhMDU5Y2ZhYzA5YTRkZjUiLCJwIjoiZ2l0aHViLWNvbS1KU1cifQ

Author: sburuiana-snyk

bundle/bundle.go

@@ -18,6 +18,7 @@ package bundle
 
 import (
 	"context"
+	"fmt"
 
 	"github.com/rs/zerolog"
 
@@ -134,6 +135,23 @@ func NewBatch(documents map[string]deepcode.BundleFile) *Batch {
 	}
 }
 
+func NewBatchFromRawContent(documents map[string][]byte) (*Batch, error) {
+	bundleFiles := make(map[string]deepcode.BundleFile)
+
+	for key, rawData := range documents {
+		bundleFile, err := deepcode.BundleFileFrom(rawData)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create file from raw data: %v", err)
+		}
+
+		bundleFiles[key] = bundleFile
+	}
+
+	return &Batch{
+		documents: bundleFiles,
+	}, nil
+}
+
 // todo simplify the size computation
 // maybe consider an addFile / canFitFile interface with proper error handling
 func (b *Batch) canFitFile(uri string, content []byte) bool {

bundle/bundle_test.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"crypto/sha256"
 	"encoding/hex"
+	"fmt"
 	"os"
 	"testing"
 
@@ -39,6 +40,29 @@ var bundleWithMultipleFiles = bundle.NewBatch(map[string]deepcode.BundleFile{
 	"file":    {},
 	"another": {},
 })
+var bundleFromRawContent, batchErr = bundle.NewBatchFromRawContent(map[string][]byte{"hello": []byte("world")})
+
+// Matcher for BundleFile that matches on key and content (ignores hash)
+type bundleFilePartialMatcher struct {
+	expectedKey     string
+	expectedContent string
+}
+
+func (m bundleFilePartialMatcher) Matches(x interface{}) bool {
+	files, ok := x.(map[string]deepcode.BundleFile)
+	if !ok {
+		return false
+	}
+	file, exists := files[m.expectedKey]
+	if !exists {
+		return false
+	}
+	return file.Content == m.expectedContent
+}
+
+func (m bundleFilePartialMatcher) String() string {
+	return fmt.Sprintf("{ Key : '%s', Content : '%s' }", m.expectedKey, m.expectedContent)
+}
 
 func Test_UploadBatch(t *testing.T) {
 	testLogger := zerolog.Nop()
@@ -108,6 +132,31 @@ func Test_UploadBatch(t *testing.T) {
 	})
 }
 
+func Test_RawContentBatch(t *testing.T) {
+	testLogger := zerolog.Nop()
+
+	t.Run("create a batch from raw content and upload the bundle", func(t *testing.T) {
+		ctrl := gomock.NewController(t)
+		mockSnykCodeClient := deepcodeMocks.NewMockDeepcodeClient(ctrl)
+		mockSnykCodeClient.EXPECT().ExtendBundle(gomock.Any(), "testBundleHash", bundleFilePartialMatcher{expectedKey: "hello", expectedContent: "world"}, []string{}).Return("newBundleHash", []string{}, nil).Times(1)
+
+		mockSpan := mocks.NewMockSpan(ctrl)
+		mockSpan.EXPECT().Context().AnyTimes()
+		mockInstrumentor := mocks.NewMockInstrumentor(ctrl)
+		mockInstrumentor.EXPECT().StartSpan(gomock.Any(), gomock.Any()).Return(mockSpan).AnyTimes()
+		mockInstrumentor.EXPECT().Finish(gomock.Any()).AnyTimes()
+		mockErrorReporter := mocks.NewMockErrorReporter(ctrl)
+		b := bundle.NewBundle(mockSnykCodeClient, mockInstrumentor, mockErrorReporter, &testLogger, "testRootPath", "testBundleHash", map[string]deepcode.BundleFile{}, []string{}, []string{})
+
+		require.NoError(t, batchErr)
+		oldHash := b.GetBundleHash()
+		err := b.UploadBatch(context.Background(), "testRequestId", bundleFromRawContent)
+		require.NoError(t, err)
+		newHash := b.GetBundleHash()
+		assert.NotEqual(t, oldHash, newHash)
+	})
+}
+
 func Test_BundleEncoding(t *testing.T) {
 	t.Run("utf-8 encoded content", func(t *testing.T) {
 		content := []byte("hello")