Login via browser's cookies (Selenium WebDriver) or Local Storage (JavaScript)


How to avoid repeating login action before running each test?

It depends on how this was implemented in your Web application (via cookies or Local Storage).

So, let’s start with cookies. The following algorithm should be used in this case (with help of Selenium WebDriver):

  • Login as by usual (via email and password);
  • Save cookies of your current session;
  • Open the site and load saved cookies.

After step 3 you will login into the application as the same after step 1. Now I will demonstrate this with the following code snippets:

"""Module Utilities with different useful methods."""

...
import pickle
import json
...


class Utilities:
    """Class Utilities with different useful methods."""

    ...

    def save_cookie(self, name):
        """A method for saving cookies for domain."""

        with open(name + ".pkl", "wb") as filehandler:

            pickle.dump(self.driver.get_cookies(), filehandler)

    def load_cookie(self, name):
        """A method for loading cookies for domain."""

        with open(name + ".pkl", "rb") as cookiesfile:

            cookies = pickle.load(cookiesfile)

            for cookie in cookies:

                self.driver.add_cookie(cookie)

Note that you can save and load cookies only for current domain.

What if is your Web application using Local Storage? In this case you can use the following method in Utilities class:

def set_local_storage(self, key, value):
    """A method for setting key in local storage."""

    self.driver.execute_script("window.localStorage.setItem('{}',{})".format(key, json.dumps(value)))

Using this method you can set up all needed pair of key/value in Local Storage. Then you will see that you are logged in (after update the page).

Now let’s consider more practical example -- login to Facebook via pre-saved cookies of user. First of all, we need to login as by usual and save cookies of user:


"""Module for saving cookies for the domain."""

import time

import config
from utilities import Utilities


class GetCookies(Utilities):
    """Class for saving cookies for the domain."""

    def __init__(self, email="", password=""):
        """Method for saving cookies for the domain."""

        config.DOMAIN = "https://facebook.com"
        self._set_up()

        # Go to URL.
        self.driver.get(config.DOMAIN)

        # Finding elements on the page and actions.
        email_text_field_element = self.wait_visibility_by_id("email")
        email_text_field_element.clear()
        email_text_field_element.send_keys(email)

        password_text_field_element = self.find_by_id("pass")
        password_text_field_element.clear()
        password_text_field_element.send_keys(password)

        login_button = self.find_by_id("loginbutton")
        self.click(login_button)

        # Saving user's cookies.
        self.save_cookie("ratmir.asanov.demo")
        time.sleep(config.DELAY2)

        self._tear_down()


if __name__ == '__main__':
    COOKIE = GetCookies(config.USER1['email'], config.USER1['password'])

Then we need to login with using pre-saved user’s cookies:


"""Module for loading cookies for the domain."""

import time

import config
from utilities import Utilities


class LoadCookies(Utilities):
    """Class for loading cookies for the domain."""

    def __init__(self, name_of_cookies_file):
        """Method for loading cookies for the domain."""

        config.DOMAIN = "https://www.facebook.com"
        self._set_up()

        # Go to URL.
        self.driver.get(config.DOMAIN)
        time.sleep(config.DELAY2)

        # Loading user's pre-saved cookies.
        self.load_cookie(name_of_cookies_file)
        time.sleep(config.DELAY2)

        # Refresh the page to see the changes (user should be logged in).
        self.refresh_page()
        time.sleep(config.DELAY2 * 5)

        self._tear_down()


if __name__ == '__main__':
    COOKIE = LoadCookies("ratmir.asanov.demo")

Hope it helps you!

Click here for source code.
Used references: