import org.junit.Assert;
import org.junit.Test;

public class SimpleTest {

    @Test
    public void testExample() {
        SocialNetwork network = new MySocialNetwork();
        for (int i = 0; i < 6; i++) network.addPerson(i);
        network.friendRequest(0, 1);
        network.friendRequest(2, 3);
        network.friendRequest(4, 5);
        Assert.assertTrue(network.canMessage(2, 3));
        Assert.assertFalse(network.canMessage(0, 5));
        network.friendRequest(0, 2);
        Assert.assertEquals(4, network.numberOfFriends(1));
        Assert.assertTrue(network.canMessage(0, 3));
        Assert.assertFalse(network.canMessage(3, 4));
        network.friendRequest(3, 4);
        Assert.assertEquals(6, network.numberOfFriends(5));
        Assert.assertTrue(network.canMessage(1, 5));
    }

    @Test
    public void testOef66() {
        SocialNetwork network = new MySocialNetwork();
        for (int i = 0; i < 9; i++) network.addPerson(i);
        network.friendRequest(0, 2);
        network.friendRequest(3, 1);
        network.friendRequest(4, 6);
        network.friendRequest(5, 7);
        network.friendRequest(8, 4);
        network.friendRequest(5, 1);
        network.friendRequest(7, 2);
        network.friendRequest(0, 1);
        Assert.assertTrue(network.canMessage(4, 8));
        Assert.assertTrue(network.canMessage(1, 5));
        Assert.assertFalse(network.canMessage(6, 3));
        Assert.assertEquals(3, network.numberOfFriends(8));
        Assert.assertEquals(6, network.numberOfFriends(7));
    }

}