/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "planTestUtil.h"
#include "planner.h"

using namespace std;

class PlanOrderByTest : public PlannerTestBase {};

TEST_F(PlanOrderByTest, basic) {
  useDb("root", "test");

  // ORDER BY key is in the projection list
  run("SELECT c1 FROM t1 ORDER BY c1");
  // ORDER BY key is not in the projection list
  run("SELECT c1 FROM t1 ORDER BY c2");

  run("SELECT c1 AS a FROM t1 ORDER BY a");

  run("SELECT c1 + 10 AS a FROM t1 ORDER BY a");

  run("SELECT 1 FROM t1 ORDER BY c1");
}

TEST_F(PlanOrderByTest, expr) {
  useDb("root", "test");

  run("SELECT * FROM t1 ORDER BY c1 + 10, c2");

  run("SELECT c1 FROM st1 ORDER BY ts, _C0");
}

TEST_F(PlanOrderByTest, nullsOrder) {
  useDb("root", "test");

  run("SELECT * FROM t1 ORDER BY c1 DESC NULLS FIRST");
}

TEST_F(PlanOrderByTest, withGroupBy) {
  useDb("root", "test");

  run("SELECT SUM(c1) AS a FROM t1 GROUP BY c2 ORDER BY a");
}

TEST_F(PlanOrderByTest, withSubquery) {
  useDb("root", "test");

  run("SELECT ts FROM (SELECT * FROM t1 ORDER BY ts DESC) ORDER BY ts DESC");
}

TEST_F(PlanOrderByTest, stable) {
  useDb("root", "test");

  // ORDER BY key is in the projection list
  run("SELECT c1 FROM st1 ORDER BY c1");

  // ORDER BY key is not in the projection list
  run("SELECT c2 FROM st1 ORDER BY c1");

  run("SELECT c2 FROM st1 PARTITION BY c2 ORDER BY c1");

  run("SELECT c1 AS a FROM st1 ORDER BY a");
}