configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/TestTenantBuilder.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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.http.v2;

import com.google.common.base.Function;
import com.google.common.collect.Collections2;
import com.yahoo.config.provision.TenantName;
import com.yahoo.path.Path;
import com.yahoo.vespa.config.server.*;
import com.yahoo.vespa.config.server.application.MemoryTenantApplications;
import com.yahoo.vespa.config.server.monitoring.Metrics;
import com.yahoo.vespa.config.server.session.LocalSessionRepo;
import com.yahoo.vespa.config.server.session.RemoteSessionRepo;
import com.yahoo.vespa.config.server.tenant.Tenant;
import com.yahoo.vespa.config.server.tenant.TenantBuilder;
import com.yahoo.vespa.config.server.tenant.Tenants;

import java.util.*;

/**
 * Test utility for creating tenants used for testing and setup wiring of tenant stuff.
 *
 * @author lulf
 * @since 5.1
 */
public class TestTenantBuilder {

    private GlobalComponentRegistry componentRegistry;
    private Map<TenantName, TenantBuilder> tenantMap = new HashMap<>();

    public TestTenantBuilder() {
        componentRegistry = new TestComponentRegistry.Builder().build();
    }

    public TenantBuilder createTenant(TenantName tenantName) {
        MemoryTenantApplications applicationRepo = new MemoryTenantApplications();
        TenantBuilder builder = TenantBuilder.create(componentRegistry, tenantName, Path.createRoot().append(tenantName.value()))
                .withSessionFactory(new SessionCreateHandlerTest.MockSessionFactory())
                .withLocalSessionRepo(new LocalSessionRepo(applicationRepo))
                .withRemoteSessionRepo(new RemoteSessionRepo())
                .withApplicationRepo(applicationRepo);
        tenantMap.put(tenantName, builder);
        return builder;
    }

    public Map<TenantName, TenantBuilder> tenants() {
        return Collections.unmodifiableMap(tenantMap);
    }

    public Tenants createTenants() {
        Collection<Tenant> tenantList = Collections2.transform(tenantMap.values(), new Function<TenantBuilder, Tenant>() {
            @Override
            public Tenant apply(TenantBuilder builder) {
                try {
                    return builder.build();
                } catch (Exception e) {
                    throw new IllegalArgumentException("Unable to build tenant", e);
                }
            }
        });
        return new Tenants(componentRegistry, Metrics.createTestMetrics(), tenantList);
    }
}