public virtual void TestSingleCollector()
{
// Tests that if a single Collector is input, it is returned (and not MultiCollector).
DummyCollector dc = new DummyCollector();
Assert.AreSame(dc, MultiCollector.Wrap(dc));
Assert.AreSame(dc, MultiCollector.Wrap(dc, null));
}
public void CannotAddDuplicatedCollectors()
{
var registry = new CollectorRegistry();
var collector = new DummyCollector("testName", "metric");
var collector1 = new DummyCollector("testName", "metric2");
registry.Add(collector);
Assert.Throws <ArgumentException>(() => registry.Add(collector1));
}
public void ShouldNotThrowOnRemoveNonRegisteredCollector()
{
var registry = new CollectorRegistry();
var collector = new DummyCollector("collector", "metric");
var res = registry.Remove(collector);
Assert.False(res);
}
public void CannotAddWithDuplicatedMetricNames(string[] first, string[] second)
{
var registry = new CollectorRegistry();
var collector1 = new DummyCollector("testName1", first);
var collector2 = new DummyCollector("testName2", second);
registry.Add(collector1);
Assert.Throws <ArgumentException>(() => registry.Add(collector2));
}
public void DoNotCallFactoryIfCollectorExists()
{
var registry = new CollectorRegistry();
var originalCollector = new DummyCollector("testName", "metric");
var fn = Substitute.For <Func <CollectorConfiguration, ICollector> >();
registry.Add(originalCollector);
var result = registry.GetOrAdd(originalCollector.Configuration, fn);
Assert.Equal(originalCollector, result);
fn.DidNotReceiveWithAnyArgs();
}
public void ShouldNotThrowOnRemoveByNameNonRegisteredCollector(string keyToRemove)
{
var registry = new CollectorRegistry();
var collector = new DummyCollector("collector", "metric");
registry.Add(collector);
var res = registry.Remove(keyToRemove);
Assert.Equal(null, res);
Assert.True(registry.TryGet("collector", out var _));
}
public void CanRemoveCollector()
{
var registry = new CollectorRegistry();
var collector = new DummyCollector("collector", "metric");
registry.Add(collector);
var collector1 = new DummyCollector("collector1", "metric1");
registry.Add(collector1);
var res = registry.Remove(collector1);
Assert.True(res);
Assert.False(registry.TryGet("collector1", out var _));
Assert.True(registry.TryGet("collector", out var _));
}
public virtual void TestCollector()
{
// Tests that the collector delegates calls to input collectors properly.
// Tests that the collector handles some null collectors well. If it
// doesn't, an NPE would be thrown.
DummyCollector[] dcs = new DummyCollector[] { new DummyCollector(), new DummyCollector() };
Collector c = MultiCollector.Wrap(dcs);
Assert.IsTrue(c.AcceptsDocsOutOfOrder());
c.Collect(1);
c.NextReader = null;
c.Scorer = null;
foreach (DummyCollector dc in dcs)
{
Assert.IsTrue(dc.AcceptsDocsOutOfOrderCalled);
Assert.IsTrue(dc.CollectCalled);
Assert.IsTrue(dc.SetNextReaderCalled);
Assert.IsTrue(dc.SetScorerCalled);
}
}
public virtual void TestCollector()
{
// Tests that the collector delegates calls to input collectors properly.
// Tests that the collector handles some null collectors well. If it
// doesn't, an NPE would be thrown.
DummyCollector[] dcs = new DummyCollector[] { new DummyCollector(), new DummyCollector() };
ICollector c = MultiCollector.Wrap(dcs);
Assert.IsTrue(c.AcceptsDocsOutOfOrder);
c.Collect(1);
c.SetNextReader(null);
c.SetScorer(null);
foreach (DummyCollector dc in dcs)
{
Assert.IsTrue(dc.AcceptsDocsOutOfOrderCalled);
Assert.IsTrue(dc.CollectCalled);
Assert.IsTrue(dc.SetNextReaderCalled);
Assert.IsTrue(dc.SetScorerCalled);
}
}
