test_that("all expected views register on session open", { skip_if_no_corpus() con <- cog_open() on.exit(cog_close()) views <- DBI::dbGetQuery(con, "SELECT table_name FROM information_schema.tables WHERE table_schema = 'main' AND table_type = 'VIEW'" ) expected <- c( "long", "spending_long", "revenue_long", "canonical_fips_xwalk", "summary_categories", "spending_annotated", "revenue_annotated" ) expect_true(all(expected %in% views$table_name)) }) test_that("spending_long filters to E/F/G/K prefixes and excludes aggregates", { skip_if_no_corpus() con <- cog_open() on.exit(cog_close()) prefixes <- DBI::dbGetQuery(con, "SELECT DISTINCT LEFT(item_code, 1) AS pfx FROM spending_long" )$pfx expect_true(all(prefixes %in% c("E", "F", "G", "K"))) agg_count <- DBI::dbGetQuery(con, "SELECT count(*) AS n FROM spending_long WHERE is_aggregate" )$n expect_equal(agg_count, 0) }) test_that("revenue_long filters to T/A/U/B/C/D prefixes and excludes aggregates", { skip_if_no_corpus() con <- cog_open() on.exit(cog_close()) prefixes <- DBI::dbGetQuery(con, "SELECT DISTINCT LEFT(item_code, 1) AS pfx FROM revenue_long" )$pfx expect_true(all(prefixes %in% c("T", "A", "U", "B", "C", "D"))) agg_count <- DBI::dbGetQuery(con, "SELECT count(*) AS n FROM revenue_long WHERE is_aggregate" )$n expect_equal(agg_count, 0) }) test_that("spending_annotated carries category + xwalk columns", { skip_if_no_corpus() con <- cog_open() on.exit(cog_close()) row <- DBI::dbGetQuery(con, "SELECT * FROM spending_annotated LIMIT 1" ) for (nm in c("canonical_govid", "item_code", "amt", "xwalk_gov_name", "govs_type", "population_acs", "category", "spend_subtype")) { expect_true(nm %in% names(row), info = paste("missing column:", nm)) } }) test_that("gov_population_yearly exposes one row per (year, canonical_govid)", { skip_if_no_corpus() with_fixture_corpus({ con <- uscogdata:::.ensure_session() df <- DBI::dbGetQuery( con, "SELECT year, canonical_govid, population, popyear FROM gov_population_yearly WHERE canonical_govid = '101006006' ORDER BY year" ) expect_setequal(df$year, c(2019L, 2020L)) expect_equal(nrow(df), 2L) expect_true(all(!is.na(df$population))) expect_equal(df$population[df$year == 2019L], 1935878L) expect_equal(df$population[df$year == 2020L], 1952778L) # Uniqueness on (year, canonical_govid) across the whole view. dup <- DBI::dbGetQuery( con, "SELECT year, canonical_govid, COUNT(*) AS n FROM gov_population_yearly GROUP BY year, canonical_govid HAVING n > 1" ) expect_equal(nrow(dup), 0L) }) })