container-search/src/test/java/com/yahoo/search/federation/sourceref/VirtualSourceResolverTestCase.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.federation.sourceref;

import org.junit.jupiter.api.Test;

import java.util.Set;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertSame;

/**
 * Test of VirtualSourceResolver
 *
 * @author baldersheim
 */
public class VirtualSourceResolverTestCase {
    @Test
    void testThatOriginalIsReturnedIfNoMapping() {
        var input = Set.of("a","b", "b.c");
        assertSame(input, VirtualSourceResolver.of().resolve(input));
        assertSame(input, VirtualSourceResolver.of(Set.of("x.a","x.b")).resolve(input));
    }
    @Test
    void testResolution() {
        var input = Set.of("a","b", "b.c");
        assertEquals(Set.of("a.x", "a.y", "b.c", "b.x"),
                     VirtualSourceResolver.of(Set.of("a.x","a.y", "b.x")).resolve(input));
    }
}