1 files changed, 7 insertions, 2 deletions
diff --git a/src/test/java/com/google/devtools/build/skydoc/SkydocTest.java b/src/test/java/com/google/devtools/build/skydoc/SkydocTest.java
index 870fa67e52..a1d7acbe29 100644
--- a/src/test/java/com/google/devtools/build/skydoc/SkydocTest.java
+++ b/src/test/java/com/google/devtools/build/skydoc/SkydocTest.java
@@ -97,11 +97,16 @@ public final class SkydocTest extends SkylarkTestCase {
     Entry<String, RuleInfo> ruleInfo = Iterables.getOnlyElement(ruleInfos.entrySet());
     assertThat(ruleInfo.getKey()).isEqualTo("my_rule");
     assertThat(ruleInfo.getValue().getDocString()).isEqualTo("This is my rule. It does stuff.");
-    assertThat(ruleInfo.getValue().getAttrNames()).containsExactly(
-        "first", "second", "third", "fourth");
+    assertThat(getAttrNames(ruleInfo.getValue())).containsExactly(
+        "first", "fourth", "second", "third").inOrder();
     assertThat(unexportedRuleInfos.build()).isEmpty();
   }
 
+  private static Iterable<String> getAttrNames(RuleInfo ruleInfo) {
+    return ruleInfo.getAttributes().stream().map(attr -> attr.getName())
+        .collect(Collectors.toList());
+  }
+
   @Test
   public void testMultipleRuleNames() throws Exception {
     scratch.file(