parse rawendpoints

buildbot · Sep 27, 2016 · aef24e6 · aef24e6
1 parent a493f8c
commit aef24e6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/master/buildbot/test/unit/test_util_raml.py b/master/buildbot/test/unit/test_util_raml.py
@@ -81,3 +81,8 @@ def test_iter_actions(self):
         build = self.api.endpoints_by_type['build']
         actions = dict(self.api.iter_actions(build['/builds/{buildid}']))
         self.assertEqual(sorted(actions.keys()), sorted(['rebuild', 'stop']))
+
+    def test_rawendpoints(self):
+        self.assertIn(
+            "/steps/{stepid}/logs/{log_slug}/raw",
+            self.api.rawendpoints.keys())
diff --git a/master/buildbot/util/raml.py b/master/buildbot/util/raml.py
@@ -17,8 +17,8 @@
 import os
 
 import ramlfications
-
 from future.utils import iteritems
+
 try:
     from collections import OrderedDict
 except ImportError:  # pragma: no cover
@@ -43,6 +43,7 @@ def __init__(self):
             os.path.dirname(__file__), os.pardir, 'spec', 'api.raml'))
         endpoints = {}
         self.endpoints_by_type = {}
+        self.rawendpoints = {}
         self.endpoints = self.parse_endpoints(endpoints, "", self.api)
         self.types = self.parse_types()
 
@@ -66,6 +67,9 @@ def parse_endpoints(self, endpoints, base, api, uriParameters=None):
                             v['eptype'] = _is['bbget']['bbtype']
                             self.endpoints_by_type.setdefault(v['eptype'], {})
                             self.endpoints_by_type[v['eptype']][base] = api
+                        if 'bbgetraw' in _is:
+                            self.rawendpoints.setdefault(base, {})
+                            self.rawendpoints[base] = api
         return endpoints
 
     def reindent(self, s, indent):