python test
import unittest
from unittest.mock import MagicMock
Example CRUD class
class CRUDOperations:
def create(self, data):
pass
def read(self, id):
pass
def update(self, id, data):
pass
def delete(self, id):
pass
Test class
class TestCRUDOperations(unittest.TestCase):
def setUp(self):
self.crud = CRUDOperations()
self.crud.create = MagicMock(return_value=True)
self.crud.read = MagicMock(return_value={'id': 1, 'data': 'test'})
self.crud.update = MagicMock(return_value=True)
self.crud.delete = MagicMock(return_value=True)
def test_create(self):
result = self.crud.create({'data': 'test'})
self.assertTrue(result)
self.crud.create.assert_called_once_with({'data': 'test'})
def test_read(self):
result = self.crud.read(1)
self.assertEqual(result, {'id': 1, 'data': 'test'})
self.crud.read.assert_called_once_with(1)
def test_update(self):
result = self.crud.update(1, {'data': 'updated'})
self.assertTrue(result)
self.crud.update.assert_called_once_with(1, {'data': 'updated'})
def test_delete(self):
result = self.crud.delete(1)
self.assertTrue(result)
self.crud.delete.assert_called_once_with(1)
if name == 'main':
unittest.main()