Support needs for reusable jobs

actions-languageservice/src/context-providers/needs.test.ts

@@ -0,0 +1,75 @@
+import {DescriptionDictionary} from "@github/actions-expressions/.";
+import {WorkflowContext} from "../context/workflow-context";
+import {testGetWorkflowContext} from "../test-utils/test-workflow-context";
+import {getNeedsContext} from "./needs";
+
+describe("needs context", () => {
+  describe("invalid workflow context", () => {
+    it("jobs not defined", () => {
+      const workflowContext = {} as WorkflowContext;
+      expect(workflowContext.job).toBeUndefined();
+      expect(workflowContext.reusableWorkflowJob).toBeUndefined();
+
+      const context = getNeedsContext(workflowContext);
+      expect(context).toEqual(new DescriptionDictionary());
+    });
+  });
+
+  it("job without needs", async () => {
+    const workflowContext = await testGetWorkflowContext(`on: push
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+    - run: ec|ho`);
+
+    const context = getNeedsContext(workflowContext);
+    expect(context).toEqual(new DescriptionDictionary());
+  });
+
+  it("job with needs", async () => {
+    const workflowContext = await testGetWorkflowContext(`on: push
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+    - run: echo
+  build:
+    runs-on: ubuntu-latest
+    needs: [test]
+    steps:
+    - run: ec|ho`);
+
+    const context = getNeedsContext(workflowContext);
+    expect(context.pairs().map(x => x.key)).toEqual(["test"]);
+  });
+
+  it("reusable job without needs", async () => {
+    const workflowContext = await testGetWorkflowContext(`on: push
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+    - run: echo
+  build:
+    uses: ./.github/workflows/some-reusable-wor|kflow.yml`);
+
+    const context = getNeedsContext(workflowContext);
+    expect(context).toEqual(new DescriptionDictionary());
+  });
+
+  it("reusable job with needs", async () => {
+    const workflowContext = await testGetWorkflowContext(`on: push
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+    - run: echo
+  build:
+    uses: ./.github/workflows/some-reusable-wor|kflow.yml
+    needs: [test]`);
+
+    const context = getNeedsContext(workflowContext);
+    expect(context.pairs().map(x => x.key)).toEqual(["test"]);
+  });
+});