Wsrtjtyk

I have an issue where I have a dictionary stub inside a mock and my test crashes. Shouldn't the mock replace everything to begin with so I shouldn't even run into this issue?

script.py

from selenium import webdriver



def nothing_here_really_just_setting_up_the_logic_flow():

   """

   This function is completely irrelevant. It just loads a website

   returns the webdriver session as "driver" and a dictionary 

   "add extra xpath" with the addtional xpath syntax for the function

   where my issue is at

   """     



    # Load a website

    driver = webdriver.Chrome()

    driver.get(#some website)



    # Prepare extra xpath syntax

    add_extra_xpath = {}

    add_extra_xpath['add me'] = 1 # The actual dict is more complex then this and 

                                  # this number keeps changing so that's why I'm using a 

                                  # dict, if you're wondering



    return driver, add_extra_xpath





def function_where_I_have_a_problem(driver, add_extra_xpath):

    """

    Simply find an element on the site via a connotated xpath

    """



    # Find an element by xpath with the help of "add_extra_path"

    find = driver.find_element_by_xpath('begin writing xpath' + str(add_extra_xpath['add me']) + 'finish off the xpath')

test_script.py

import pytest

import script



from unittest.mock import MagicMock



# Mock the "driver" and stub the "add_extra_xpath" dictionary

@pytest.fixture

def fixture():



    driver = MagicMock()

    driver.find_element_by_xpath.side_effect = driver



    add_extra_xpath = {}

    add_extra_xpath['add me'] = 1



    return driver, add_extra_xpath



def test_function_where_I_have_a_problem(fixture):



    driver, add_extra_xpath = fixture



    script.function_where_I_have_a_problem(driver, add_extra_xpath)



    assert driver.call_count == 1

Running the test crashes due to an:

AttributeError: 'dict' object has no attribute 'find_element_by_xpath'

at

find = driver.find_element_by_xpath('begin writing xpath' + str(add_extra_xpath['add me']) + 'finish off the xpath')

Shouldn't the side_effect I set back in my fixture override this 'find_element_by_xpath'?