mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-19 00:36:38 +00:00
acec87dd45
* integrate api integrate api with stream and implement some client side * fix typo and fix validation * use real daemon address * redo the connect via address * Refactor the debug bundle generator to be ready to use from engine (#4469) * fix tests * fix lint * fix bug with stream * try refactor status 1 * fix convert fullStatus to statusOutput & add logFile * fix tests * fix tests * fix not enough arguments in call to nbstatus.ConvertToStatusOutputOverview * fix status_test * fix(engine): avoid deadlock when stopping engine during debug bundle * use atomic for lock-free * use new lock --------- Co-authored-by: Zoltan Papp <zoltan.pmail@gmail.com>
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package debug
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/netbirdio/netbird/upload-server/server"
|
|
"github.com/netbirdio/netbird/upload-server/types"
|
|
)
|
|
|
|
func TestUpload(t *testing.T) {
|
|
if os.Getenv("DOCKER_CI") == "true" {
|
|
t.Skip("Skipping upload test on docker ci")
|
|
}
|
|
testDir := t.TempDir()
|
|
testURL := "http://localhost:8080"
|
|
t.Setenv("SERVER_URL", testURL)
|
|
t.Setenv("STORE_DIR", testDir)
|
|
srv := server.NewServer()
|
|
go func() {
|
|
if err := srv.Start(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
t.Errorf("Failed to start server: %v", err)
|
|
}
|
|
}()
|
|
t.Cleanup(func() {
|
|
if err := srv.Stop(); err != nil {
|
|
t.Errorf("Failed to stop server: %v", err)
|
|
}
|
|
})
|
|
|
|
file := filepath.Join(t.TempDir(), "tmpfile")
|
|
fileContent := []byte("test file content")
|
|
err := os.WriteFile(file, fileContent, 0640)
|
|
require.NoError(t, err)
|
|
key, err := UploadDebugBundle(context.Background(), testURL+types.GetURLPath, testURL, file)
|
|
require.NoError(t, err)
|
|
id := getURLHash(testURL)
|
|
require.Contains(t, key, id+"/")
|
|
expectedFilePath := filepath.Join(testDir, key)
|
|
createdFileContent, err := os.ReadFile(expectedFilePath)
|
|
require.NoError(t, err)
|
|
require.Equal(t, fileContent, createdFileContent)
|
|
}
|