2

我正在尝试测试登录用户是否可以使用 Lettuce、Selenium 和 lettuce_webdriver 在我的 Django 站点上注销。

在我的 terrain.py 中，我有：

@before.all
def setup_browser():
    profile = webdriver.FirefoxProfile()
    profile.set_preference('network.dns.disableIPv6', True)
    world.browser = webdriver.Firefox(profile)
    world.client = Client(HTTP_USER_AGENT='Mozilla/5.0')

然后当我“登录”时：

@step(r'I am logged in as "(\w*)"')
def log_in(step, name):
    world.client.login(username=name, password=name)

我去我的网站：

And I go to "localhost:8000"
    I find a link called "Logout ?" that goes to "/logout"

@step(r'I find a link called "(.*?)" that goes to "(.*?)"$')
def find_link(step, link_name, link_url):
    print(world.browser.page_source)
    elem = world.browser.find_element_by_xpath(r'//a[@href="%s"]' % link_url)
    eq_(elem.text, link_name)

但是我的 page_source 显示我没有登录。这种情况是有道理的......在这方面client并且browser没有互相交谈。但这可能吗，还是我需要通过单击 selenium 等链接“手动”登录？

我想这样做：

 world.browser.page_source = world.client.get(world.browser.current_url).content

但是 page_source 不能改变。我可以以某种方式从 django 的客户端提供 Selenium 吗？

编辑：按照下面的 Loius建议，我的“我以 ... 身份登录”步骤如下。我添加了 if/else 来检查我的怀疑。我的客户仍然如上设置（见setup_browser上面的步骤）

@step(r'I am logged in as "(\w*)"')
def log_in(step, name):
    world.client.login(username=name, password=name)
    if world.client.cookies:
        session_key = world.client.cookies["sessionid"].value
        world.browser.add_cookie({'name':'sessionid', 'value':session_key})
        world.browser.refresh()
    else: 
        raise Exception("No Cookies!")

我看到的所有建议都是先登录。没有我的检查，我得到这个：

  Scenario: Logged in users can logout                       # \gantt_charts\features\index.feature:12
    Given I am logged in as "elsepeth"                                # \gantt_charts\features\steps.py:25
    Traceback (most recent call last):
      File "C:\Python34\lib\site-packages\lettuce\core.py", line 144, in __call__
        ret = self.function(self.step, *args, **kw)
      File "D:\Django_Projects\gAnttlr\gantt_charts\features\steps.py", line 27, in log_in
        session_key = world.client.cookies["sessionid"].value
    KeyError: 'sessionid'
4

1 回答 1

3

我没有尝试过完全按照您的要求做，但我做过类似的事情。在您使用实例登录 Django 站点后，您需要做的是在您的 Selenium 实例上设置这样的 cookie Client：

driver.add_cookie({'name': 'sessionid', 'value': session_key})

该名称应与您SESSION_COOKIE_NAME在 Django 站点上的设置相同（sessionid默认设置）。你需要弄清楚 的价值session_key。

您可以从这样的Client实例中获取它：

    session_key = client.cookies["sessionid"].value

请注意，如果是，Selenium 将无法为某些浏览器设置 cookie 。您应该将此设置用于您的生产服务器，但如果您希望 Selenium 测试设置您的会话 cookie，那么您必须将其用于测试。SESSION_COOKIE_SECURETrueTrueFalse

一旦你的 Selenium 实例拥有了 cookie，它会在 Django 中看起来好像你已经使用 Selenium 登录了它。正如我所说，我在我的测试套件中做了类似的事情。（我使用了其他东西，Client但原理是一样的。）

于 2015-01-16T18:20:42.000 回答