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);
  }
}