|
| 1 | +package cli |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "database/sql" |
| 6 | + "os" |
| 7 | + "path/filepath" |
| 8 | + "testing" |
| 9 | +) |
| 10 | + |
| 11 | +// dnfPrimarySchema mirrors the columns extractPackages reads from repodata primary.sqlite. |
| 12 | +const dnfPrimarySchema = ` |
| 13 | +CREATE TABLE packages ( |
| 14 | + pkgKey INTEGER PRIMARY KEY, |
| 15 | + pkgId TEXT, |
| 16 | + name TEXT, |
| 17 | + arch TEXT, |
| 18 | + version TEXT, |
| 19 | + epoch TEXT, |
| 20 | + release TEXT, |
| 21 | + summary TEXT, |
| 22 | + description TEXT, |
| 23 | + url TEXT, |
| 24 | + time_file INTEGER, |
| 25 | + time_build INTEGER, |
| 26 | + rpm_license TEXT, |
| 27 | + rpm_vendor TEXT, |
| 28 | + rpm_group TEXT, |
| 29 | + rpm_buildhost TEXT, |
| 30 | + rpm_sourcerpm TEXT, |
| 31 | + rpm_header_start INTEGER, |
| 32 | + rpm_header_end INTEGER, |
| 33 | + rpm_packager TEXT, |
| 34 | + size_package INTEGER, |
| 35 | + size_installed INTEGER, |
| 36 | + size_archive INTEGER, |
| 37 | + location_href TEXT, |
| 38 | + location_base TEXT, |
| 39 | + checksum_type TEXT |
| 40 | +)` |
| 41 | + |
| 42 | +// makeDNFPrimary builds a primary.sqlite file and returns its raw bytes. |
| 43 | +func makeDNFPrimary(t *testing.T, inserts ...string) []byte { |
| 44 | + t.Helper() |
| 45 | + path := filepath.Join(t.TempDir(), "primary.sqlite") |
| 46 | + db, err := sql.Open("sqlite", path) |
| 47 | + if err != nil { |
| 48 | + t.Fatalf("open sqlite: %v", err) |
| 49 | + } |
| 50 | + if _, err := db.Exec(dnfPrimarySchema); err != nil { |
| 51 | + t.Fatalf("create schema: %v", err) |
| 52 | + } |
| 53 | + for _, stmt := range inserts { |
| 54 | + if _, err := db.Exec(stmt); err != nil { |
| 55 | + t.Fatalf("insert: %v", err) |
| 56 | + } |
| 57 | + } |
| 58 | + if err := db.Close(); err != nil { |
| 59 | + t.Fatalf("close sqlite: %v", err) |
| 60 | + } |
| 61 | + data, err := os.ReadFile(path) |
| 62 | + if err != nil { |
| 63 | + t.Fatalf("read sqlite file: %v", err) |
| 64 | + } |
| 65 | + return data |
| 66 | +} |
| 67 | + |
| 68 | +func TestDNFParserExtractPackages(t *testing.T) { |
| 69 | + data := makeDNFPrimary(t, |
| 70 | + `INSERT INTO packages (pkgKey, pkgId, name, arch, version, epoch, release, |
| 71 | + summary, url, time_build, rpm_license, rpm_sourcerpm, |
| 72 | + size_package, location_href, checksum_type) |
| 73 | + VALUES (1, 'abc123', 'pg_duckdb_18', 'x86_64', '1.1.2', '0', '1PGDG.rhel9', |
| 74 | + 'DuckDB embedded in Postgres', 'https://github.com/duckdb/pg_duckdb', 1750000000, |
| 75 | + 'MIT', 'pg_duckdb_18-1.1.2-1PGDG.rhel9.src.rpm', |
| 76 | + 12345678, 'pg_duckdb_18-1.1.2-1PGDG.rhel9.x86_64.rpm', 'sha256')`, |
| 77 | + `INSERT INTO packages (pkgKey, pkgId, name) VALUES (2, 'def456', 'sparse-pkg')`, |
| 78 | + ) |
| 79 | + |
| 80 | + tmp := filepath.Join(t.TempDir(), "extract.sqlite") |
| 81 | + if err := os.WriteFile(tmp, data, 0644); err != nil { |
| 82 | + t.Fatalf("write temp sqlite: %v", err) |
| 83 | + } |
| 84 | + db, err := sql.Open("sqlite", "file:"+tmp+"?mode=ro") |
| 85 | + if err != nil { |
| 86 | + t.Fatalf("open sqlite ro: %v", err) |
| 87 | + } |
| 88 | + defer db.Close() |
| 89 | + |
| 90 | + parser := newDNFParser(context.Background(), liveTable("dnf"), false) |
| 91 | + packages, err := parser.extractPackages(db) |
| 92 | + if err != nil { |
| 93 | + t.Fatalf("extractPackages: %v", err) |
| 94 | + } |
| 95 | + if len(packages) != 2 { |
| 96 | + t.Fatalf("got %d packages, want 2", len(packages)) |
| 97 | + } |
| 98 | + |
| 99 | + full := packages[0] |
| 100 | + if full.Name != "pg_duckdb_18" || full.PkgKey != 1 || full.PkgId != "abc123" { |
| 101 | + t.Fatalf("unexpected identity: %+v", full) |
| 102 | + } |
| 103 | + if !full.Version.Valid || full.Version.String != "1.1.2" { |
| 104 | + t.Fatalf("version = %+v, want 1.1.2", full.Version) |
| 105 | + } |
| 106 | + if !full.TimeBuild.Valid || full.TimeBuild.Int64 != 1750000000 { |
| 107 | + t.Fatalf("time_build = %+v, want 1750000000", full.TimeBuild) |
| 108 | + } |
| 109 | + if !full.SizePackage.Valid || full.SizePackage.Int64 != 12345678 { |
| 110 | + t.Fatalf("size_package = %+v, want 12345678", full.SizePackage) |
| 111 | + } |
| 112 | + |
| 113 | + sparse := packages[1] |
| 114 | + if sparse.Name != "sparse-pkg" { |
| 115 | + t.Fatalf("sparse name = %q", sparse.Name) |
| 116 | + } |
| 117 | + if sparse.Arch.Valid || sparse.Summary.Valid || sparse.TimeBuild.Valid || sparse.SizePackage.Valid { |
| 118 | + t.Fatalf("sparse row should scan as NULLs: %+v", sparse) |
| 119 | + } |
| 120 | + |
| 121 | + // mode=ro must reject writes |
| 122 | + if _, err := db.Exec(`INSERT INTO packages (pkgKey, pkgId, name) VALUES (3, 'x', 'y')`); err == nil { |
| 123 | + t.Fatal("write on mode=ro connection should fail") |
| 124 | + } |
| 125 | +} |
| 126 | + |
| 127 | +func TestDNFParserParseRepositoryEmpty(t *testing.T) { |
| 128 | + // An empty packages table returns before any PostgreSQL access, |
| 129 | + // exercising the temp-file write and read-only DSN open path. |
| 130 | + data := makeDNFPrimary(t) |
| 131 | + parser := newDNFParser(context.Background(), liveTable("dnf"), false) |
| 132 | + count, err := parser.ParseRepository("test-repo", data) |
| 133 | + if err != nil { |
| 134 | + t.Fatalf("ParseRepository: %v", err) |
| 135 | + } |
| 136 | + if count != 0 { |
| 137 | + t.Fatalf("count = %d, want 0", count) |
| 138 | + } |
| 139 | +} |
0 commit comments