[FLINK-12812] [runtime] (follow-up) Test refers to slot profile compu…

…ted in ResourceManager.

This helps catch a previous bug in which the test re-computed the managed memory, passing a test value
in MEGABYTES. The original call in the ResourceManager passed a value in BYTES.

flink-yarn/src/main/java/org/apache/flink/yarn/YarnResourceManager.java

@@ -323,6 +323,11 @@ Resource getContainerResource() {
 		return resource;
 	}
 
+	@VisibleForTesting
+	Collection<ResourceProfile> getSlotsPerWorker() {
+		return slotsPerWorker;
+	}
+
 	@Override
 	public boolean stopWorker(final YarnWorkerNode workerNode) {
 		final Container container = workerNode.getContainer();

flink-yarn/src/test/java/org/apache/flink/yarn/YarnResourceManagerTest.java

@@ -45,7 +45,6 @@
 import org.apache.flink.runtime.metrics.groups.UnregisteredMetricGroups;
 import org.apache.flink.runtime.registration.RegistrationResponse;
 import org.apache.flink.runtime.resourcemanager.JobLeaderIdService;
-import org.apache.flink.runtime.resourcemanager.ResourceManager;
 import org.apache.flink.runtime.resourcemanager.ResourceManagerGateway;
 import org.apache.flink.runtime.resourcemanager.SlotRequest;
 import org.apache.flink.runtime.resourcemanager.slotmanager.SlotManager;
@@ -564,10 +563,7 @@ private void testCreateSlotsPerWorker(Configuration config, Resource resource) t
 		new Context(config) {{
 			runTest(() -> {
 
-				ResourceProfile rmCalculatedResourceProfile =
-					ResourceManager.updateTaskManagerConfigAndCreateWorkerSlotProfiles(
-						config, resource.getMemory(), config.getInteger(TaskManagerOptions.NUM_TASK_SLOTS))
-								.iterator().next();
+				ResourceProfile rmCalculatedResourceProfile = resourceManager.getSlotsPerWorker().iterator().next();
 
 				ResourceProfile tmCalculatedResourceProfile =
 					TaskManagerServices.computeSlotResourceProfile(