Exemplo n.º 1
0
 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))
Exemplo n.º 2
0
 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})