diff --git a/src/mango/test/09-text-sort-test.py b/src/mango/test/09-text-sort-test.py deleted file mode 100644 index c0c36ccd0d5..00000000000 --- a/src/mango/test/09-text-sort-test.py +++ /dev/null @@ -1,115 +0,0 @@ -# Licensed under the Apache License, Version 2.0 (the "License"); you may not -# use this file except in compliance with the License. You may obtain a copy of -# the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations under -# the License. - -import mango -import unittest - - -@unittest.skipUnless(mango.has_text_service(), "requires text service") -class SortTests(mango.UserDocsTextTests): - def test_number_sort(self): - q = {"age": {"$gt": 0}} - docs = self.db.find(q, sort=["age:number"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["age"], 22) - - def test_number_sort_desc(self): - q = {"age": {"$gt": 0}} - docs = self.db.find(q, sort=[{"age": "desc"}]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["age"], 79) - - q = {"manager": True} - docs = self.db.find(q, sort=[{"age:number": "desc"}]) - self.assertEqual(len(docs), 10) - self.assertEqual(docs[0]["age"], 79) - - def test_string_sort(self): - q = {"email": {"$gt": None}} - docs = self.db.find(q, sort=["email:string"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["email"], "abbottwatson@talkola.com") - - def test_notype_sort(self): - q = {"email": {"$gt": None}} - try: - self.db.find(q, sort=["email"]) - except Exception as e: - self.assertEqual(e.response.status_code, 400) - else: - raise AssertionError("Should have thrown error for sort") - - def test_array_sort(self): - q = {"favorites": {"$exists": True}} - docs = self.db.find(q, sort=["favorites.[]:string"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["user_id"], 8) - - def test_multi_sort(self): - q = {"name": {"$exists": True}} - docs = self.db.find(q, sort=["name.last:string", "age:number"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["name"], {"last": "Ewing", "first": "Shelly"}) - self.assertEqual(docs[1]["age"], 22) - - def test_guess_type_sort(self): - q = {"$or": [{"age": {"$gt": 0}}, {"email": {"$gt": None}}]} - docs = self.db.find(q, sort=["age"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["age"], 22) - - def test_guess_dup_type_sort(self): - q = { - "$and": [ - {"age": {"$gt": 0}}, - {"email": {"$gt": None}}, - {"age": {"$lte": 100}}, - ] - } - docs = self.db.find(q, sort=["age"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["age"], 22) - - def test_ambiguous_type_sort(self): - q = {"$or": [{"age": {"$gt": 0}}, {"email": {"$gt": None}}, {"age": "34"}]} - try: - self.db.find(q, sort=["age"]) - except Exception as e: - self.assertEqual(e.response.status_code, 400) - else: - raise AssertionError("Should have thrown error for sort") - - def test_guess_multi_sort(self): - q = { - "$or": [ - {"age": {"$gt": 0}}, - {"email": {"$gt": None}}, - {"name.last": "Harvey"}, - ] - } - docs = self.db.find(q, sort=["name.last", "age"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["name"], {"last": "Ewing", "first": "Shelly"}) - self.assertEqual(docs[1]["age"], 22) - - def test_guess_mix_sort(self): - q = { - "$or": [ - {"age": {"$gt": 0}}, - {"email": {"$gt": None}}, - {"name.last": "Harvey"}, - ] - } - docs = self.db.find(q, sort=["name.last:string", "age"]) - self.assertEqual(len(docs), 15) - self.assertEqual(docs[0]["name"], {"last": "Ewing", "first": "Shelly"}) - self.assertEqual(docs[1]["age"], 22) diff --git a/test/elixir/test/config/search.elixir b/test/elixir/test/config/search.elixir index 87715d4caf3..ee6434f4f3f 100644 --- a/test/elixir/test/config/search.elixir +++ b/test/elixir/test/config/search.elixir @@ -38,5 +38,18 @@ ], "LimitTests": [ "limit field" + ], + "SortTests": [ + "number sort", + "number sort desc", + "string sort", + "notype sort", + "array sort", + "multi sort", + "guess type sort", + "guess dup type sort", + "guess ambiguous type sort", + "guess multi sort", + "guess mix sort" ] } diff --git a/test/elixir/test/mango/09_text_sort_test.exs b/test/elixir/test/mango/09_text_sort_test.exs new file mode 100644 index 00000000000..948eba6050e --- /dev/null +++ b/test/elixir/test/mango/09_text_sort_test.exs @@ -0,0 +1,148 @@ +# Licensed under the Apache License, Version 2.0 (the "License"); you may not +# use this file except in compliance with the License. You may obtain a copy of +# the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT +# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the +# License for the specific language governing permissions and limitations under +# the License. + +defmodule SortTests do + use CouchTestCase + + @db_name "sort-docs" + + setup do + UserDocs.setup(@db_name, "text") + end + + test "number sort" do + q = %{"age" => %{"$gt" => 0}} + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["age:number"]) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["age"] == 22 + end + + test "number sort desc" do + q = %{"age" => %{"$gt" => 0}} + sort = [%{"age" => "desc"}] + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: sort) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["age"] == 79 + + q = %{"manager" => true} + sort = [%{"age:number" => "desc"}] + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: sort) + + assert length(docs) == 10 + assert Enum.at(docs, 0)["age"] == 79 + end + + test "string sort" do + q = %{"email" => %{"$gt" => nil}} + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["email:string"]) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["email"] == "abbottwatson@talkola.com" + end + + test "notype sort" do + q = %{"email" => %{"$gt" => nil}} + {:error, resp} = MangoDatabase.find(@db_name, q, sort: ["email"]) + + assert resp.status_code == 400 + assert resp.body["error"] == "text_sort_error" + assert resp.body["reason"] == "Unspecified or ambiguous sort type. Try appending :number or :string to the sort field. email" + end + + test "array sort" do + q = %{"favorites" => %{"$exists" => true}} + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["favorites.[]:string"]) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["user_id"] == 8 + end + + test "multi sort" do + q = %{"name" => %{"$exists" => true}} + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["name.last:string", "age:number"]) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["name"] == %{"first" => "Shelly", "last" => "Ewing"} + assert Enum.at(docs, 1)["age"] == 22 + end + + test "guess type sort" do + q = %{ + "$or" => [ + %{"age" => %{"$gt" => 0}}, + %{"email" => %{"$gt" => nil}} + ] + } + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["age"]) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["age"] == 22 + end + + test "guess dup type sort" do + q = %{ + "$and" => [ + %{"age" => %{"$gt" => 0}}, + %{"email" => %{"$gt" => nil}}, + %{"age" => %{"$lte" => 100}}, + ] + } + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["age"]) + assert length(docs) == 15 + assert Enum.at(docs, 0)["age"] == 22 + end + + test "guess ambiguous type sort" do + q = %{ + "$or" => [ + %{"age" => %{"$gt" => 0}}, + %{"email" => %{"$gt" => nil}}, + %{"age" => "34"} + ] + } + {:error, resp} = MangoDatabase.find(@db_name, q, sort: ["age"]) + + assert resp.status_code == 400 + end + + test "guess multi sort" do + q = %{ + "$or" => [ + %{"age" => %{"$gt" => 0}}, + %{"email" => %{"$gt" => nil}}, + %{"name.last" => "Harvey"} + ] + } + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["name.last", "age"]) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["name"] == %{"first" => "Shelly", "last" => "Ewing"} + assert Enum.at(docs, 1)["age"] == 22 + end + + test "guess mix sort" do + q = %{ + "$or" => [ + %{"age" => %{"$gt" => 0}}, + %{"email" => %{"$gt" => nil}}, + %{"name.last" => "Harvey"} + ] + } + {:ok, docs} = MangoDatabase.find(@db_name, q, sort: ["name.last", "age"]) + + assert length(docs) == 15 + assert Enum.at(docs, 0)["name"] == %{"first" => "Shelly", "last" => "Ewing"} + assert Enum.at(docs, 1)["age"] == 22 + end +end \ No newline at end of file