edgehog-device-manager · davidebriani · Jan 9, 2025 · Jan 9, 2025
diff --git a/backend/lib/edgehog/devices/devices.ex b/backend/lib/edgehog/devices/devices.ex
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule Edgehog.Devices do
   @moduledoc """
@@ -60,7 +58,8 @@ defmodule Edgehog.Devices do
 
       list SystemModel, :system_models, :read do
         description "Returns a list of system models."
-        paginate_with nil
+        relay? true
+        paginate_with :keyset
       end
     end
 

diff --git a/backend/lib/edgehog/devices/system_model/system_model.ex b/backend/lib/edgehog/devices/system_model/system_model.ex
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule Edgehog.Devices.SystemModel do
   @moduledoc false
@@ -43,6 +41,8 @@ defmodule Edgehog.Devices.SystemModel do
 
   graphql do
     type :system_model
+
+    paginate_relationship_with part_numbers: :relay
   end
 
   actions do

diff --git a/backend/lib/edgehog/devices/system_model_part_number.ex b/backend/lib/edgehog/devices/system_model_part_number.ex
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule Edgehog.Devices.SystemModelPartNumber do
   @moduledoc false
@@ -28,6 +26,8 @@ defmodule Edgehog.Devices.SystemModelPartNumber do
 
   graphql do
     type :system_model_part_number
+
+    paginate_relationship_with devices: :relay
   end
 
   actions do

diff --git a/backend/test/edgehog_web/schema/mutation/create_system_model_test.exs b/backend/test/edgehog_web/schema/mutation/create_system_model_test.exs
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule EdgehogWeb.Schema.Mutation.CreateSystemModelTest do
   use EdgehogWeb.GraphqlCase, async: true
@@ -49,15 +47,15 @@ defmodule EdgehogWeb.Schema.Mutation.CreateSystemModelTest do
                "name" => "Foobar",
                "handle" => "foobar",
                "pictureUrl" => nil,
-               "partNumbers" => part_numbers,
+               "partNumbers" => %{"edges" => part_numbers},
                "hardwareType" => %{
                  "id" => ^hardware_type_id
                }
              } = system_model
 
       assert length(part_numbers) == 2
-      assert %{"partNumber" => "123"} in part_numbers
-      assert %{"partNumber" => "456"} in part_numbers
+      assert %{"node" => %{"partNumber" => "123"}} in part_numbers
+      assert %{"node" => %{"partNumber" => "456"}} in part_numbers
     end
 
     test "allows passing localized descriptions", %{tenant: tenant} do
@@ -233,7 +231,11 @@ defmodule EdgehogWeb.Schema.Mutation.CreateSystemModelTest do
           handle
           pictureUrl
           partNumbers {
-            partNumber
+            edges {
+              node {
+                partNumber
+              }
+            }
           }
           hardwareType {
             id

diff --git a/backend/test/edgehog_web/schema/mutation/update_system_model_test.exs b/backend/test/edgehog_web/schema/mutation/update_system_model_test.exs
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule EdgehogWeb.Schema.Mutation.UpdateSystemModelTest do
   use EdgehogWeb.GraphqlCase, async: true
@@ -57,9 +55,13 @@ defmodule EdgehogWeb.Schema.Mutation.UpdateSystemModelTest do
                "id" => _id,
                "name" => "Updated Name",
                "handle" => "updatedhandle",
-               "partNumbers" => [
-                 %{"partNumber" => "updated-1234"}
-               ]
+               "partNumbers" => %{
+                 "edges" => [
+                   %{
+                     "node" => %{"partNumber" => "updated-1234"}
+                   }
+                 ]
+               }
              } = system_model
     end
 
@@ -78,13 +80,15 @@ defmodule EdgehogWeb.Schema.Mutation.UpdateSystemModelTest do
       assert %{
                "name" => "Only Name Update",
                "handle" => ^old_handle,
-               "partNumbers" => part_numbers
+               "partNumbers" => %{
+                 "edges" => part_numbers
+               }
              } = system_model
 
       assert length(part_numbers) == length(old_part_numbers)
 
       Enum.each(old_part_numbers, fn pn ->
-        assert %{"partNumber" => pn} in part_numbers
+        assert %{"node" => %{"partNumber" => pn}} in part_numbers
       end)
     end
 
@@ -100,10 +104,10 @@ defmodule EdgehogWeb.Schema.Mutation.UpdateSystemModelTest do
 
       system_model = extract_result!(result)
 
-      assert %{"partNumbers" => part_numbers} = system_model
+      assert %{"partNumbers" => %{"edges" => part_numbers}} = system_model
       assert length(part_numbers) == 2
-      assert %{"partNumber" => "B"} in part_numbers
-      assert %{"partNumber" => "D"} in part_numbers
+      assert %{"node" => %{"partNumber" => "B"}} in part_numbers
+      assert %{"node" => %{"partNumber" => "D"}} in part_numbers
     end
 
     test "allows updating localized descriptions", %{tenant: tenant} do
@@ -387,7 +391,11 @@ defmodule EdgehogWeb.Schema.Mutation.UpdateSystemModelTest do
           handle
           pictureUrl
           partNumbers {
-            partNumber
+            edges {
+              node {
+                partNumber
+              }
+            }
           }
         }
       }

diff --git a/backend/test/edgehog_web/schema/query/device_test.exs b/backend/test/edgehog_web/schema/query/device_test.exs
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule EdgehogWeb.Schema.Query.DeviceTest do
   use EdgehogWeb.GraphqlCase, async: true
@@ -58,7 +56,11 @@ defmodule EdgehogWeb.Schema.Query.DeviceTest do
           systemModel {
             id
             partNumbers {
-              partNumber
+              edges {
+                node {
+                  partNumber
+                }
+              }
             }
           }
         }
@@ -71,7 +73,10 @@ defmodule EdgehogWeb.Schema.Query.DeviceTest do
         |> extract_result!()
 
       assert device["systemModel"]["id"] == system_model_id
-      assert device["systemModel"]["partNumbers"] == [%{"partNumber" => part_number}]
+
+      assert device["systemModel"]["partNumbers"]["edges"] == [
+               %{"node" => %{"partNumber" => part_number}}
+             ]
     end
 
     test "queries associated OTA operations", %{tenant: tenant} do

diff --git a/backend/test/edgehog_web/schema/query/devices_test.exs b/backend/test/edgehog_web/schema/query/devices_test.exs
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule EdgehogWeb.Schema.Query.DevicesTest do
   use EdgehogWeb.GraphqlCase, async: true
@@ -57,27 +55,36 @@ defmodule EdgehogWeb.Schema.Query.DevicesTest do
           systemModel {
             id
             partNumbers {
-              partNumber
+              edges {
+                node {
+                  partNumber
+                }
+              }
             }
           }
         }
       }
       """
 
