98 lines
3.0 KiB
C#
98 lines
3.0 KiB
C#
// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.
|
|
|
|
using System.Collections.Generic;
|
|
using System.Reactive.Concurrency;
|
|
using Microsoft.Reactive.Testing;
|
|
using System;
|
|
|
|
namespace ReactiveTests
|
|
{
|
|
public class MockEnumerable<T> : IEnumerable<T>
|
|
{
|
|
public readonly TestScheduler Scheduler;
|
|
public readonly List<Subscription> Subscriptions = new List<Subscription>();
|
|
|
|
IEnumerable<T> underlyingEnumerable;
|
|
|
|
public MockEnumerable(TestScheduler scheduler, IEnumerable<T> underlyingEnumerable)
|
|
{
|
|
if (scheduler == null)
|
|
throw new ArgumentNullException("scheduler");
|
|
if (underlyingEnumerable == null)
|
|
throw new ArgumentNullException("underlyingEnumerable");
|
|
|
|
this.Scheduler = scheduler;
|
|
this.underlyingEnumerable = underlyingEnumerable;
|
|
}
|
|
|
|
public IEnumerator<T> GetEnumerator()
|
|
{
|
|
return new MockEnumerator(Scheduler, Subscriptions, underlyingEnumerable.GetEnumerator());
|
|
}
|
|
|
|
System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
|
|
{
|
|
return GetEnumerator();
|
|
}
|
|
|
|
class MockEnumerator : IEnumerator<T>
|
|
{
|
|
List<Subscription> subscriptions;
|
|
IEnumerator<T> enumerator;
|
|
TestScheduler scheduler;
|
|
int index;
|
|
bool disposed = false;
|
|
|
|
public MockEnumerator(TestScheduler scheduler, List<Subscription> subscriptions, IEnumerator<T> enumerator)
|
|
{
|
|
this.subscriptions = subscriptions;
|
|
this.enumerator = enumerator;
|
|
this.scheduler = scheduler;
|
|
|
|
index = subscriptions.Count;
|
|
subscriptions.Add(new Subscription(scheduler.Clock));
|
|
}
|
|
|
|
public T Current
|
|
{
|
|
get
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException("this");
|
|
return enumerator.Current;
|
|
}
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
if (!disposed)
|
|
{
|
|
disposed = true;
|
|
enumerator.Dispose();
|
|
subscriptions[index] = new Subscription(subscriptions[index].Subscribe, scheduler.Clock);
|
|
}
|
|
}
|
|
|
|
object System.Collections.IEnumerator.Current
|
|
{
|
|
get { return Current; }
|
|
}
|
|
|
|
public bool MoveNext()
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException("this");
|
|
return enumerator.MoveNext();
|
|
}
|
|
|
|
public void Reset()
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException("this");
|
|
enumerator.Reset();
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|