package datadog.trace.agent.test.datastreams;

import datadog.communication.ddagent.DDAgentFeaturesDiscovery;
import datadog.communication.monitor.Monitoring;

// TODO Ideally, DDAgentFeaturesDiscovery would be an interface to create a proper testable stubs
public class MockFeaturesDiscovery extends DDAgentFeaturesDiscovery {
  private final boolean supportsDataStreams;

  public MockFeaturesDiscovery(boolean supportsDataStreams) {
    super(null, Monitoring.DISABLED, null, true, true);
    this.supportsDataStreams = supportsDataStreams;
  }

  @Override
  public void discover() {}

  @Override
  public void discoverIfOutdated() {}

  @Override
  public boolean supportsDataStreams() {
    return supportsDataStreams;
  }
}
