Add a few more tests

votingworks · Oct 4, 2023 · 6731e39 · 6731e39
1 parent 4e2090f
commit 6731e39
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 8 deletions.
diff --git a/apps/scan/frontend/src/components/cast_vote_record_sync_modal.test.tsx b/apps/scan/frontend/src/components/cast_vote_record_sync_modal.test.tsx
@@ -27,7 +27,6 @@ afterEach(() => {
 
 test('CVR sync modal success case', async () => {
   renderComponent();
-
   expect(screen.queryByRole('alertdialog')).not.toBeInTheDocument();
 
   apiMock.expectGetUsbDriveStatus('mounted', {
@@ -55,7 +54,6 @@ test('CVR sync modal success case', async () => {
 
 test('CVR sync modal error case', async () => {
   renderComponent();
-
   expect(screen.queryByRole('alertdialog')).not.toBeInTheDocument();
 
   apiMock.expectGetUsbDriveStatus('mounted', {

diff --git a/apps/scan/frontend/src/screens/election_manager_screen.test.tsx b/apps/scan/frontend/src/screens/election_manager_screen.test.tsx
@@ -367,12 +367,14 @@ test('switching to test mode when ballots have been counted', async () => {
     expect(screen.queryByRole('alertdialog')).not.toBeInTheDocument()
   );
   // For some reason, getting by role doesn't work here
-  expect(
-    screen
-      .getByText('Test Ballot Mode')
-      .closest('button')
-      ?.getAttribute('aria-selected')
-  ).toEqual('true');
+  await waitFor(() => {
+    expect(
+      screen
+        .getByText('Test Ballot Mode')
+        .closest('button')
+        ?.getAttribute('aria-selected')
+    ).toEqual('true');
+  });
   expect(
     screen
       .getByText('Official Ballot Mode')
@@ -413,6 +415,39 @@ test('machine cannot be switched to test mode if CVR sync is required and ballot
   );
 });
 
+test('machine *can* be switched to test mode if CVR sync is required but no ballots have been counted', async () => {
+  apiMock.mockApiClient.getUsbDriveStatus.reset();
+  apiMock.expectGetUsbDriveStatus('mounted', {
+    doesUsbDriveRequireCastVoteRecordSync: true,
+  });
+  apiMock.expectGetConfig({ isTestMode: false });
+  renderScreen({ scannerStatus: { ...statusNoPaper, ballotsCounted: 0 } });
+  await screen.findByRole('heading', { name: 'Election Manager Settings' });
+
+  userEvent.click(screen.getByRole('tab', { name: 'Configuration' }));
+
+  const testBallotModeButton = await screen.findByRole('option', {
+    name: 'Test Ballot Mode',
+    selected: false,
+  });
+  screen.getByRole('option', {
+    name: 'Official Ballot Mode',
+    selected: true,
+  });
+
+  apiMock.expectSetTestMode(true);
+  apiMock.expectGetConfig({ isTestMode: true });
+  userEvent.click(testBallotModeButton);
+  await screen.findByRole('option', {
+    name: 'Test Ballot Mode',
+    selected: true,
+  });
+  screen.getByRole('option', {
+    name: 'Official Ballot Mode',
+    selected: false,
+  });
+});
+
 test('machine cannot be unconfigured if CVR sync is required and not in test mode', async () => {
   apiMock.mockApiClient.getUsbDriveStatus.reset();
   apiMock.expectGetUsbDriveStatus('mounted', {
@@ -439,3 +474,25 @@ test('machine cannot be unconfigured if CVR sync is required and not in test mod
     expect(screen.queryByRole('alertdialog')).not.toBeInTheDocument()
   );
 });
+
+test('machine *can* be unconfigured if CVR sync is required but in test mode', async () => {
+  apiMock.mockApiClient.getUsbDriveStatus.reset();
+  apiMock.expectGetUsbDriveStatus('mounted', {
+    doesUsbDriveRequireCastVoteRecordSync: true,
+  });
+  apiMock.expectGetConfig({ isTestMode: true });
+  renderScreen();
+  await screen.findByRole('heading', { name: 'Election Manager Settings' });
+
+  userEvent.click(screen.getByRole('tab', { name: 'Election Data' }));
+
+  userEvent.click(
+    await screen.findByRole('button', {
+      name: 'Delete All Election Data from VxScan',
+    })
+  );
+  const modal = await screen.findByRole('alertdialog');
+  within(modal).getByText(
+    'Do you want to remove all election information and data from this machine?'
+  );
+});