##########################################################################
#
# pgAdmin 4 - PostgreSQL Tools
#
# Copyright (C) 2013 - 2019, The pgAdmin Development Team
# This software is released under the PostgreSQL Licence
#
##########################################################################

from __future__ import print_function
import sys
import random
import time

from regression.python_test_utils import test_utils
from regression.feature_utils.base_feature_test import BaseFeatureTest
from selenium.common.exceptions import TimeoutException, \
    StaleElementReferenceException


class BrowserToolBarFeatureTest(BaseFeatureTest):
    """
        This feature test will test the tool bar on Browser panel.
    """

    scenarios = [
        ("Browser tool bar feature test", dict())
    ]

    test_table_name = ""

    def before(self):
        self.page.wait_for_spinner_to_disappear()
        self.page.add_server(self.server)
        self.test_table_name = "test_table" + str(random.randint(1000, 3000))
        test_utils.create_table(self.server, self.test_db,
                                self.test_table_name)

    def runTest(self):
        # Check for query tool button
        print("\nQuery Tool ToolBar Button ",
              file=sys.stderr, end="")
        self.test_query_tool_button()
        print("OK.", file=sys.stderr)

        # Check for view data button
        print("\nView Data ToolBar Button ",
              file=sys.stderr, end="")
        self.test_view_data_tool_button()
        print("OK.", file=sys.stderr)

        # Check for filtered rows button
        print("\nFiltered Rows ToolBar Button ",
              file=sys.stderr, end="")
        self.test_filtered_rows_tool_button()
        print("OK.", file=sys.stderr)

    def after(self):
        self.page.remove_server(self.server)

    def _locate_database_tree_node(self):
        self.page.toggle_open_tree_item(self.server['name'])
        self.page.toggle_open_tree_item('Databases')
        self.page.toggle_open_tree_item(self.test_db)

    def test_query_tool_button(self):
        self._locate_database_tree_node()

        retry_count = 0
        while retry_count < 5:
            try:
                self.page.find_by_css_selector(
                    ".wcFrameButton[title='Query Tool']:not(.disabled)")\
                    .click()
                break
            except StaleElementReferenceException:
                retry_count += 1

        time.sleep(0.5)
        self.page.find_by_css_selector(".wcPanelTab .wcTabIcon.fa.fa-bolt")

    def test_view_data_tool_button(self):
        self.page.select_tree_item(self.test_db)
        self.page.toggle_open_tree_item('Schemas')
        self.page.toggle_open_tree_item('public')
        self.page.toggle_open_tables_node()
        self.page.select_tree_item(self.test_table_name)

        retry_count = 0
        while retry_count < 5:
            try:
                self.page.find_by_css_selector(
                    ".wcFrameButton[title='View Data']:not(.disabled)").click()
                break
            except StaleElementReferenceException:
                retry_count += 1

        time.sleep(0.5)
        self.page.find_by_css_selector(".wcPanelTab .wcTabIcon.fa.fa-bolt")

    def test_filtered_rows_tool_button(self):
        retry_count = 0
        while retry_count < 5:
            try:
                self.page.find_by_css_selector(
                    ".wcFrameButton[title='Filtered Rows']:not(.disabled)")\
                    .click()
                break
            except StaleElementReferenceException:
                retry_count += 1

        time.sleep(0.5)
        self.page.find_by_css_selector(
            ".alertify .ajs-header[data-title~='Filter']")
        self.page.click_modal('Cancel')