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

mock-dnf-json: Add support for search command

Browse source 
The search response from mocks/dnfjson is a map of responses indexed by the
comma-separated list of packages and globs being requested. Add support
for this.
This commit is contained in:
Brian C. Lane 2022-08-22 15:06:26 -07:00 • committed by Tom Gundersen
parent 760be50069
commit 3c0fa94c69
1 changed files with 21 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

24
cmd/mock-dnf-json/dnf-json.go
View file

@ -14,6 +14,7 @@ import (
"fmt"
"io"
"os"
"strings"
"github.com/osbuild/osbuild-composer/internal/dnfjson"
)
@ -49,11 +50,28 @@ func readTestCase() string {
return os.Args[1]
}
func parseResponse(resp []byte, command string) json.RawMessage {
func parseResponse(resp []byte, req dnfjson.Request) json.RawMessage {
parsedResponse := make(map[string]json.RawMessage)
err := json.Unmarshal(resp, &parsedResponse)
maybeFail(err)
return parsedResponse[command]
if req.Command == "search" {
// Search requests need to return results based on the search
// The key to the search is a comma-separated list of the requested packages
key := strings.Join(req.Arguments.Search.Packages, ",")
// Extract the possible response map
var searches map[string]json.RawMessage
err = json.Unmarshal(parsedResponse["search"], &searches)
maybeFail(err)
if _, ok := searches[key]; !ok {
fail(fmt.Errorf("search response map is missing key = %s", key))
}
return searches[key]
} else {
return parsedResponse[req.Command]
}
}
func checkForError(msg json.RawMessage) bool {
@ -79,7 +97,7 @@ func main() {
fail(fmt.Errorf("failed to read test file %q\n", testFilePath))
}
res := parseResponse(response, req.Command)
res := parseResponse(response, req)
if req.Command == "depsolve" {
// add repo ID to packages