discourse/plugins/discourse-data-explorer/spec/system/query_runner_spec.rb

# frozen_string_literal: true

RSpec.describe "Data explorer query runner" do
  fab!(:admin)
  fab!(:group) { Fabricate(:group, name: "group") }
  fab!(:group_user) { Fabricate(:group_user, user: admin, group: group) }

  let(:query_runner) { PageObjects::Pages::DataExplorerQueryRunner.new }

  before do
    SiteSetting.data_explorer_enabled = true
    sign_in admin
  end

  context "with the mode switch in the header" do
    fab!(:query) { Fabricate(:query, name: "Query", sql: "SELECT 1", user: admin) }

    it "hides the switch when AI queries are disabled" do
      SiteSetting.data_explorer_ai_queries_enabled = false
      visit("/admin/plugins/discourse-data-explorer/queries/#{query.id}")
      expect(page).to have_no_css(".query-mode-switch")
    end

    it "shows the switch defaulted to AI and toggles to manual" do
      SiteSetting.data_explorer_ai_queries_enabled = true
      visit("/admin/plugins/discourse-data-explorer/queries/#{query.id}")
      expect(page).to have_css(".query-mode-switch input[value='ai']:checked", visible: :all)

      find(".query-mode-switch .d-segmented-control__label", text: "Write SQL").click
      expect(page).to have_css(".query-mode-switch input[value='manual']:checked", visible: :all)
    end
  end

  context "when navigating between queries" do
    fab!(:query_a) do
      Fabricate(:query, name: "Query A", sql: "SELECT * FROM users LIMIT 1", user: admin)
    end
    fab!(:query_b) do
      Fabricate(:query, name: "Query B", sql: "SELECT * FROM users LIMIT 1", user: admin)
    end

    it "clears results from a previously run query" do
      visit("/admin/plugins/discourse-data-explorer/queries/#{query_a.id}")
      find(".query-run-split__primary").click
      expect(page).to have_css(".query-results .result-header")

      find(".back-button").click
      first("a[href='/admin/plugins/discourse-data-explorer/queries/#{query_b.id}']").click
      expect(page).to have_no_css(".query-results .result-header")
    end
  end

  context "with the edit page layout" do
    fab!(:query) { Fabricate(:query, name: "Layout query", sql: "SELECT 1", user: admin) }

    it "shows the SQL editor with Help in the action bar (no Edit toggle)" do
      visit("/admin/plugins/discourse-data-explorer/queries/#{query.id}")

      expect(page).to have_no_css(".btn-edit-query")
      expect(page).to have_css(".query-action-bar__left .query-action-bar__help")
      expect(page).to have_css(".query-action-bar__left .query-run-split__primary")
    end

    it "puts Discard on the left and Delete on the right of the action bar" do
      visit("/admin/plugins/discourse-data-explorer/queries/#{query.id}")

      expect(page).to have_css(".query-action-bar__left .btn-discard-query[disabled]")
      expect(page).to have_css(".query-action-bar__right .btn-danger")
      expect(page).to have_no_css(".btn-save-query")
      expect(page).to have_no_css(".query-footer-actions")
    end

    it "switches the Run button label to 'Save changes and run' when dirty" do
      visit("/admin/plugins/discourse-data-explorer/queries/#{query.id}")

      expect(page).to have_css(".query-run-split__primary", text: I18n.t("js.explorer.run"))

      PageObjects::Components::AceEditor.new.set_input("SELECT 2")

      expect(page).to have_css(".query-run-split__primary", text: I18n.t("js.explorer.saverun"))
    end

    it "renders the chart/table toggle + Export-as dropdown in the action bar after results load" do
      visit("/admin/plugins/discourse-data-explorer/queries/#{query.id}")
      find(".query-run-split__primary").click

      expect(page).to have_css(".query-action-bar__right .query-results-modes")
      expect(page).to have_css(".query-action-bar__right .query-result-download-buttons")

      find(".query-action-bar__right .query-result-download-buttons").click

      expect(page).to have_css(".query-result-export__results-json")
      expect(page).to have_css(".query-result-export__results-csv")
      expect(page).to have_css(".query-result-export__query-json")
    end

    it "hides Edit/Save/Discard on default queries" do
      visit("/admin/plugins/discourse-data-explorer/queries/-1")

      expect(page).to have_no_css(".btn-edit-query")
      expect(page).to have_no_css(".btn-save-query")
      expect(page).to have_no_css(".btn-discard-query")
    end
  end

  context "with a query using a default param" do
    fab!(:query_1) do
      Fabricate(
        :query,
        name: "My default param query",
        description: "Test default param query",
        sql: "-- [params]\n-- string :limit = 42\n\nSELECT * FROM users LIMIT :limit",
        user: admin,
      )
    end
    fab!(:query_group_1) { Fabricate(:query_group, query: query_1, group: group) }

    it "pre-fills the field with the default param" do
      query_runner.visit_group_report("group", query_1.id)

      expect(query_runner).to have_param_field("limit", "42")
    end

    it "allows to edit custom name" do
      query_runner.visit_admin_query(query_1.id).run_query.click_edit_name
      query_runner.fill_query_name("My custom name edited").click_save_and_run

      expect(query_runner).to have_query_name("My custom name edited")
    end
  end

  context "with the old url format" do
    fab!(:query_1) do
      Fabricate(
        :query,
        name: "My query",
        description: "Test query",
        sql: "SELECT * FROM users",
        user: admin,
      )
    end

    it "redirects /explorer/queries/:id to the new url format" do
      visit("/admin/plugins/explorer/queries/#{query_1.id}")

      expect(page).to have_current_path(
        "/admin/plugins/discourse-data-explorer/queries/#{query_1.id}",
      )
    end

    it "redirects /explorer/queries to the new url format" do
      visit("/admin/plugins/explorer/queries")

      expect(page).to have_current_path("/admin/plugins/discourse-data-explorer/queries")
    end

    it "redirects /explorer to the new url format" do
      visit("/admin/plugins/explorer")

      expect(page).to have_current_path("/admin/plugins/discourse-data-explorer/queries")
    end
  end

  describe "caching results with query params" do
    fab!(:user_jan) { Fabricate(:user, created_at: Time.parse("2025-01-15")) }
    fab!(:user_feb) { Fabricate(:user, created_at: Time.parse("2025-02-15")) }
    fab!(:user_mar) { Fabricate(:user, created_at: Time.parse("2025-03-15")) }
    fab!(:user_apr) { Fabricate(:user, created_at: Time.parse("2025-04-15")) }

    fab!(:query) { Fabricate(:query, name: "Monthly signups", sql: <<~SQL, user: admin) }
      -- [params]
      -- date :start_date = 2025-01-01
      -- date :end_date = 2025-04-30
      SELECT
        date_trunc('month', u.created_at)::date AS month,
        COUNT(*) AS signups
      FROM users u
      WHERE u.created_at >= :start_date::date
        AND u.created_at < :end_date::date
      GROUP BY month
      ORDER BY month
    SQL

    let(:query_runner) { PageObjects::Pages::DataExplorerQueryRunner.new }

    before do
      DiscourseDataExplorer::QueryRunner.invalidate(query.id)
      SiteSetting.data_explorer_enabled = true
      sign_in admin
    end

    it "caches results and shows them on reload" do
      query_runner.visit_admin_query(query.id)
      expect(query_runner).to have_no_result_header

      query_runner.run_query

      expect(query_runner).to have_result_header
      expect(query_runner).to have_no_cached_result_notice
      expect(query_runner).to have_result_row_count(4)

      query_runner.visit_admin_query(
        query.id,
        params: {
          start_date: "2025-01-01",
          end_date: "2025-03-31",
        },
      )
      expect(query_runner).to have_no_result_header

      query_runner.run_query
      expect(query_runner).to have_result_header
      expect(query_runner).to have_result_row_count(3)
      expect(query_runner).to have_no_cached_result_notice

      query_runner.visit_admin_query(query.id)

      expect(query_runner).to have_result_header
      expect(query_runner).to have_result_row_count(4)
      expect(query_runner).to have_cached_result_notice
    end

    it "forces fresh execution with ?run=1 even when cache exists" do
      query_runner.visit_admin_query(query.id)
      query_runner.run_query
      expect(query_runner).to have_result_header

      query_runner.visit_admin_query(query.id, query_string: "run=true")
      expect(query_runner).to have_result_header
      expect(query_runner).to have_no_cached_result_notice
    end
  end
end