From 726f781c50e88f557437f6490b8479c3d6fabfc2 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Mon, 22 Feb 2021 16:22:47 -0800
Subject: [PATCH] Fix for arraycontains bug, closes #1239

---
 datasette/filters.py  | 4 ++--
 tests/test_filters.py | 8 ++++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/datasette/filters.py b/datasette/filters.py
index 152a26b4..2b859d99 100644
--- a/datasette/filters.py
+++ b/datasette/filters.py
@@ -150,7 +150,7 @@ class Filters:
                     "arraycontains",
                     "array contains",
                     """rowid in (
-            select {t}.rowid from {t}, json_each({t}.{c}) j
+            select {t}.rowid from {t}, json_each([{t}].[{c}]) j
             where j.value = :{p}
         )""",
                     '{c} contains "{v}"',
@@ -159,7 +159,7 @@ class Filters:
                     "arraynotcontains",
                     "array does not contain",
                     """rowid not in (
-            select {t}.rowid from {t}, json_each({t}.{c}) j
+            select {t}.rowid from {t}, json_each([{t}].[{c}]) j
             where j.value = :{p}
         )""",
                     '{c} does not contain "{v}"',
diff --git a/tests/test_filters.py b/tests/test_filters.py
index 75a779b9..f22b7b5c 100644
--- a/tests/test_filters.py
+++ b/tests/test_filters.py
@@ -56,6 +56,14 @@ import pytest
         # Not in, and JSON array not in
         ((("foo__notin", "1,2,3"),), ["foo not in (:p0, :p1, :p2)"], ["1", "2", "3"]),
         ((("foo__notin", "[1,2,3]"),), ["foo not in (:p0, :p1, :p2)"], [1, 2, 3]),
+        # JSON arraycontains
+        (
+            (("Availability+Info__arraycontains", "yes"),),
+            [
+                "rowid in (\n            select table.rowid from table, json_each([table].[Availability+Info]) j\n            where j.value = :p0\n        )"
+            ],
+            ["yes"],
+        ),
     ],
 )
 def test_build_where(args, expected_where, expected_params):