Skip to content
This repository has been archived by the owner on Feb 19, 2022. It is now read-only.

Add test to confirm that we can still insert after a restore #17

Merged
merged 1 commit into from
Oct 21, 2020
Merged

Add test to confirm that we can still insert after a restore #17

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 26 additions & 0 deletions pkg/test/restore_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -161,6 +161,8 @@ func TestBackupRestore(t *testing.T) {
t.Fatal("Failed on restore: ", err)
}
confirmTablesCongruent(t, pgx.Identifier{"public"}, pgx.Identifier{"two_Partitions"}, dumpConfig.DbURI, restoreConfig.DbURI)
confirmTablesCongruent(t, pgx.Identifier{"public"}, pgx.Identifier{"insert_test"}, dumpConfig.DbURI, restoreConfig.DbURI)
confirmCanStillInsert(t, restoreConfig.DbURI)
})
}
}
Expand Down Expand Up @@ -207,6 +209,18 @@ func confirmTablesCongruent(t *testing.T, tableSchema pgx.Identifier, tableName
}
}

func confirmCanStillInsert(t *testing.T, restoredURI string) {
restoredConn, err := util.GetDBConn(context.Background(), restoredURI)
if err != nil {
t.Fatal("Unable to connect to restore db: ", err)
}
defer restoredConn.Close(context.Background())
_, err = restoredConn.Exec(context.Background(), `INSERT INTO public."insert_test"(tstamp, device_id, series_0, series_1) VALUES ('2020-10-04 14:21:08+00', 'dev2', 1.5, 1)`)
if err != nil {
t.Fatalf("Post-restore insert failure")
}
}

func setupOrigDB(t *testing.T, db dbInfo, tsSchema string, tsVersion string) {

createTestDB(t, db)
Expand Down Expand Up @@ -255,6 +269,18 @@ func setupOrigDB(t *testing.T, db dbInfo, tsSchema string, tsVersion string) {
mustExec(t, conn, `ALTER TABLE public."two_Partitions" SET (timescaledb.compress=true)`)
mustExec(t, conn, `SELECT compress_chunk((SELECT chunk FROM show_chunks('public."two_Partitions"'::regclass) c (chunk) LIMIT 1))`)

mustExec(t, conn, `CREATE TABLE PUBLIC."insert_test" (
tstamp timestamptz NOT NULL,
device_id TEXT NOT NULL,
series_0 DOUBLE PRECISION NULL,
series_1 DOUBLE PRECISION NULL) ;`)

mustExec(t, conn, `SELECT * FROM create_hypertable('"public"."insert_test"'::regclass, 'tstamp'::name, chunk_time_interval=>'1 day'::interval);`)
mustExec(t, conn, `INSERT INTO public."insert_test"(tstamp, device_id, series_0, series_1) VALUES
('2020-10-04 14:21:08+00', 'dev1', 1.5, 1),
('2020-10-04 14:21:30+00', 'dev1', 1.5, 2),
('2020-10-10 14:21:30+00', 'dev2', 1.5, 1),
('2020-10-14 14:21:30+00', 'dev1', 2.5, 3)`)
}

func PGConnectURI(db dbInfo, useDefault bool) string {
Expand Down