AttributeError：'TestLogin'对象没有属性'driver'

Question

So, my question is related to AttributeError: 'LoginPage' object has no attribute 'driver' , but answer doesn't solve my issue.

I'm creating test cases using Python and Appium framework with page object model and I need some code to be executed before every test case, so I use setUp method for this in base_test.py file.

base_test.py

class BaseTest(object):
    def setUp(self):
        self.driver = webdriver.Remote(server_url,device_config)
        self.os = str(self.driver.caps['platformName']).lower()

TestLogin class extends BaseTest class.

test_login.py

class TestLogin(BaseTest):
    def __init__(self):
        self.login_page = LoginPage(self.driver)

    def test_00_login(self):
        self.login_page.openLoginPage()

LoginPage class needs driver object because all actions happen in page classes.

If I execute this script I get an error AttributeError: 'TestLogin' object has no attribute 'driver' .

Now, if I change my TestLogin class to:

class TestLogin(BaseTest):
    def __init__(self):
        setup = BaseTest()
        setup.setUp()
        self.login_page = LoginPage(setup.driver)

    def test_00_login(self):
        self.login_page.openLoginPage()

I won't receive an error but setUp method will be executed 2 times, first on setup.setUp() and then in test_00_login(self) method. This is not what I want.

Third way would be to move self.login_page = LoginPage(self.driver) to the method test_00_login(self) , but this way I would need to have self.login_page = LoginPage(self.driver) in every future method I write in TestLogin class.

class TestLogin(BaseTest):
    def test_00_login(self):
        self.login_page = LoginPage(self.driver)
        self.login_page.openLoginPage()

Is there a way of having self.login_page = LoginPage(self.driver) on top level of TestLogin class, like in __init__(self) method, but without getting attribute error? I need a better way of handling it.

Thank you.

Answer 1

that because you didn't defined (driver):

class TestLogin(BaseTest):
driver = 'driver'
def __init__(self):
    self.login_page = LoginPage(self.driver)

then:

TestLogin(Something).__init__()

you just forgot to define driver you must define it in the class (TestLogin) not in others class

Answer 2

i found other answer for you , you must define self.driver before everything:

from webdriver import Remote
import webdriver 

self = webdriver
driver  = Remote(server_url,device_config)

class BaseTest(object):
def setUp(self):
    self.os = str(self.driver.caps['platformName']).lower()

class TestLogin(BaseTest):
    def __init__(self):
        self.login_page = LoginPage(self.driver)

    def test_00_login(self):
        self.login_page.openLoginPage()

    class TestLogin(BaseTest):
       def test_00_login(self):
         self.login_page = LoginPage(self.driver)
         self.login_page.openLoginPage()

then what ever you want , don't forget to define other syntaxs