diff --git a/pubspec.yaml b/pubspec.yaml
index 9952d7a..0beb6cc 100644
--- a/pubspec.yaml
+++ b/pubspec.yaml
@@ -7,3 +7,5 @@
   sdk: '>=1.0.0 <2.0.0'
 dependencies:
   which: '>=0.1.2 <0.2.0'
+dev_dependencies:
+  unittest: '>=0.11.0 <0.12.0'
diff --git a/test/cli_util_test.dart b/test/cli_util_test.dart
new file mode 100644
index 0000000..b7b7822
--- /dev/null
+++ b/test/cli_util_test.dart
@@ -0,0 +1,21 @@
+// Copyright (c) 2015, 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.
+
+import 'package:cli_util/cli_util.dart';
+import 'package:unittest/unittest.dart';
+
+void defineTests() {
+  group('getSdkDir', () {
+    test('arg parsing', () {
+      expect(getSdkDir(['--dart-sdk', '/dart/sdk']).path, equals('/dart/sdk'));
+      expect(getSdkDir(['--dart-sdk=/dart/sdk']).path, equals('/dart/sdk'));
+    });
+  });
+}
+
+main() {
+  groupSep = ' | ';
+
+  defineTests();
+}
diff --git a/tool/travis.sh b/tool/travis.sh
index b21d421..e8e10b7 100755
--- a/tool/travis.sh
+++ b/tool/travis.sh
@@ -9,10 +9,11 @@
 
 # Verify that the libraries are error free.
 dartanalyzer --fatal-warnings \
-  lib/cli_util.dart
+  lib/cli_util.dart \
+  test/cli_util_test.dart
 
 # Run the tests.
-#dart test/all.dart
+dart test/cli_util_test.dart
 
 # Install dart_coveralls; gather and send coverage data.
 if [ "$COVERALLS_TOKEN" ]; then
@@ -21,5 +22,5 @@
     --token $COVERALLS_TOKEN \
     --retry 2 \
     --exclude-test-files \
-    test/all.dart
+    test/cli_util_test.dart
 fi
\ No newline at end of file