80 lines
2.9 KiB
C
80 lines
2.9 KiB
C
#define _POSIX_C_SOURCE 200809L
|
|
|
|
#include <sqlite3.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
#include "trainlog/database.h"
|
|
|
|
#define CHECK(value) \
|
|
do { \
|
|
if (!(value)) { \
|
|
fprintf(stderr, "CHECK failed line %d: %s\n", __LINE__, #value); \
|
|
goto cleanup; \
|
|
} \
|
|
} while (0)
|
|
|
|
static int scalar(sqlite3 *database, const char *sql) {
|
|
sqlite3_stmt *statement = NULL;
|
|
int value = -1;
|
|
if (sqlite3_prepare_v2(database, sql, -1, &statement, NULL) == SQLITE_OK &&
|
|
sqlite3_step(statement) == SQLITE_ROW) {
|
|
value = sqlite3_column_int(statement, 0);
|
|
}
|
|
(void)sqlite3_finalize(statement);
|
|
return value;
|
|
}
|
|
|
|
int main(void) {
|
|
char path[] = "/tmp/trainlog-schema-v31-XXXXXX";
|
|
TrainlogDatabase *production = NULL;
|
|
sqlite3 *raw = NULL;
|
|
int descriptor = mkstemp(path);
|
|
int version = 0;
|
|
int result = EXIT_FAILURE;
|
|
|
|
CHECK(descriptor >= 0 && close(descriptor) == 0);
|
|
CHECK(trainlog_database_open(path, &production) == TRAINLOG_STATUS_OK);
|
|
trainlog_database_close(production);
|
|
production = NULL;
|
|
|
|
CHECK(sqlite3_open(path, &raw) == SQLITE_OK);
|
|
CHECK(sqlite3_exec(raw,
|
|
"DROP TABLE session_exercise_timeline;"
|
|
"PRAGMA user_version=30;",
|
|
NULL,
|
|
NULL,
|
|
NULL) == SQLITE_OK);
|
|
CHECK(sqlite3_close(raw) == SQLITE_OK);
|
|
raw = NULL;
|
|
|
|
CHECK(trainlog_database_open(path, &production) == TRAINLOG_STATUS_OK);
|
|
CHECK(trainlog_database_schema_version(production, &version) == TRAINLOG_STATUS_OK);
|
|
CHECK(version == TRAINLOG_DATABASE_SCHEMA_VERSION);
|
|
trainlog_database_close(production);
|
|
production = NULL;
|
|
|
|
CHECK(sqlite3_open(path, &raw) == SQLITE_OK);
|
|
CHECK(scalar(raw,
|
|
"SELECT COUNT(*) FROM sqlite_master WHERE type='table' "
|
|
"AND name='session_exercise_timeline'") == 1);
|
|
CHECK(scalar(raw,
|
|
"SELECT COUNT(*) FROM sqlite_master WHERE type='index' "
|
|
"AND name='session_exercise_timeline_time'") == 1);
|
|
CHECK(scalar(raw, "SELECT COUNT(*) FROM pragma_foreign_key_check") == 0);
|
|
result = EXIT_SUCCESS;
|
|
|
|
cleanup:
|
|
if (production != NULL) {
|
|
trainlog_database_close(production);
|
|
}
|
|
if (raw != NULL) {
|
|
(void)sqlite3_close(raw);
|
|
}
|
|
(void)unlink(path);
|
|
if (result == EXIT_SUCCESS) {
|
|
puts("PASS schema v30 to v31 session timeline migration");
|
|
}
|
|
return result;
|
|
}
|