#!/usr/bin/env python
from absl import app

from grr_response_server import data_store
from grr_response_server.gui import gui_test_lib
from grr.test_lib import test_lib


class ClientTest(gui_test_lib.GRRSeleniumTest):

  def testLabelsAreShownAndCanBeRemoved(self):
    client_id = self.SetupClient(0)
    data_store.REL_DB.AddClientLabels(
        client_id, self.test_username, ["mylabel", "foobar"]
    )

    self.Open(f"/v2/clients/{client_id}")
    self.WaitUntil(self.IsTextPresent, client_id)
    self.WaitUntil(self.IsTextPresent, "mylabel")
    self.WaitUntil(self.IsTextPresent, "foobar")

    self.Click("css=mat-chip:contains('mylabel') [matchipremove]")
    self.WaitUntilNot(self.IsTextPresent, "mylabel")

    self.assertEqual(
        ["foobar"],
        [l.name for l in data_store.REL_DB.ReadClientLabels(client_id)],
    )

  def testClientTimelineOpensInDrawer(self):
    mem_size = "1.00 KiB"
    client_id = self.SetupClient(0, fqdn="foo.bar", memory_size=1024)

    self.Open(f"/v2/clients/{client_id}")
    self.WaitUntil(self.IsTextPresent, "foo.bar")
    # Test that data only shows up after opening the details drawer. If
    # the data is every moved to the client summary shown on the client page,
    # use a different field for testing.
    self.WaitUntilNot(self.IsTextPresent, mem_size)

    element = self.WaitUntil(
        self.GetVisibleElement, "css=a:contains('View details')"
    )
    self.assertEndsWith(
        element.get_attribute("href"),
        "/v2/clients/C.1000000000000000/flows(drawer:details/C.1000000000000000)",
    )
    element.click()

    self.WaitUntilContains("drawer", self.GetCurrentUrlPath)
    self.WaitUntil(self.IsTextPresent, mem_size)

  def testDeepLinkToClientTimelineOpensInDrawer(self):
    mem_size = "1.00 KiB"
    client_id = self.SetupClient(0, fqdn="foo.bar", memory_size=1024)

    self.Open(f"/v2/clients/{client_id}")
    self.WaitUntil(self.IsTextPresent, "foo.bar")
    # Test that data only shows up after opening the details drawer. If
    # the data is every moved to the client summary shown on the client page,
    # use a different field for testing.
    self.WaitUntilNot(self.IsTextPresent, mem_size)

    element = self.WaitUntil(
        self.GetVisibleElement, "css=a:contains('View details')"
    )
    self.assertEndsWith(
        element.get_attribute("href"),
        "/v2/clients/C.1000000000000000/flows(drawer:details/C.1000000000000000)",
    )
    element.click()

    self.WaitUntilContains("drawer", self.GetCurrentUrlPath)
    self.WaitUntil(self.IsTextPresent, mem_size)

    # Reload the web page to simulate a deep link directly to the drawer.
    self.Open(self.GetCurrentUrlPath())
    self.WaitUntil(self.IsTextPresent, mem_size)

  def testFileFlowMenuHasLinkToVfsView(self):
    client_id = self.SetupClient(0)
    self.RequestAndGrantClientApproval(client_id)

    self.Open(f"/v2/clients/{client_id}")
    self.Click("css=button:contains('Collect files')")
    self.Click("css=button:contains('Browse the filesystem')")

    self.WaitUntil(self.GetVisibleElement, "css=app-vfs-section")

  def testBackButtonNavigatesToOldUi(self):
    client_id = self.SetupClient(0)
    self.Open(f"/v2/clients/{client_id}")

    self.WaitUntil(self.IsElementPresent, "css=a#fallback-link")
    self.Click("css=a#fallback-link")

    self.WaitUntilEqual(
        f"/legacy#/clients/{client_id}/host-info", self.GetCurrentUrlPath
    )


if __name__ == "__main__":
  app.run(test_lib.main)