|
32 | 32 | import static org.junit.Assert.assertThat;
|
33 | 33 | import static org.junit.Assert.fail;
|
34 | 34 |
|
| 35 | +import java.util.ArrayList; |
35 | 36 | import java.util.Arrays;
|
36 | 37 | import java.util.Collection;
|
37 | 38 | import java.util.HashMap;
|
| 39 | +import java.util.List; |
38 | 40 | import java.util.Map;
|
39 | 41 |
|
| 42 | +import org.hamcrest.Matcher; |
40 | 43 | import org.junit.Test;
|
41 | 44 | import org.junit.runner.RunWith;
|
42 | 45 | import org.junit.runners.Parameterized;
|
|
60 | 63 |
|
61 | 64 | /**
|
62 | 65 | * @author Mark Vollmary
|
63 |
| - * |
| 66 | + * @author Reşat SABIQ |
64 | 67 | */
|
65 | 68 | @RunWith(Parameterized.class)
|
66 | 69 | public class ArangoDBTest {
|
@@ -191,12 +194,23 @@ public void getUsersOnlyRoot() {
|
191 | 194 | @Test
|
192 | 195 | public void getUsers() {
|
193 | 196 | try {
|
| 197 | + // Allow & account for pre-existing users other than ROOT: |
| 198 | + final Collection<UserEntity> initialUsers = arangoDB.getUsers(); |
| 199 | + |
194 | 200 | arangoDB.createUser(USER, PW, null);
|
195 | 201 | final Collection<UserEntity> users = arangoDB.getUsers();
|
196 | 202 | assertThat(users, is(notNullValue()));
|
197 |
| - assertThat(users.size(), is(2)); |
| 203 | + assertThat(users.size(), is(initialUsers.size()+1)); |
| 204 | + |
| 205 | + List<Matcher<? super String>> matchers = new ArrayList<Matcher<? super String>>(users.size()); |
| 206 | + // Add initial users, including root: |
| 207 | + for (final UserEntity userEntity : initialUsers) |
| 208 | + matchers.add(is(userEntity.getUser())); |
| 209 | + // Add USER: |
| 210 | + matchers.add(is(USER)); |
| 211 | + |
198 | 212 | for (final UserEntity user : users) {
|
199 |
| - assertThat(user.getUser(), anyOf(is(ROOT), is(USER))); |
| 213 | + assertThat(user.getUser(), anyOf(matchers)); |
200 | 214 | }
|
201 | 215 | } finally {
|
202 | 216 | arangoDB.deleteUser(USER);
|
|
0 commit comments