async def test_get_by_username(self, *args): mock_username = '******' mock_db = CoroutineMock() mock_admin = {} mock_db.find_one = CoroutineMock() mock_db.find_one.return_value = mock_admin admin = await Admin.get_by_username(mock_username, mock_db) mock_db.find_one.assert_called() mock_db.find_one.assert_called_with({'username': mock_username}) expect(admin).to(equal(mock_admin))
async def test_get_by_id(self, *args): with patch('bson.ObjectId') as object_id_mock: mock_id = 'some-value' mock_db = CoroutineMock() mock_db.find_one = CoroutineMock() object_id_mock.return_value = mock_id await Insights.get_by_id(mock_id, mock_db) object_id_mock.assert_called_with(mock_id) mock_db.find_one.assert_called() mock_db.find_one.assert_awaited_with({'_id': mock_id})