handle learning track REST category page (#46146)

rsese · web-flow · commit d36f23ef3b24 · 2023-11-15T16:19:09.000Z
diff --git a/src/fixtures/fixtures/data/learning-tracks/code-security.yml b/src/fixtures/fixtures/data/learning-tracks/code-security.yml
@@ -11,3 +11,4 @@ bar_foo:
   guides:
     - /code-security/getting-started/securing-your-organization
     - /code-security/getting-started/quickstart
+    - /rest/actions
diff --git a/src/fixtures/tests/guides.js b/src/fixtures/tests/guides.js
@@ -1,4 +1,4 @@
-import { getDOMCached as getDOM } from '#src/tests/helpers/e2etest.js'
+import { get, getDOMCached as getDOM } from '#src/tests/helpers/e2etest.js'
 
 describe('guides', () => {
   test("page's title should be document title", async () => {
@@ -94,4 +94,11 @@ describe('learning tracks', () => {
       expect($('[data-testid=learning-track-nav]').length).toBe(0)
     }
   })
+
+  test('REST category learning track article works', async () => {
+    const response = await get('/rest/actions?learnProduct=code-security&learn=bar_foo', {
+      followAllRedirects: true,
+    })
+    expect(response.statusCode).toBe(200)
+  })
 })
diff --git a/src/rest/pages/category.tsx b/src/rest/pages/category.tsx
@@ -1,7 +1,12 @@
 import { GetServerSideProps } from 'next'
 import { Operation } from 'src/rest/components/types'
 import { RestReferencePage } from 'src/rest/components/RestReferencePage'
-import { getMainContext, MainContext, MainContextT } from 'src/frame/components/context/MainContext'
+import {
+  addUINamespaces,
+  getMainContext,
+  MainContext,
+  MainContextT,
+} from 'src/frame/components/context/MainContext'
 import {
   AutomatedPageContext,
   AutomatedPageContextT,
@@ -200,10 +205,15 @@ export const getServerSideProps: GetServerSideProps<Props> = async (context) =>
   // created.
   tocLandingContext.tocItems = restCategoryTocItems
 
+  const mainContext = await getMainContext(req, res)
+  if (tocLandingContext.currentLearningTrack?.trackName) {
+    addUINamespaces(req, mainContext.data.ui, ['learning_track_nav'])
+  }
+
   return {
     props: {
       restOperations,
-      mainContext: await getMainContext(req, res),
+      mainContext,
       automatedPageContext: getAutomatedPageContextFromRequest(req),
       tocLandingContext,
     },
