#!/usr/bin/env python
"""Tests for API client and flows-related API calls."""

from absl import app

from grr_response_proto import objects_pb2
from grr_response_server import data_store
from grr_response_server.gui import api_integration_test_lib
from grr.test_lib import test_lib


class AuditTest(api_integration_test_lib.ApiIntegrationTest):

  def testFlowIsAudited(self):

    self.api.SearchClients(query=".")

    entries = data_store.REL_DB.ReadAPIAuditEntries()
    self.assertNotEmpty(entries)
    entry = entries[-1]

    self.assertEqual(
        entry.http_request_path, "/api/v2/clients?count=50&offset=0&query=."
    )
    self.assertEqual(entry.response_code, objects_pb2.APIAuditEntry.Code.OK)
    self.assertEqual(entry.router_method_name, "SearchClients")
    self.assertEqual(entry.username, "api_test_robot_user")


def main(argv):
  test_lib.main(argv)


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