jdisc_http_service/src/test/java/com/yahoo/jdisc/http/guiceModules/ConnectorFactoryRegistryModule.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
63
64
65
66
67
68

// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.jdisc.http.guiceModules;

import com.google.inject.Binder;
import com.google.inject.Module;
import com.google.inject.Provides;
import com.yahoo.component.ComponentId;
import com.yahoo.component.provider.ComponentRegistry;
import com.yahoo.jdisc.http.ConnectorConfig;
import com.yahoo.jdisc.http.ConnectorConfig.Builder;

import com.yahoo.jdisc.http.server.jetty.ConnectorFactory;
import com.yahoo.jdisc.http.server.jetty.TestDrivers;
import com.yahoo.jdisc.http.ssl.DefaultSslKeyStoreConfigurator;
import com.yahoo.jdisc.http.ssl.DefaultSslTrustStoreConfigurator;

/**
 * Guice module for test ConnectorFactories
 *
 * @author Tony Vaagenes
 */
public class ConnectorFactoryRegistryModule implements Module {

    private final Builder connectorConfigBuilder;

    public ConnectorFactoryRegistryModule(Builder connectorConfigBuilder) {
        this.connectorConfigBuilder = connectorConfigBuilder;
    }

    public ConnectorFactoryRegistryModule() {
        this(new Builder());
    }

    @Provides
    public ComponentRegistry<ConnectorFactory> connectorFactoryComponentRegistry() {
        ComponentRegistry<ConnectorFactory> registry = new ComponentRegistry<>();
        registry.register(ComponentId.createAnonymousComponentId("connector-factory"),
                new StaticKeyDbConnectorFactory(new ConnectorConfig(connectorConfigBuilder)));

        registry.freeze();
        return registry;
    }

    @Override
    public void configure(Binder binder) {
    }

    private static class StaticKeyDbConnectorFactory extends ConnectorFactory {

        public StaticKeyDbConnectorFactory(ConnectorConfig connectorConfig) {
            super(connectorConfig,
                  new DefaultSslKeyStoreConfigurator(connectorConfig, new MockSecretStore()),
                  new DefaultSslTrustStoreConfigurator(connectorConfig, new MockSecretStore()));
        }

    }

    @SuppressWarnings("deprecation")
    private static final class MockSecretStore implements com.yahoo.jdisc.http.SecretStore {

        @Override
        public String getSecret(String key) {
            return TestDrivers.KEY_STORE_PASSWORD;
        }

    }

}