lingwei.kong/hawkbit
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Add Rollout Execution ACM Test (#2731)

Browse Source 
Signed-off-by: Avgustin Marinov <Avgustin.Marinov@bosch.com>
This commit is contained in:
Avgustin Marinov
2025-10-09 12:29:20 +03:00
committed by GitHub
parent 222ca5c826
commit 95e37268bd
1 changed files with 89 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
hawkbit-repository/hawkbit-repository-jpa/src/test/java/org/eclipse/hawkbit/repository/jpa/acm/RolloutExecutionTest.java Normal file
View File

@@ -0,0 +1,89 @@
/**
* Copyright (c) 2025 Contributors to the Eclipse Foundation
*
* This program and the accompanying materials are made
* available under the terms of the Eclipse Public License 2.0
* which is available at https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
*/
package org.eclipse.hawkbit.repository.jpa.acm;
import static org.assertj.core.api.Assertions.assertThat;
import static org.eclipse.hawkbit.im.authentication.SpPermission.CREATE_ROLLOUT;
import static org.eclipse.hawkbit.im.authentication.SpPermission.HANDLE_ROLLOUT;
import static org.eclipse.hawkbit.im.authentication.SpPermission.READ_DISTRIBUTION_SET;
import static org.eclipse.hawkbit.im.authentication.SpPermission.READ_ROLLOUT;
import static org.eclipse.hawkbit.im.authentication.SpPermission.READ_TARGET;
import static org.eclipse.hawkbit.im.authentication.SpPermission.UPDATE_TARGET;
import static org.eclipse.hawkbit.repository.test.util.SecurityContextSwitch.runAs;
import java.util.Optional;
import org.eclipse.hawkbit.repository.jpa.rollout.RolloutScheduler;
import org.eclipse.hawkbit.repository.model.DistributionSet;
import org.eclipse.hawkbit.repository.model.Rollout;
import org.junit.jupiter.api.Test;
class RolloutExecutionTest extends AbstractAccessControllerTest {
@Test
void verifyOnlyUpdatableTargetsArePartOfRolloutExecutedByScheduler() {
verify(new RolloutScheduler(rolloutHandler, systemManagement, systemSecurityContext, 1, Optional.empty())::runningRolloutScheduler);
}
@Test
void verifyOnlyUpdatableTargetsArePartOfRollout() {
verify(() -> systemSecurityContext.runAsSystem(() -> {
rolloutHandler.handleAll();
return null;
}));
}
void verify(final Runnable run) {
runAs(withAuthorities(
READ_TARGET, UPDATE_TARGET + "/type.id==" + targetType1.getId(),
READ_DISTRIBUTION_SET,
CREATE_ROLLOUT, READ_ROLLOUT, HANDLE_ROLLOUT),
() -> {
assertThat(targetManagement.findAll(UNPAGED))
.as("targetManagement#findAll operation should see all created targets")
.hasSize(3);
final Rollout rollout = createRolloutConsideringAllTargets(ds2Type2);
rolloutManagement.start(rollout.getId());
final RolloutScheduler rolloutScheduler = new RolloutScheduler(
rolloutHandler, systemManagement, systemSecurityContext, 1, Optional.empty());
rolloutScheduler.runningRolloutScheduler();
assertThat(rolloutGroupManagement.findByRollout(rollout.getId(), UNPAGED).getContent().stream()
.flatMap(rolloutGroup -> rolloutGroupManagement.findTargetsOfRolloutGroup(rolloutGroup.getId(), UNPAGED).stream()))
.as("Only updatable targets should be part of the rollout")
.containsExactly(target1Type1);
});
runAs(withAuthorities(
READ_TARGET, UPDATE_TARGET + "/type.id==" + targetType2.getId(),
READ_DISTRIBUTION_SET,
CREATE_ROLLOUT, READ_ROLLOUT, HANDLE_ROLLOUT),
() -> {
assertThat(targetManagement.findAll(UNPAGED))
.as("targetManagement#findAll operation should see all created targets")
.hasSize(3);
final Rollout rollout = createRolloutConsideringAllTargets(ds2Type2);
rolloutManagement.start(rollout.getId());
run.run();
assertThat(rolloutGroupManagement.findByRollout(rollout.getId(), UNPAGED).getContent().stream()
.flatMap(rolloutGroup -> rolloutGroupManagement.findTargetsOfRolloutGroup(rolloutGroup.getId(), UNPAGED).stream()))
.as("Only updatable targets should be part of the rollout")
.containsExactly(target2Type2, target3Type2);
});
}
private Rollout createRolloutConsideringAllTargets(final DistributionSet ds) {
return testdataFactory.createRolloutByVariables(randomString(16), "description", 5, "id==*", ds, "50", "80");
}
}