From b33633ce1839aa1c08b01dfc6b47ca91715f8e97 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicolas=20Jim=C3=A9nez?= <nicolas.jimenez@edunext.co>
Date: Mon, 17 Jun 2024 11:38:03 -0500
Subject: [PATCH] feat: staff endpoint in view class buttons

---
 .../Classes/Class/ClassPage/Actions.jsx       |  6 +-
 src/features/Classes/ClassesTable/columns.jsx |  6 +-
 .../Courses/CourseDetailTable/columns.jsx     |  4 +-
 src/features/Main/data/_test_/api.test.js     | 25 +++++++-
 src/features/Main/data/api.js                 | 20 ++++++
 src/helpers/__test__/index.test.js            | 62 ++++++++++++++++++-
 src/helpers/index.js                          | 16 +++++
 7 files changed, 128 insertions(+), 11 deletions(-)

diff --git a/src/features/Classes/Class/ClassPage/Actions.jsx b/src/features/Classes/Class/ClassPage/Actions.jsx
index 482c5484..ac2d62c3 100644
--- a/src/features/Classes/Class/ClassPage/Actions.jsx
+++ b/src/features/Classes/Class/ClassPage/Actions.jsx
@@ -10,6 +10,8 @@ import {
 } from '@edx/paragon';
 import { MoreVert } from '@edx/paragon/icons';
 
+import { setAssignStaffRole } from 'helpers';
+
 import AddClass from 'features/Courses/AddClass';
 import EnrollStudent from 'features/Classes/EnrollStudent';
 
