particle-os/debian-forge-composer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cloudapi: support worker server target artifact retrieval

Browse source 
In order to get the artifact location from the cloudapi, add a helper
function in the worker server.
This commit is contained in:
Sanne Raymaekers 2025-01-22 19:23:09 +01:00
parent df16f7fc63
commit 7bfcac30dd
8 changed files with 147 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

10
internal/worker/server_test.go
View file

@ -314,7 +314,8 @@ func TestUpload(t *testing.T) {
if err != nil {
t.Fatalf("error creating osbuild manifest: %v", err)
}
server := newTestServer(t, t.TempDir(), defaultConfig, true)
storeDir := t.TempDir()
server := newTestServer(t, storeDir, defaultConfig, true)
mf, err := manifest.Serialize(nil, nil, nil, nil)
if err != nil {
t.Fatalf("error creating osbuild manifest: %v", err)
@ -334,6 +335,13 @@ func TestUpload(t *testing.T) {
test.TestRoute(t, handler, false, "PUT", fmt.Sprintf("/api/worker/v1/jobs/%s/artifacts/foobar", token), `this is my artifact`, http.StatusOK, `?`)
require.Equal(t, float64(0), promtest.ToFloat64(prometheus.PendingJobs))
require.Equal(t, float64(1), promtest.ToFloat64(prometheus.RunningJobs))
err = server.FinishJob(token, []byte("{}"))
require.NoError(t, err)
path, err := server.JobArtifactLocation(jobID, "foobar")
require.NoError(t, err)
require.Equal(t, fmt.Sprintf("%s/artifacts/%s/foobar", storeDir, jobID), path)
}
func TestUploadNotAcceptingArtifacts(t *testing.T) {