Merge "statsd_testdrive: added support to look for atom id among extensions"
diff --git a/statsd/tools/localtools/src/com/android/statsd/shelltools/testdrive/TestDrive.java b/statsd/tools/localtools/src/com/android/statsd/shelltools/testdrive/TestDrive.java
index a223674..501dc40 100644
--- a/statsd/tools/localtools/src/com/android/statsd/shelltools/testdrive/TestDrive.java
+++ b/statsd/tools/localtools/src/com/android/statsd/shelltools/testdrive/TestDrive.java
@@ -295,7 +295,8 @@
         }
 
         void addAtom(Integer atom, List<String> protoIncludes) {
-            if (Atom.getDescriptor().findFieldByNumber(atom) == null) {
+            if (Atom.getDescriptor().findFieldByNumber(atom) == null &&
+                Atom.getDescriptor().isExtensionNumber(atom) == false) {
                 // try to look in alternative locations
                 if (protoIncludes != null) {
                     boolean isAtomDefined = false;