@@ -47,9 +49,7 @@ const Actions = ({ previousPage }) => {
       </Button>
       <Button
         as="a"
-        href={classLink}
-        target="_blank"
-        rel="noopener noreferrer"
+        onClick={() => setAssignStaffRole(classLink, classId)}
         className="text-decoration-none text-white button-view-class mr-3"
       >
         <i className="fa-solid fa-arrow-up-right-from-square mr-2 mb-1" />
diff --git a/src/features/Classes/ClassesTable/columns.jsx b/src/features/Classes/ClassesTable/columns.jsx
index bc707b64..dd9d5350 100644
--- a/src/features/Classes/ClassesTable/columns.jsx
+++ b/src/features/Classes/ClassesTable/columns.jsx
@@ -13,7 +13,7 @@ import { getConfig } from '@edx/frontend-platform';
 
 import AddClass from 'features/Courses/AddClass';
 
-import { formatUTCDate } from 'helpers';
+import { formatUTCDate, setAssignStaffRole } from 'helpers';
 
 const columns = [
   {
@@ -107,9 +107,7 @@ const columns = [
               Edit Class
             </Dropdown.Item>
             <Dropdown.Item
-              href={`${getConfig().LEARNING_MICROFRONTEND_URL}/course/${classId}/home`}
-              target="_blank"
-              rel="noreferrer"
+              onClick={() => setAssignStaffRole(`${getConfig().LEARNING_MICROFRONTEND_URL}/course/${classId}/home`, classId)}
               className="text-truncate text-decoration-none custom-text-black"
             >
               <i className="fa-solid fa-arrow-up-right-from-square mr-2 mb-1" />
diff --git a/src/features/Courses/CourseDetailTable/columns.jsx b/src/features/Courses/CourseDetailTable/columns.jsx
index 81fe9e22..7abc52ce 100644
--- a/src/features/Courses/CourseDetailTable/columns.jsx
+++ b/src/features/Courses/CourseDetailTable/columns.jsx
@@ -8,7 +8,7 @@ import { Badge } from 'react-paragon-topaz';
 import { MoreHoriz } from '@edx/paragon/icons';
 import { getConfig } from '@edx/frontend-platform';
 
-import { formatUTCDate } from 'helpers';
+import { formatUTCDate, setAssignStaffRole } from 'helpers';
 
 import AddClass from 'features/Courses/AddClass';
 
@@ -121,9 +121,9 @@ const columns = [
           />
           <Dropdown.Menu>
             <Dropdown.Item
-              href={`${getConfig().LEARNING_MICROFRONTEND_URL}/course/${classId}/home`}
               target="_blank"
               rel="noreferrer"
+              onClick={() => setAssignStaffRole(`${getConfig().LEARNING_MICROFRONTEND_URL}/course/${classId}/home`, classId)}
               className="text-truncate text-decoration-none custom-text-black"
             >
               <i className="fa-regular fa-eye mr-2 mb-1" />
diff --git a/src/features/Main/data/_test_/api.test.js b/src/features/Main/data/_test_/api.test.js
index 40232094..97c78557 100644
--- a/src/features/Main/data/_test_/api.test.js
+++ b/src/features/Main/data/_test_/api.test.js
@@ -1,10 +1,17 @@
-import { getInstitutionName } from 'features/Main/data/api';
+import { getInstitutionName, assignStaffRole } from 'features/Main/data/api';
 import { getAuthenticatedHttpClient } from '@edx/frontend-platform/auth';
 
 jest.mock('@edx/frontend-platform/auth', () => ({
   getAuthenticatedHttpClient: jest.fn(),
 }));
 
+jest.mock('@edx/frontend-platform', () => ({
+  getConfig: jest.fn(() => ({
+    LMS_BASE_URL: 'http://localhost:18000',
+    COURSE_OPERATIONS_API_V2_BASE_URL: 'http://localhost:18000/pearson_course_operation/api/v2',
+  })),
+}));
+
 describe('getInstitutionName', () => {
   test('should call getAuthenticatedHttpClient with the correct parameters', () => {
     const httpClientMock = {
@@ -18,3 +25,19 @@ describe('getInstitutionName', () => {
     expect(httpClientMock.get).toHaveBeenCalledTimes(1);
   });
 });
+
+describe('assignStaffRole', () => {
+  test('Should call getAuthenticatedHttpClient with the correct parameters', () => {
+    const httpClientMock = {
+      post: jest.fn(),
+    };
+
+    getAuthenticatedHttpClient.mockReturnValue(httpClientMock);
+
+    assignStaffRole('example+123');
+    const data = new FormData();
+    data.append('class_id', 'example+123');
+
+    expect(httpClientMock.post).toHaveBeenCalledWith('http://localhost:18000/pearson_course_operation/api/v2/assign-staff/', data);
+  });
+});
diff --git a/src/features/Main/data/api.js b/src/features/Main/data/api.js
index d1bc0ae3..31ecff38 100644
--- a/src/features/Main/data/api.js
+++ b/src/features/Main/data/api.js
@@ -7,6 +7,26 @@ function getInstitutionName() {
   );
 }
 
+/**
+ * Assigns a staff role to a class.
+ *
+ * This function makes an authenticated HTTP request to the COURSE_OPERATIONS_API_V2
+ * to assign a staff role to the specified class ID.
+ *
+ * @param {string} classId - The ID of the class to which the staff role will be assigned.
+ * @returns {Promise} A promise that resolves with the HTTP response of the request.
+ */
+function assignStaffRole(classId) {
+  const formData = new FormData();
+  formData.append('class_id', classId);
+
+  return getAuthenticatedHttpClient().post(
+    `${getConfig().COURSE_OPERATIONS_API_V2_BASE_URL}/assign-staff/`,
+    formData,
+  );
+}
+
 export {
   getInstitutionName,
+  assignStaffRole,
 };
diff --git a/src/helpers/__test__/index.test.js b/src/helpers/__test__/index.test.js
index 53ff73eb..f8a5cc14 100644
--- a/src/helpers/__test__/index.test.js
+++ b/src/helpers/__test__/index.test.js
@@ -1,4 +1,21 @@
-import { formatDateRange, formatUTCDate, getInitials } from 'helpers';
+import { logError } from '@edx/frontend-platform/logging';
+
+import {
+  formatDateRange,
+  formatUTCDate,
+  getInitials,
+  setAssignStaffRole,
+} from 'helpers';
+
+import { assignStaffRole } from 'features/Main/data/api';
+
+jest.mock('@edx/frontend-platform/logging', () => ({
+  logError: jest.fn(),
+}));
+
+jest.mock('features/Main/data/api', () => ({
+  assignStaffRole: jest.fn(),
+}));
 
 describe('formatDateRange', () => {
   test('Should return "-" when startDate is not provided', () => {
@@ -60,3 +77,46 @@ describe('getInitials', () => {
     expect(getInitials('Mary-Jane Watson')).toBe('MW');
   });
 });
+
+describe('setAssignStaffRole', () => {
+  const originalWindowOpen = window.open;
+
+  beforeAll(() => {
+    window.open = jest.fn();
+  });
+
+  afterAll(() => {
+    window.open = originalWindowOpen;
+  });
+
+  beforeEach(() => {
+    jest.clearAllMocks();
+  });
+
+  test('Should open a new window with the correct URL on success', async () => {
+    assignStaffRole.mockResolvedValueOnce();
+
+    const url = 'http://example.com';
+    const classId = '12345';
+
+    await setAssignStaffRole(url, classId);
+
+    expect(assignStaffRole).toHaveBeenCalledWith(classId);
+    expect(logError).not.toHaveBeenCalled();
+    expect(window.open).toHaveBeenCalledWith(url, '_blank', 'noopener,noreferrer');
+  });
+
+  test('Should log an error and open a new window with the correct URL on failure', async () => {
+    const error = new Error('Assignment failed');
+    assignStaffRole.mockRejectedValueOnce(error);
+
+    const url = 'http://example.com';
+    const classId = '12345';
+
+    await setAssignStaffRole(url, classId);
+
+    expect(assignStaffRole).toHaveBeenCalledWith(classId);
+    expect(logError).toHaveBeenCalledWith(error);
+    expect(window.open).toHaveBeenCalledWith(url, '_blank', 'noopener,noreferrer');
+  });
+});
diff --git a/src/helpers/index.js b/src/helpers/index.js
index 539a8939..38132165 100644
--- a/src/helpers/index.js
+++ b/src/helpers/index.js
@@ -1,4 +1,7 @@
 import { format } from 'date-fns';
+import { logError } from '@edx/frontend-platform/logging';
+
+import { assignStaffRole } from 'features/Main/data/api';
 
 /**
  * Format a UTC date
@@ -102,3 +105,16 @@ export const getInitials = (name) => {
 
   return name.trim().split(/\s+/).map(word => word.charAt(0).toUpperCase()).join('');
 };
+
+/**
+ * Assigns a staff role to a class and then opens a new window or tab with a specific URL.
+ *
+ * @param {string} url - The URL to open in a new window or tab after assigning the role.
+ * @param {string} classId - The ID of the class to which the staff role will be assigned.
+ *
+ * @returns {Promise} - The promise returned by the assignStaffRole function, which resolves once
+ * the role assignment is complete.
+ */
+export const setAssignStaffRole = (url, classId) => assignStaffRole(classId).catch(logError).finally(() => {
+  window.open(url, '_blank', 'noopener,noreferrer');
+});