test: test delete with command line

evanp · Nov 20, 2023 · 9c6ffd8 · 9c6ffd8
1 parent a4bb517
commit 9c6ffd8
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/tests/test_delete.py b/tests/test_delete.py
@@ -1,6 +1,6 @@
 import unittest
 from unittest.mock import patch, mock_open, MagicMock
-from ap.commands.delete import DeleteCommand
+from ap.main import run_command
 from argparse import Namespace
 import io
 import sys
@@ -44,14 +44,12 @@ def tearDown(self):
     def test_delete_with_confirmation(
         self, mock_input, mock_requests_get, mock_oauth_post, mock_file
     ):
-        args = Namespace(id=NOTE_ID, force=False)
-        delete_cmd = DeleteCommand(args)
 
         mock_oauth_post.return_value = MagicMock(
             status_code=200, json=lambda: {"success": True}
         )
 
-        delete_cmd.run()
+        run_command(["delete", NOTE_ID], {})
 
         # Assertions
         self.assertEqual(mock_requests_get.call_count, 2)
@@ -66,14 +64,12 @@ def test_delete_with_confirmation(
     def test_delete_with_force(
         self, mock_input, mock_requests_get, mock_oauth_post, mock_file
     ):
-        args = Namespace(id=NOTE_ID, force=True)
-        delete_cmd = DeleteCommand(args)
 
         mock_oauth_post.return_value = MagicMock(
             status_code=200, json=lambda: {"success": True}
         )
 
-        delete_cmd.run()
+        run_command(["delete", "--force", NOTE_ID], {})
 
         # Assertions
         self.assertEqual(mock_requests_get.call_count, 2)