Skip to content

Commit d5d1704

Browse files
committed
IGNITE-26966 Update testRollingUpgrade
1 parent 86afbc2 commit d5d1704

File tree

1 file changed

+18
-10
lines changed

1 file changed

+18
-10
lines changed

modules/control-utility/src/test/java/org/apache/ignite/internal/commandline/SecurityCommandHandlerPermissionsTest.java

Lines changed: 18 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -144,25 +144,33 @@ public void testCacheCreate() throws Exception {
144144
/** */
145145
@Test
146146
public void testRollingUpgrade() throws Exception {
147-
IgniteEx ign = startGrid(0);
147+
IgniteEx ign = startGrid(
148+
0,
149+
userData(TEST_NO_PERMISSIONS_LOGIN, NO_PERMISSIONS),
150+
userData(TEST_LOGIN, systemPermissions(ADMIN_ROLLING_UPGRADE))
151+
);
148152

149153
IgniteProductVersion curVer = IgniteProductVersion.fromString(ign.localNode().attribute(ATTR_BUILD_VER));
150-
151154
String targetVerStr = curVer.major() + "." + (curVer.minor() + 1) + ".0";
152155

156+
List<String> cmdArgs = asList("--rolling-upgrade", "enable", targetVerStr);
153157

154-
checkCommandPermissions(
155-
asList("--rolling-upgrade", "enable", targetVerStr),
156-
systemPermissions(ADMIN_ROLLING_UPGRADE)
157-
);
158+
assertEquals(EXIT_CODE_UNEXPECTED_ERROR, execute(enrichWithConnectionArguments(cmdArgs, TEST_NO_PERMISSIONS_LOGIN)));
159+
160+
assertFalse(ign.context().rollingUpgrade().enabled());
161+
162+
assertEquals(EXIT_CODE_OK, execute(enrichWithConnectionArguments(cmdArgs, TEST_LOGIN)));
158163

159164
assertTrue(ign.context().rollingUpgrade().enabled());
160165
assertEquals(IgniteProductVersion.fromString(targetVerStr), ign.context().rollingUpgrade().versions().get2());
161166

162-
checkCommandPermissions(
163-
asList("--rolling-upgrade", "disable"),
164-
systemPermissions(ADMIN_ROLLING_UPGRADE)
165-
);
167+
cmdArgs = asList("--rolling-upgrade", "disable");
168+
169+
assertEquals(EXIT_CODE_UNEXPECTED_ERROR, execute(enrichWithConnectionArguments(cmdArgs, TEST_NO_PERMISSIONS_LOGIN)));
170+
171+
assertTrue(ign.context().rollingUpgrade().enabled());
172+
173+
assertEquals(EXIT_CODE_OK, execute(enrichWithConnectionArguments(cmdArgs, TEST_LOGIN)));
166174

167175
assertFalse(ign.context().rollingUpgrade().enabled());
168176
}

0 commit comments

Comments
 (0)