move to pub version of vm_service_client

packages/flutter_driver/lib/src/driver.dart

diff --git a/packages/flutter_driver/lib/src/driver.dart b/packages/flutter_driver/lib/src/driver.dart
index 10d7763..b4fbfe3 100644
--- a/packages/flutter_driver/lib/src/driver.dart
+++ b/packages/flutter_driver/lib/src/driver.dart
@@ -74,8 +74,8 @@
 
       // Waits for a signal from the VM service that the extension is registered
       Future waitForServiceExtension() {
-        return isolate.onServiceExtensionAdded.firstWhere((VMServiceExtension ext) {
-          return ext.method == _flutterExtensionMethod;
+        return isolate.onExtensionAdded.firstWhere((String extension) {
+          return extension == _flutterExtensionMethod;
         });
       }
 

packages/flutter_driver/pubspec.yaml

diff --git a/packages/flutter_driver/pubspec.yaml b/packages/flutter_driver/pubspec.yaml
index 2918a68..d15ffba 100644
--- a/packages/flutter_driver/pubspec.yaml
+++ b/packages/flutter_driver/pubspec.yaml
@@ -8,10 +8,7 @@
   sdk: '>=1.12.0 <2.0.0'
 
 dependencies:
-  vm_service_client:
-    git:
-      url: git://github.com/yjbanov/vm_service_client.git
-      ref: 54085d1
+  vm_service_client: '>=0.1.2 <1.0.0'
   json_rpc_2: any
   logging: '>=0.11.0 <1.0.0'
   flutter:

packages/flutter_driver/test/flutter_driver_test.dart

diff --git a/packages/flutter_driver/test/flutter_driver_test.dart b/packages/flutter_driver/test/flutter_driver_test.dart
index 3dee762..c2386b9 100644
--- a/packages/flutter_driver/test/flutter_driver_test.dart
+++ b/packages/flutter_driver/test/flutter_driver_test.dart
@@ -46,10 +46,8 @@
     test('connects to isolate paused at start', () async {
       when(mockIsolate.pauseEvent).thenReturn(new MockVMPauseStartEvent());
       when(mockIsolate.resume()).thenReturn(new Future.value());
-      MockVMServiceExtension mockExtension = new MockVMServiceExtension();
-      when(mockExtension.method).thenReturn('ext.flutter_driver');
-      when(mockIsolate.onServiceExtensionAdded)
-          .thenReturn(new Stream.fromIterable([mockExtension]));
+      when(mockIsolate.onExtensionAdded)
+          .thenReturn(new Stream.fromIterable(['ext.flutter_driver']));
 
       FlutterDriver driver = await FlutterDriver.connect();
       expect(driver, isNotNull);
@@ -203,6 +201,3 @@
 
 @proxy
 class MockVMResumeEvent extends Mock implements VMResumeEvent { }
-
-@proxy
-class MockVMServiceExtension extends Mock implements VMServiceExtension { }