automations crate and various chores

crates/agent-sql/tests/connector_tags.rs

@@ -15,8 +15,8 @@ async fn resource_path_pointers_cannot_be_changed() {
     let row = sqlx::query!(
       r#"
       with setup_connectors as (
-        insert into connectors (image_name, external_url, title, short_description, logo_url)
-          values ('foo/image', 'http://test.test', '{"en-US": "foo"}', '{"en-US": "foo"}', '{"en-US": "foo"}')
+        insert into connectors (image_name, external_url, title, short_description, logo_url, recommended)
+          values ('foo/image', 'http://test.test', '{"en-US": "foo"}', '{"en-US": "foo"}', '{"en-US": "foo"}', false)
           returning id
       )
       insert into connector_tags (connector_id, image_tag) select id, ':test' as image_tag from setup_connectors

crates/agent-sql/tests/publications.rs

@@ -20,8 +20,8 @@ async fn test_finding_forbidden_connectors() {
           ('bb00000000000000', 'testConnectors/Allowed', '{}'::json, 'capture', 'allowed_image', 'bbbbbbbbbbbbbbbb', 'bbbbbbbbbbbbbbbb')
         ),
         p2 as (
-            insert into connectors (external_url, image_name, title, short_description, logo_url) values
-                ('http://example.com', 'allowed_image', '{"en-US": "foo"}'::json, '{"en-US": "foo"}'::json, '{"en-US": "foo"}'::json)
+            insert into connectors (external_url, image_name, title, short_description, logo_url, recommended) values
+                ('http://example.com', 'allowed_image', '{"en-US": "foo"}'::json, '{"en-US": "foo"}'::json, '{"en-US": "foo"}'::json, false)
         )
         select 1;
         "#,

crates/agent/src/integration_tests/harness.rs

@@ -126,14 +126,14 @@ impl TestHarness {
     async fn setup_test_connectors(&mut self) {
         sqlx::query!(r##"
             with source_image as (
-                insert into connectors (external_url, image_name, title, short_description, logo_url)
-                values ('http://test.test/', 'source/test', '{"en-US": "test"}', '{"en-US": "test"}', '{"en-US": "http://test.test/"}')
+                insert into connectors (external_url, image_name, title, short_description, logo_url, recommended)
+                values ('http://test.test/', 'source/test', '{"en-US": "test"}', '{"en-US": "test"}', '{"en-US": "http://test.test/"}', false)
                 on conflict(image_name) do update set title = excluded.title
                 returning id
             ),
             materialize_image as (
-                insert into connectors (external_url, image_name, title, short_description, logo_url)
-                values ('http://test.test/', 'materialize/test', '{"en-US": "test"}', '{"en-US": "test"}', '{"en-US": "http://test.test/"}')
+                insert into connectors (external_url, image_name, title, short_description, logo_url, recommended)
+                values ('http://test.test/', 'materialize/test', '{"en-US": "test"}', '{"en-US": "test"}', '{"en-US": "http://test.test/"}', false)
                 on conflict(image_name) do update set title = excluded.title
                 returning id
             ),

crates/agent/src/integration_tests/source_captures.rs

@@ -283,7 +283,7 @@ async fn test_source_captures_collection_name() {
 #[tokio::test]
 #[serial_test::serial]
 async fn test_source_capture_no_annotations() {
-    let mut harness = TestHarness::init("test_source_capture_no_annotations").await;
+    let harness = TestHarness::init("test_source_capture_no_annotations").await;
     let user_id = harness.setup_tenant("sheep").await;
 
     let draft = draft_catalog(serde_json::json!({

crates/agent/src/integration_tests/unknown_connectors.rs

@@ -6,7 +6,7 @@ use crate::integration_tests::harness::{draft_catalog, TestHarness};
 #[tokio::test]
 #[serial_test::serial]
 async fn test_forbidden_connector() {
-    let mut harness = TestHarness::init("test_forbidden_connector").await;
+    let harness = TestHarness::init("test_forbidden_connector").await;
     let user_id = harness.setup_tenant("sheep").await;
 
     let draft = draft_catalog(serde_json::json!({

supabase/migrations/00_polyfill.sql

@@ -0,0 +1,54 @@
+begin;
+
+-- Roles which are created by supabase: anon, authenticated, supabase_admin, service_role.
+
+create role stats_loader with login password 'stats_loader_password' bypassrls;
+create role marketplace_integration;
+create role gatsby_reader;
+create role github_action_connector_refresh;
+create role wgd_automation;
+create role reporting_user;
+
+-- Required for postgres to give ownership of catalog_stats to stats_loader.
+grant stats_loader to postgres;
+
+-- Required for stats materialization to create flow_checkpoints_v1 and flow_materializations_v2.
+grant create on schema public to stats_loader;
+
+-- TODO(johnny): Required for `authenticated` to own `drafts_ext` and `publication_specs_ext`.
+-- We should make them owed by postgres and grant usage instead.
+grant create on schema public to authenticated;
+
+-- The production database has a Flow materialization of Stripe customer data.
+-- This is a partial table which matches the portions we use today.
+create schema stripe;
+
+create table stripe.customers (
+    id text primary key,
+    address json,
+    "address/city" text,
+    "address/country" text,
+    "address/line1" text,
+    "address/line2" text,
+    "address/postal_code" text,
+    "address/state" text,
+    balance bigint,
+    created bigint,
+    currency text,
+    default_source text,
+    delinquent boolean,
+    description text,
+    email text,
+    invoice_prefix text,
+    invoice_settings json,
+    "invoice_settings/custom_fields" json,
+    "invoice_settings/default_payment_method" text,
+    metadata json,
+    name text,
+    phone text,
+    flow_document json not null
+);
+
+grant usage on schema stripe to postgres;
+
+commit;