mirror of
https://codeberg.org/Mercury-IM/Smack
synced 2024-12-23 21:17:58 +01:00
c06b0a7720
Roster versioning is defined in RFC 6121, section 2.2.6; the protocol was originally described in XEP-0237. Fixes SMACK-399
222 lines
8.4 KiB
Java
222 lines
8.4 KiB
Java
/**
|
|
* All rights reserved. Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package org.jivesoftware.smack;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertNotNull;
|
|
import static org.junit.Assert.assertNull;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import org.jivesoftware.smack.packet.RosterPacket;
|
|
import org.jivesoftware.smack.packet.RosterPacket.Item;
|
|
import org.jivesoftware.smack.packet.RosterPacket.ItemStatus;
|
|
import org.jivesoftware.smack.packet.RosterPacket.ItemType;
|
|
import org.junit.After;
|
|
import org.junit.Before;
|
|
import org.junit.Rule;
|
|
import org.junit.Test;
|
|
import org.junit.rules.TemporaryFolder;
|
|
|
|
/**
|
|
* Tests the implementation of {@link DefaultRosterStore}.
|
|
*
|
|
* @author Lars Noschinski
|
|
*/
|
|
public class DefaultRosterStoreTest {
|
|
|
|
@Rule
|
|
public TemporaryFolder tmpFolder = new TemporaryFolder();
|
|
|
|
@Before
|
|
public void setUp() throws Exception {
|
|
}
|
|
|
|
@After
|
|
public void tearDown() throws Exception {
|
|
}
|
|
|
|
/**
|
|
* Tests that opening an uninitialized directory fails.
|
|
*/
|
|
@Test
|
|
public void testStoreUninitialized() throws IOException {
|
|
File storeDir = tmpFolder.newFolder();
|
|
assertNull(DefaultRosterStore.open(storeDir));
|
|
}
|
|
|
|
/**
|
|
* Tests that an initialized directory is empty.
|
|
*/
|
|
@Test
|
|
public void testStoreInitializedEmpty() throws IOException {
|
|
File storeDir = tmpFolder.newFolder();
|
|
DefaultRosterStore store = DefaultRosterStore.init(storeDir);
|
|
assertNotNull("Initialization returns store", store);
|
|
assertEquals("Freshly initialized store must have empty version",
|
|
"", store.getRosterVersion());
|
|
assertEquals("Freshly initialized store must have no entries",
|
|
0, store.getEntries().size());
|
|
}
|
|
|
|
/**
|
|
* Tests adding and removing entries
|
|
*/
|
|
@Test
|
|
public void testStoreAddRemove() throws IOException {
|
|
File storeDir = tmpFolder.newFolder();
|
|
DefaultRosterStore store = DefaultRosterStore.init(storeDir);
|
|
|
|
assertEquals("Initial roster version", "", store.getRosterVersion());
|
|
|
|
String userName = "user@example.com";
|
|
|
|
final RosterPacket.Item item1 = new Item(userName, null);
|
|
final String version1 = "1";
|
|
store.addEntry(item1, version1);
|
|
|
|
assertEquals("Adding entry sets version correctly", version1, store.getRosterVersion());
|
|
|
|
RosterPacket.Item storedItem = store.getEntry(userName);
|
|
assertNotNull("Added entry not found found", storedItem);
|
|
assertEquals("User of added entry",
|
|
item1.getUser(), storedItem.getUser());
|
|
assertEquals("Name of added entry",
|
|
item1.getName(), storedItem.getName());
|
|
assertEquals("Groups", item1.getGroupNames(), storedItem.getGroupNames());
|
|
assertEquals("ItemType of added entry",
|
|
item1.getItemType(), storedItem.getItemType());
|
|
assertEquals("ItemStatus of added entry",
|
|
item1.getItemStatus(), storedItem.getItemStatus());
|
|
|
|
|
|
final String version2 = "2";
|
|
final RosterPacket.Item item2 = new Item(userName, "Ursula Example");
|
|
item2.addGroupName("users");
|
|
item2.addGroupName("examples");
|
|
item2.setItemStatus(ItemStatus.subscribe);
|
|
item2.setItemType(ItemType.none);
|
|
store.addEntry(item2,version2);
|
|
assertEquals("Updating entry sets version correctly", version2, store.getRosterVersion());
|
|
storedItem = store.getEntry(userName);
|
|
assertNotNull("Added entry not found", storedItem);
|
|
assertEquals("User of added entry",
|
|
item2.getUser(), storedItem.getUser());
|
|
assertEquals("Name of added entry",
|
|
item2.getName(), storedItem.getName());
|
|
assertEquals("Groups", item2.getGroupNames(), storedItem.getGroupNames());
|
|
assertEquals("ItemType of added entry",
|
|
item2.getItemType(), storedItem.getItemType());
|
|
assertEquals("ItemStatus of added entry",
|
|
item2.getItemStatus(), storedItem.getItemStatus());
|
|
|
|
List<Item> entries = store.getEntries();
|
|
assertEquals("Number of entries", 1, entries.size());
|
|
|
|
final RosterPacket.Item item3 = new Item("foobar@example.com", "Foo Bar");
|
|
item3.addGroupName("The Foo Fighters");
|
|
item3.addGroupName("Bar Friends");
|
|
item3.setItemStatus(ItemStatus.unsubscribe);
|
|
item3.setItemType(ItemType.both);
|
|
|
|
final RosterPacket.Item item4 = new Item("baz@example.com", "Baba Baz");
|
|
item4.addGroupName("The Foo Fighters");
|
|
item4.addGroupName("Bar Friends");
|
|
item4.setItemStatus(ItemStatus.subscribe);
|
|
item4.setItemType(ItemType.both);
|
|
|
|
ArrayList<Item> items34 = new ArrayList<RosterPacket.Item>();
|
|
items34.add(item3);
|
|
items34.add(item4);
|
|
|
|
String version3 = "3";
|
|
store.resetEntries(items34, version3);
|
|
|
|
storedItem = store.getEntry("foobar@example.com");
|
|
assertNotNull("Added entry not found", storedItem);
|
|
assertEquals("User of added entry",
|
|
item3.getUser(), storedItem.getUser());
|
|
assertEquals("Name of added entry",
|
|
item3.getName(), storedItem.getName());
|
|
assertEquals("Groups", item3.getGroupNames(), storedItem.getGroupNames());
|
|
assertEquals("ItemType of added entry",
|
|
item3.getItemType(), storedItem.getItemType());
|
|
assertEquals("ItemStatus of added entry",
|
|
item3.getItemStatus(), storedItem.getItemStatus());
|
|
|
|
|
|
storedItem = store.getEntry("baz@example.com");
|
|
assertNotNull("Added entry not found", storedItem);
|
|
assertEquals("User of added entry",
|
|
item4.getUser(), storedItem.getUser());
|
|
assertEquals("Name of added entry",
|
|
item4.getName(), storedItem.getName());
|
|
assertEquals("Groups", item4.getGroupNames(), storedItem.getGroupNames());
|
|
assertEquals("ItemType of added entry",
|
|
item4.getItemType(), storedItem.getItemType());
|
|
assertEquals("ItemStatus of added entry",
|
|
item4.getItemStatus(), storedItem.getItemStatus());
|
|
|
|
entries = store.getEntries();
|
|
assertEquals("Number of entries", 2, entries.size());
|
|
|
|
String version4 = "4";
|
|
store.removeEntry("baz@example.com", version4);
|
|
assertEquals("Removing entry sets version correctly",
|
|
version4, store.getRosterVersion());
|
|
assertNull("Removed entry is gone", store.getEntry(userName));
|
|
|
|
entries = store.getEntries();
|
|
assertEquals("Number of entries", 1, entries.size());
|
|
}
|
|
|
|
/**
|
|
* Tests adding entries with evil characters
|
|
*/
|
|
@Test
|
|
public void testAddEvilChars() throws IOException {
|
|
File storeDir = tmpFolder.newFolder();
|
|
DefaultRosterStore store = DefaultRosterStore.init(storeDir);
|
|
|
|
String user = "../_#;\"'\\&@example.com";
|
|
String name = "\n../_#\0\t;\"'&@\\";
|
|
String group1 = "\t;\"'&@\\\n../_#\0";
|
|
String group2 = "#\0\t;\"'&@\\\n../_";
|
|
|
|
Item item = new Item(user, name);
|
|
item.setItemStatus(ItemStatus.unsubscribe);
|
|
item.setItemType(ItemType.to);
|
|
item.addGroupName(group1);
|
|
item.addGroupName(group2);
|
|
store.addEntry(item, "a-version");
|
|
Item storedItem = store.getEntry(user);
|
|
|
|
assertNotNull("Added entry not found", storedItem);
|
|
assertEquals("User of added entry",
|
|
item.getUser(), storedItem.getUser());
|
|
assertEquals("Name of added entry",
|
|
item.getName(), storedItem.getName());
|
|
assertEquals("Groups", item.getGroupNames(), storedItem.getGroupNames());
|
|
assertEquals("ItemType of added entry",
|
|
item.getItemType(), storedItem.getItemType());
|
|
assertEquals("ItemStatus of added entry",
|
|
item.getItemStatus(), storedItem.getItemStatus());
|
|
|
|
}
|
|
}
|
|
|