runtime/observatory/tests/observatory_ui/mocks/repositories/class.dart

// Copyright (c) 2016, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

part of mocks;

typedef Future<M.Class> ClassRepositoryMockObjectCallback(M.Isolate isolate);
typedef Future<M.Class> ClassRepositoryMockGetterCallback(
    M.Isolate isolate, String id);

class ClassRepositoryMock implements M.ClassRepository {
  final ClassRepositoryMockObjectCallback _object;
  final ClassRepositoryMockGetterCallback _get;

  ClassRepositoryMock(
      {ClassRepositoryMockObjectCallback object,
      ClassRepositoryMockGetterCallback getter})
      : _object = object,
        _get = getter;

  Future<M.Class> getObject(M.IsolateRef i) {
    if (_object != null) {
      return _object(i);
    }
    return new Future.value(null);
  }

  Future<M.Class> get(M.IsolateRef i, String id) {
    if (_get != null) {
      return _get(i, id);
    }
    return new Future.value(null);
  }
}