Tests generated by Copilot

Author: S7evinK

fclient/federationclient_test.go

@@ -282,3 +282,111 @@ func jsonify(x interface{}) string {
 	b, _ := json.Marshal(x)
 	return string(b)
 }
+
+func TestQueryKeysReturnsDeviceKeys(t *testing.T) {
+	serverName := spec.ServerName("local.server.name")
+	targetServerName := spec.ServerName("target.server.name")
+	keyID := gomatrixserverlib.KeyID("ed25519:auto")
+	_, privateKey, _ := ed25519.GenerateKey(nil)
+
+	respQueryKeysJSON := []byte(`{
+		"device_keys": {
+			"@user:target.server.name": {
+				"device1": {
+					"algorithms": ["m.olm.curve25519-aes-sha2"],
+					"keys": {
+						"curve25519:device1": "key1",
+						"ed25519:device1": "key2"
+					}
+				}
+			}
+		}
+	}`)
+
+	fc := fclient.NewFederationClient(
+		[]*fclient.SigningIdentity{
+			{
+				ServerName: serverName,
+				KeyID:      keyID,
+				PrivateKey: privateKey,
+			},
+		},
+		fclient.WithSkipVerify(true),
+		fclient.WithTransport(
+			&roundTripper{
+				fn: func(req *http.Request) (*http.Response, error) {
+					if strings.HasPrefix(req.URL.Path, "/_matrix/federation/v1/user/keys/query") {
+						return &http.Response{
+							StatusCode: 200,
+							Body:       io.NopCloser(bytes.NewReader(respQueryKeysJSON)),
+						}, nil
+					}
+					return &http.Response{
+						StatusCode: 404,
+						Body:       io.NopCloser(strings.NewReader("404 not found")),
+					}, nil
+				},
+			},
+		),
+	)
+
+	keys := map[string][]string{
+		"@user:target.server.name": {"device1"},
+	}
+	res, err := fc.QueryKeys(context.Background(), serverName, targetServerName, keys)
+	if err != nil {
+		t.Fatalf("QueryKeys returned an error: %s", err)
+	}
+	if len(res.DeviceKeys["@user:target.server.name"]["device1"].Keys) == 0 {
+		t.Fatalf("QueryKeys response missing device keys")
+	}
+}
+
+func TestQueryKeysHandlesNilDeviceKeys(t *testing.T) {
+	serverName := spec.ServerName("local.server.name")
+	targetServerName := spec.ServerName("target.server.name")
+	keyID := gomatrixserverlib.KeyID("ed25519:auto")
+	_, privateKey, _ := ed25519.GenerateKey(nil)
+
+	respQueryKeysJSON := []byte(`{
+		"device_keys": {}
+	}`)
+
+	fc := fclient.NewFederationClient(
+		[]*fclient.SigningIdentity{
+			{
+				ServerName: serverName,
+				KeyID:      keyID,
+				PrivateKey: privateKey,
+			},
+		},
+		fclient.WithSkipVerify(true),
+		fclient.WithTransport(
+			&roundTripper{
+				fn: func(req *http.Request) (*http.Response, error) {
+					if strings.HasPrefix(req.URL.Path, "/_matrix/federation/v1/user/keys/query") {
+						return &http.Response{
+							StatusCode: 200,
+							Body:       io.NopCloser(bytes.NewReader(respQueryKeysJSON)),
+						}, nil
+					}
+					return &http.Response{
+						StatusCode: 404,
+						Body:       io.NopCloser(strings.NewReader("404 not found")),
+					}, nil
+				},
+			},
+		),
+	)
+
+	keys := map[string][]string{
+		"@user:target.server.name": nil,
+	}
+	res, err := fc.QueryKeys(context.Background(), serverName, targetServerName, keys)
+	if err != nil {
+		t.Fatalf("QueryKeys returned an error: %s", err)
+	}
+	if len(res.DeviceKeys) != 0 {
+		t.Fatalf("QueryKeys response should be empty for nil device keys")
+	}
+}