-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
WIP: spike out unit test suite for prop defs query gen
- Loading branch information
Showing
4 changed files
with
303 additions
and
11 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,292 @@ | ||
| use property_defs_rs::api::v1::query::Manager; | ||
|
|
||
| use sqlx::{postgres::PgArguments, Arguments, PgPool}; | ||
| use uuid::Uuid; | ||
|
|
||
| #[sqlx::test(migrations = "./tests/test_migrations")] | ||
| async fn test_simple_events_query(test_pool: PgPool) { | ||
| // seed the test DB | ||
| bootstrap_seed_data(test_pool.clone()).await.unwrap(); | ||
|
|
||
| // run some raw prop defs queries with various parameters | ||
| // and assert expected rows are returned | ||
| let _qmgr = Manager::new(test_pool.clone()); | ||
|
|
||
| // TODO: sanity check count and props queries; explore corner cases in other tests later :) | ||
| } | ||
|
|
||
| async fn bootstrap_seed_data(test_pool: PgPool) -> Result<(), sqlx::Error> { | ||
| // posthog_propertydefinition: (id, name, project_id, team_id, is_numerical, type, property_type, group_type_index) | ||
| let pd_rows = [ | ||
| // PropertyParentType::Event | ||
| (Uuid::now_v7(), "user_email", 1, 1, false, 1, "String", -1), | ||
| (Uuid::now_v7(), "utm_source", 1, 1, false, 1, "String", -1), | ||
| (Uuid::now_v7(), "$time", 1, 1, false, 1, "DateTime", -1), | ||
| (Uuid::now_v7(), "$sent_at", 1, 1, false, 1, "DateTime", -1), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "attempted_event_type", | ||
| 1, | ||
| 1, | ||
| true, | ||
| 1, | ||
| "Numeric", | ||
| -1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "$screen_width", | ||
| 1, | ||
| 1, | ||
| true, | ||
| 1, | ||
| "Numeric", | ||
| -1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "session_timeout_ms", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 1, | ||
| "Duration", | ||
| -1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "$dead_clicks_enabled_server_side", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 1, | ||
| "Boolean", | ||
| -1, | ||
| ), | ||
| // feature flag (filterable w/query param) | ||
| ( | ||
| Uuid::now_v7(), | ||
| "$feature/foo-bar-baz", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 1, | ||
| "Boolean", | ||
| -1, | ||
| ), | ||
| // PropertyParentType::Person | ||
| ( | ||
| Uuid::now_v7(), | ||
| "$feature_enrollment/artificial-hog", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 2, | ||
| "Boolean", | ||
| -1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "$survey_dismissed/abc123", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 2, | ||
| "Boolean", | ||
| -1, | ||
| ), | ||
| (Uuid::now_v7(), "company_type", 1, 1, false, 2, "String", -1), | ||
| (Uuid::now_v7(), "$os_version", 1, 1, false, 2, "String", -1), | ||
| (Uuid::now_v7(), "created_at", 1, 1, false, 2, "DateTime", -1), | ||
| (Uuid::now_v7(), "hire_date", 1, 1, false, 2, "DateTime", -1), | ||
| // query param "latest" filter will apply to these (eliminates "initial" props) | ||
| ( | ||
| Uuid::now_v7(), | ||
| "$initial_geoip_postal_code", | ||
| 1, | ||
| 1, | ||
| true, | ||
| 2, | ||
| "Numeric", | ||
| -1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "$initial_geoip_longitude", | ||
| 1, | ||
| 1, | ||
| true, | ||
| 2, | ||
| "Numeric", | ||
| -1, | ||
| ), | ||
| // PropertyParentType::Group (1 for each valid property_type and group_type_index) | ||
| (Uuid::now_v7(), "instance_name", 1, 1, false, 3, "String", 0), | ||
| (Uuid::now_v7(), "project_name", 1, 1, false, 3, "String", 1), | ||
| (Uuid::now_v7(), "timezone", 1, 1, false, 3, "String", 2), | ||
| (Uuid::now_v7(), "city", 1, 1, false, 3, "String", 3), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "integration_id", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "String", | ||
| 4, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "total_registrations", | ||
| 1, | ||
| 1, | ||
| true, | ||
| 3, | ||
| "Numeric", | ||
| 0, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "web_events_count_in_period", | ||
| 1, | ||
| 1, | ||
| true, | ||
| 3, | ||
| "Numeric", | ||
| 1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "group_types_total", | ||
| 1, | ||
| 1, | ||
| true, | ||
| 3, | ||
| "Numeric", | ||
| 2, | ||
| ), | ||
| (Uuid::now_v7(), "min_age", 1, 1, true, 3, "Numeric", 3), | ||
| (Uuid::now_v7(), "PlanValue", 1, 1, true, 3, "Numeric", 4), | ||
| (Uuid::now_v7(), "signup_date", 1, 1, false, 3, "DateTime", 0), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "last_recorded_date", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "DateTime", | ||
| 1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "group_createdAt", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "DateTime", | ||
| 2, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "subscription_end", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "DateTime", | ||
| 3, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "subscription_next_refresh", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "DateTime", | ||
| 4, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "ingested_event", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "Boolean", | ||
| 0, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "isIssueRiskSet", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "Boolean", | ||
| 1, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "features.supplier360.products.1.enabled", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "Boolean", | ||
| 2, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "is_project_demo", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "Boolean", | ||
| 3, | ||
| ), | ||
| ( | ||
| Uuid::now_v7(), | ||
| "subscription_is_trial", | ||
| 1, | ||
| 1, | ||
| false, | ||
| 3, | ||
| "Boolean", | ||
| 4, | ||
| ), | ||
| // NOTE: some event flavors are not represented in the test seeds (yet!) | ||
| // - no records of type PropertyParentType::Session in the prod DB | ||
| // - no records in the prod DB of property_type "Duration" | ||
| ]; | ||
|
|
||
| for row in pd_rows { | ||
| let mut args = PgArguments::default(); | ||
| args.add(row.0).unwrap(); | ||
| args.add(row.1).unwrap(); | ||
| args.add(row.2).unwrap(); | ||
| args.add(row.3).unwrap(); | ||
| args.add(row.4).unwrap(); | ||
| args.add(row.5).unwrap(); | ||
| args.add(row.6).unwrap(); | ||
| args.add(row.7).unwrap(); | ||
|
|
||
| sqlx::query_with( | ||
| r#" | ||
| INSERT INTO posthog_propertydefinition | ||
| (id, name, project_id, team_id, is_numerical, type, property_type, group_type_index) | ||
| VALUES ($1, $2, $3, $4, $5, $6, $7, $8) | ||
| "#, | ||
| args, | ||
| ) | ||
| .execute(&test_pool) | ||
| .await?; | ||
| } | ||
|
|
||
| // TODO: seed the ee_enterprisepropertydefinition, posthog_eventproperty and posthog_user tables too! | ||
|
|
||
| Ok(()) | ||
| } |