-      devices =
+      system_models =
         [document: document, tenant: tenant]
         |> devices_query()
         |> extract_result!()
+        |> Enum.flat_map(fn device ->
+          system_model = get_in(device, ["systemModel", "partNumbers", "edges"])
 
-      assert Enum.count(devices, fn device ->
-               %{"partNumber" => part_number_1} in (device["systemModel"]["partNumbers"] || [])
+          if system_model,
+            do: system_model,
+            else: []
+        end)
+
+      assert Enum.count(system_models, fn element ->
+               element == %{"node" => %{"partNumber" => part_number_1}}
              end) == 1
 
-      assert Enum.count(devices, fn device ->
-               %{"partNumber" => part_number_2} in (device["systemModel"]["partNumbers"] || [])
+      assert Enum.count(system_models, fn element ->
+               element == %{"node" => %{"partNumber" => part_number_2}}
              end) == 2
-
-      assert Enum.count(devices, fn device -> device["systemModel"] == nil end) == 1
     end
 
     test "allows filtering", %{tenant: tenant} do

diff --git a/backend/test/edgehog_web/schema/query/system_model_test.exs b/backend/test/edgehog_web/schema/query/system_model_test.exs
@@ -1,7 +1,6 @@
-#
 # This file is part of Edgehog.
 #
-# Copyright 2021-2024 SECO Mind Srl
+# Copyright 2021 - 2025 SECO Mind Srl
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +15,6 @@
 # limitations under the License.
 #
 # SPDX-License-Identifier: Apache-2.0
-#
 
 defmodule EdgehogWeb.Schema.Query.SystemModelTest do
   use EdgehogWeb.GraphqlCase, async: true
@@ -46,10 +44,10 @@ defmodule EdgehogWeb.Schema.Query.SystemModelTest do
       assert system_model["name"] == fixture.name
       assert system_model["handle"] == fixture.handle
       assert system_model["pictureUrl"] == fixture.picture_url
-      assert length(system_model["partNumbers"]) == length(fixture.part_number_strings)
+      assert length(system_model["partNumbers"]["edges"]) == length(fixture.part_number_strings)
 
       Enum.each(fixture.part_number_strings, fn pn ->
-        assert(%{"partNumber" => pn} in system_model["partNumbers"])
+        assert(%{"node" => %{"partNumber" => pn}} in system_model["partNumbers"]["edges"])
       end)
 
       assert system_model["hardwareType"]["id"] ==
@@ -164,7 +162,11 @@ defmodule EdgehogWeb.Schema.Query.SystemModelTest do
         handle
         pictureUrl
         partNumbers {
-          partNumber
+          edges {
+            node {
+              partNumber
+            }
+          }
         }
         hardwareType {
           id