Allowed Context

__tests__/functions/context-check.test.js

@@ -7,8 +7,10 @@ const saveStateMock = jest.spyOn(core, 'saveState')
 var context
 beforeEach(() => {
   jest.clearAllMocks()
+  process.env.INPUT_ALLOWED_CONTEXTS = 'pull_request,issue'
   jest.spyOn(core, 'warning').mockImplementation(() => {})
   jest.spyOn(core, 'saveState').mockImplementation(() => {})
+  jest.spyOn(core, 'debug').mockImplementation(() => {})
 
   context = {
     eventName: 'issue_comment',
@@ -24,14 +26,70 @@ beforeEach(() => {
 })
 
 test('checks the event context and finds that it is valid', async () => {
-  expect(await contextCheck(context)).toBe(true)
+  expect(await contextCheck(context)).toStrictEqual({
+    valid: true,
+    context: 'pull_request'
+  })
+})
+
+test('checks the event context for an issue comment and finds that it is valid', async () => {
+  context.payload.issue = {}
+  expect(await contextCheck(context)).toStrictEqual({
+    valid: true,
+    context: 'issue'
+  })
+})
+
+test('checks the event context for an issue comment and finds that it is valid - when only issue comments are allowed', async () => {
+  process.env.INPUT_ALLOWED_CONTEXTS = 'issue'
+  context.payload.issue = {}
+  expect(await contextCheck(context)).toStrictEqual({
+    valid: true,
+    context: 'issue'
+  })
+})
+
+test('checks the event context for a pr comment and finds that it is valid - when only pr comments are allowed', async () => {
+  process.env.INPUT_ALLOWED_CONTEXTS = 'pull_request'
+  expect(await contextCheck(context)).toStrictEqual({
+    valid: true,
+    context: 'pull_request'
+  })
+})
+
+test('checks the event context for a pr comment and finds that it is invalid - when only pr comments are allowed', async () => {
+  process.env.INPUT_ALLOWED_CONTEXTS = 'pull_request'
+  context.payload.issue = {}
+  expect(await contextCheck(context)).toStrictEqual({
+    valid: false,
+    context: 'issue_comment'
+  })
+
+  expect(warningMock).toHaveBeenCalledWith(
+    'this Action can only be run in the context of a pull request comment'
+  )
+})
+
+test('checks the event context for an issue comment and finds that it is invalid - when only issue comments are allowed', async () => {
+  process.env.INPUT_ALLOWED_CONTEXTS = 'issue'
+  expect(await contextCheck(context)).toStrictEqual({
+    valid: false,
+    context: 'issue_comment'
+  })
+
+  expect(warningMock).toHaveBeenCalledWith(
+    'this Action can only be run in the context of an issue comment'
+  )
 })
 
 test('checks the event context and finds that it is invalid', async () => {
   context.eventName = 'push'
-  expect(await contextCheck(context)).toBe(false)
+  expect(await contextCheck(context)).toStrictEqual({
+    valid: false,
+    context: 'push'
+  })
   expect(warningMock).toHaveBeenCalledWith(
-    'This Action can only be run in the context of a pull request comment'
+    'this Action can only be run in the context of an issue_comment'
   )
   expect(saveStateMock).toHaveBeenCalledWith('bypass', 'true')
 })

__tests__/functions/post.test.js

@@ -35,7 +35,7 @@ beforeEach(() => {
     return validStates[name]
   })
   jest.spyOn(contextCheck, 'contextCheck').mockImplementation(() => {
-    return true
+    return {valid: true, context: 'pull_request'}
   })
 
   jest.spyOn(postReactions, 'postReactions').mockImplementation(() => {
@@ -59,7 +59,7 @@ test('successfully runs post() Action logic when "success" is false', async () =
 
 test('exits due to an invalid Actions context', async () => {
   jest.spyOn(contextCheck, 'contextCheck').mockImplementation(() => {
-    return false
+    return {valid: false, context: 'pull_request'}
   })
   expect(await post()).toBeUndefined()
 })