Add spec for this case

app/assets/javascripts/pipelines/components/graph/graph_component_wrapper.vue

@@ -138,6 +138,10 @@ export default {
         metricsPath: this.metricsPath,
       };
     },
+    graphViewType() {
+      /* This prevents reading view type off the localStorage value if it does not apply. */
+      return this.showGraphViewSelector ? this.currentViewType : STAGE_VIEW;
+    },
     showLoadingIcon() {
       /*
         Shows the icon only when the graph is empty, not when it is is
@@ -205,7 +209,7 @@ export default {
     >
       <graph-view-selector
         v-if="showGraphViewSelector"
-        :type="currentViewType"
+        :type="graphViewType"
         :show-links="showLinks"
         @updateViewType="updateViewType"
         @updateShowLinksState="updateShowLinksState"
@@ -218,7 +222,7 @@ export default {
       :pipeline="pipeline"
       :pipeline-layers="getPipelineLayers()"
       :show-links="showLinks"
-      :view-type="currentViewType"
+      :view-type="graphViewType"
       @error="reportFailure"
       @refreshPipelineGraph="refreshPipelineGraph"
     />

spec/frontend/pipelines/graph/graph_component_wrapper_spec.js

@@ -354,6 +354,36 @@ describe('Pipeline graph wrapper', () => {
       });
     });
 
+    describe('when feature flag is on and local storage is set, but the graph does not use needs', () => {
+      beforeEach(async () => {
+        const nonNeedsResponse = { ...mockPipelineResponse };
+        nonNeedsResponse.data.project.pipeline.usesNeeds = false;
+
+        localStorage.setItem(VIEW_TYPE_KEY, LAYER_VIEW);
+
+        createComponentWithApollo({
+          provide: {
+            glFeatures: {
+              pipelineGraphLayersView: true,
+            },
+          },
+          mountFn: mount,
+          getPipelineDetailsHandler: jest.fn().mockResolvedValue(nonNeedsResponse),
+        });
+
+        jest.runOnlyPendingTimers();
+        await wrapper.vm.$nextTick();
+      });
+
+      afterEach(() => {
+        localStorage.clear();
+      });
+
+      it('still passes stage type to graph', () => {
+        expect(getGraph().props('viewType')).toBe(STAGE_VIEW);
+      });
+    });
+
     describe('when feature flag is on but pipeline does not use needs', () => {
       beforeEach(async () => {
         const nonNeedsResponse = { ...mockPipelineResponse };