bug fix on return type (Azure#36950)

Author: XiaofeiCao

sdk/resourcemanager/azure-resourcemanager-compute/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "java",
   "TagPrefix": "java/resourcemanager/azure-resourcemanager-compute",
-  "Tag": "java/resourcemanager/azure-resourcemanager-compute_e09f72a36a"
+  "Tag": "java/resourcemanager/azure-resourcemanager-compute_f4e4411e9d"
 }

sdk/resourcemanager/azure-resourcemanager-compute/src/main/java/com/azure/resourcemanager/compute/implementation/VirtualMachinesImpl.java

@@ -314,24 +314,24 @@ public PagedFlux<VirtualMachine> listByVirtualMachineScaleSetIdAsync(String vmss
             listNextSinglePageAsync.setAccessible(true);
             return null;
         });
-        return new PagedFlux<>(
+        return wrapPageAsync(new PagedFlux<>(
             () -> {
                 try {
-                    return (Mono<PagedResponse<VirtualMachine>>)
+                    return (Mono<PagedResponse<VirtualMachineInner>>)
                         listSinglePageAsync.invoke(inner(), ResourceUtils.groupFromResourceId(vmssId), String.format("'virtualMachineScaleSet/id' eq '%s'", vmssId), null);
                 } catch (IllegalAccessException | InvocationTargetException e) {
                     throw new RuntimeException(e);
                 }
             },
             nextLink -> {
                 try {
-                    return (Mono<PagedResponse<VirtualMachine>>)
+                    return (Mono<PagedResponse<VirtualMachineInner>>)
                         // encode nextLink
                         listNextSinglePageAsync.invoke(inner(), ResourceUtils.encodeResourceId(nextLink), Context.NONE);
                 } catch (IllegalAccessException | InvocationTargetException e) {
                     throw new RuntimeException(e);
                 }
-            });
+            }));
     }
 
     @Override

sdk/resourcemanager/azure-resourcemanager-compute/src/test/java/com/azure/resourcemanager/compute/VirtualMachineOperationsTests.java

@@ -1753,6 +1753,7 @@ public void testListVmByVmssId() {
         Assertions.assertNull(vm2.virtualMachineScaleSetId());
 
         Assertions.assertEquals(1, computeManager.virtualMachines().listByVirtualMachineScaleSetId(vmss.id()).stream().count());
+        Assertions.assertTrue(vm.id().equalsIgnoreCase(computeManager.virtualMachines().listByVirtualMachineScaleSetId(vmss.id()).stream().iterator().next().id()));
         Assertions.assertEquals(2, computeManager.virtualMachines().listByResourceGroup(rgName).stream().count());
     }