From 260fbb598ed6936131fbcbb8e869439c09e90843 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Mon, 12 Dec 2022 21:00:40 -0800
Subject: [PATCH] Fix some failing tests, refs #1855

---
 tests/test_cli.py         |  1 +
 tests/test_permissions.py | 36 +++++++++++++++++++++---------------
 2 files changed, 22 insertions(+), 15 deletions(-)

diff --git a/tests/test_cli.py b/tests/test_cli.py
index f0d28037..d3e015fa 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -138,6 +138,7 @@ def test_metadata_yaml():
         settings=[],
         secret=None,
         root=False,
+        token=None,
         version_note=None,
         get=None,
         help_settings=False,
diff --git a/tests/test_permissions.py b/tests/test_permissions.py
index bc556e4c..1fc2c8a0 100644
--- a/tests/test_permissions.py
+++ b/tests/test_permissions.py
@@ -692,7 +692,9 @@ PermMetadataTestCase = collections.namedtuple(
         # create-table on database - no resource specified
         PermMetadataTestCase(
             metadata={
-                "databases": {"db1": {"permissions": {"create-table": {"id": "user"}}}}
+                "databases": {
+                    "perms_ds_one": {"permissions": {"create-table": {"id": "user"}}}
+                }
             },
             actor={"id": "user"},
             action="create-table",
@@ -702,11 +704,13 @@ PermMetadataTestCase = collections.namedtuple(
         # create-table on database
         PermMetadataTestCase(
             metadata={
-                "databases": {"db1": {"permissions": {"create-table": {"id": "user"}}}}
+                "databases": {
+                    "perms_ds_one": {"permissions": {"create-table": {"id": "user"}}}
+                }
             },
             actor={"id": "user"},
             action="create-table",
-            resource="db1",
+            resource="perms_ds_one",
             expected_result=True,
         ),
         # insert-row on root, wrong actor
@@ -714,7 +718,7 @@ PermMetadataTestCase = collections.namedtuple(
             metadata={"permissions": {"insert-row": {"id": "user"}}},
             actor={"id": "user2"},
             action="insert-row",
-            resource=("db1", "t1"),
+            resource=("perms_ds_one", "t1"),
             expected_result=False,
         ),
         # insert-row on root, right actor
@@ -722,24 +726,26 @@ PermMetadataTestCase = collections.namedtuple(
             metadata={"permissions": {"insert-row": {"id": "user"}}},
             actor={"id": "user"},
             action="insert-row",
-            resource=("db1", "t1"),
+            resource=("perms_ds_one", "t1"),
             expected_result=True,
         ),
         # insert-row on database
         PermMetadataTestCase(
             metadata={
-                "databases": {"db1": {"permissions": {"insert-row": {"id": "user"}}}}
+                "databases": {
+                    "perms_ds_one": {"permissions": {"insert-row": {"id": "user"}}}
+                }
             },
             actor={"id": "user"},
             action="insert-row",
-            resource="db1",
+            resource="perms_ds_one",
             expected_result=True,
         ),
         # insert-row on table, wrong table
         PermMetadataTestCase(
             metadata={
                 "databases": {
-                    "db1": {
+                    "perms_ds_one": {
                         "tables": {
                             "t1": {"permissions": {"insert-row": {"id": "user"}}}
                         }
@@ -748,14 +754,14 @@ PermMetadataTestCase = collections.namedtuple(
             },
             actor={"id": "user"},
             action="insert-row",
-            resource=("db1", "t2"),
+            resource=("perms_ds_one", "t2"),
             expected_result=False,
         ),
         # insert-row on table, right table
         PermMetadataTestCase(
             metadata={
                 "databases": {
-                    "db1": {
+                    "perms_ds_one": {
                         "tables": {
                             "t1": {"permissions": {"insert-row": {"id": "user"}}}
                         }
@@ -764,14 +770,14 @@ PermMetadataTestCase = collections.namedtuple(
             },
             actor={"id": "user"},
             action="insert-row",
-            resource=("db1", "t1"),
+            resource=("perms_ds_one", "t1"),
             expected_result=True,
         ),
         # view-query on canned query, wrong actor
         PermMetadataTestCase(
             metadata={
                 "databases": {
-                    "db1": {
+                    "perms_ds_one": {
                         "queries": {
                             "q1": {
                                 "sql": "select 1 + 1",
@@ -783,14 +789,14 @@ PermMetadataTestCase = collections.namedtuple(
             },
             actor={"id": "user2"},
             action="view-query",
-            resource=("db1", "q1"),
+            resource=("perms_ds_one", "q1"),
             expected_result=False,
         ),
         # view-query on canned query, right actor
         PermMetadataTestCase(
             metadata={
                 "databases": {
-                    "db1": {
+                    "perms_ds_one": {
                         "queries": {
                             "q1": {
                                 "sql": "select 1 + 1",
@@ -802,7 +808,7 @@ PermMetadataTestCase = collections.namedtuple(
             },
             actor={"id": "user"},
             action="view-query",
-            resource=("db1", "q1"),
+            resource=("perms_ds_one", "q1"),
             expected_result=True,
         ),
     ),