def test(save_to_db): save_to_db.return_value = None save_transaction('mock_transaction_id', '$23.22') save_to_db.assert_called_with(key='mock_transaction_id', value='$23.22', print_result=True) assert save_transaction(transaction_id=None, value='$44.99') is None save_to_db.assert_called_with(key=None, value='$44.99', print_result=True)
def test(get_database): database_mock = Mock() database_mock.__setitem__ = Mock(return_value=None) get_database.return_value = database_mock assert save_transaction('fds', '$12.00', False) == None