Examples with MockNodeRepository com.yahoo.vespa.hosted.provision.testutils.MockNodeRepository used on opensource projects

Example 1 with MockNodeRepository

use of com.yahoo.vespa.hosted.provision.testutils.MockNodeRepository in project vespa by vespa-engine.

the class AuthorizerTest method before.

@Before
public void before() {
    NodeFlavors flavors = new MockNodeFlavors();
    nodeRepository = new MockNodeRepository(new MockCurator(), flavors);
    authorizer = new Authorizer(SystemName.main, nodeRepository, () -> "cfg1");
    {
        // Populate with nodes used in this test. Note that only nodes requiring node repository lookup are added here
        Set<String> ipAddresses = new HashSet<>(Arrays.asList("127.0.0.1", "::1"));
        Flavor flavor = flavors.getFlavorOrThrow("default");
        List<Node> nodes = new ArrayList<>();
        nodes.add(nodeRepository.createNode("host1", "host1", ipAddresses, Optional.empty(), flavor, NodeType.host));
        nodes.add(nodeRepository.createNode("child1-1", "child1-1", ipAddresses, Optional.of("host1"), flavor, NodeType.tenant));
        nodes.add(nodeRepository.createNode("child1-2", "child1-2", ipAddresses, Optional.of("host1"), flavor, NodeType.tenant));
        nodes.add(nodeRepository.createNode("host2", "host2", ipAddresses, Optional.empty(), flavor, NodeType.host));
        nodes.add(nodeRepository.createNode("child2-1", "child2-1", ipAddresses, Optional.of("host1.tld"), flavor, NodeType.tenant));
        nodes.add(nodeRepository.createNode("proxy1", "proxy1", ipAddresses, Optional.empty(), flavor, NodeType.proxy));
        nodes.add(nodeRepository.createNode("proxy1-host1", "proxy1-host", ipAddresses, Optional.empty(), flavor, NodeType.proxyhost));
        nodeRepository.addNodes(nodes);
    }
}