ContentMapTest.java

package org.tenidwa.collections.utils;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import org.hamcrest.MatcherAssert;
import org.hamcrest.Matchers;
import org.junit.Test;

/**
 * Unit tests for {@link ContentMap}.
 * @author Georgy Vlasov (suseika@tendiwa.org)
 * @version $Id$
 * @since 0.6.0
 */
public final class ContentMapTest {
    @Test
    public void detectsIdenticalValues() {
        final ContentMap<Dude, String, Integer> map = new ContentMap<>(
            ImmutableMap.of(new CombinedDude("je", "ff"), 2),
            Dude::name
        );
        MatcherAssert.assertThat(
            map.containsKey(new PlainDude("jeff")),
            Matchers.is(true)
        );
        MatcherAssert.assertThat(
            map.get(new PlainDude("jeff")),
            Matchers.equalTo(2)
        );
    }

    @Test
    public void proxiesKeys() {
        final Dude jeff = new PlainDude("jeff");
        MatcherAssert.assertThat(
            new ContentMap<>(
                ImmutableMap.of(jeff, 2),
                Dude::name
            ).keySet(),
            Matchers.contains(jeff)
        );
    }

    @Test
    public void contentMapAndAugmentedMapWorkTogether() {
        final Dude jeff = new PlainDude("jeff");
        MatcherAssert.assertThat(
            new AugmentedMap<>(
                new ContentMap<>(
                    ImmutableMap.of(jeff, 4),
                    Dude::name
                ),
                ImmutableSet.of(jeff, new PlainDude("mac")),
                dude -> dude.name().length()
            ),
            Matchers.hasKey(jeff)
        );
    }

    private interface Dude {
        String name();
    }

    private final static class PlainDude implements Dude {
        private final String name;

        PlainDude(final String name) {
            this.name = name;
        }

        @Override
        public String name() {
            return this.name;
        }
    }

    private final static class CombinedDude implements Dude {
        private final String one;
        private final String two;

        @Override
        public String name() {
            return this.one + this.two;
        }

        CombinedDude(final String one, final String two) {
            this.one = one;
            this.two = two;
        }
    }
}