179 lines
7.8 KiB
C
179 lines
7.8 KiB
C
/* test_archive.c - 存档命名规则与读写往返（不需要网络） */
|
||
#include "archive.h"
|
||
#include "util.h"
|
||
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <unistd.h>
|
||
#include <sys/stat.h>
|
||
|
||
static int failures = 0;
|
||
static int checks = 0;
|
||
|
||
#define CHECK(cond, ...) \
|
||
do { \
|
||
checks++; \
|
||
if (!(cond)) { \
|
||
failures++; \
|
||
printf(" ✗ %s:%d: ", __FILE__, __LINE__); \
|
||
printf(__VA_ARGS__); \
|
||
printf("\n"); \
|
||
} \
|
||
} while (0)
|
||
|
||
static int is_alnum_lower(const char *s, size_t n)
|
||
{
|
||
for (size_t i = 0; i < n; i++)
|
||
if (!((s[i] >= 'a' && s[i] <= 'z') || (s[i] >= '0' && s[i] <= '9')))
|
||
return 0;
|
||
return 1;
|
||
}
|
||
|
||
int main(void)
|
||
{
|
||
char tmpl[] = "/tmp/ligent_arch_XXXXXX";
|
||
char *dir = mkdtemp(tmpl);
|
||
if (!dir) { perror("mkdtemp"); return 2; }
|
||
char err[256];
|
||
|
||
puts("根名：8 位小写字母数字");
|
||
char *n1 = NULL;
|
||
CHECK(archive_next_name(dir, NULL, NULL, 0, &n1, err, sizeof err) == 0, "生成失败: %s", err);
|
||
CHECK(n1 && strlen(n1) == ARCHIVE_ROOT_LEN, "根名应为 %d 位，实际 %s",
|
||
ARCHIVE_ROOT_LEN, n1 ? n1 : "(null)");
|
||
CHECK(n1 && is_alnum_lower(n1, strlen(n1)), "根名字符集应为 [a-z0-9]：%s", n1);
|
||
free(n1);
|
||
|
||
/* 空字符串与 NULL 等价：都表示"没有来源" */
|
||
char *n2 = NULL;
|
||
CHECK(archive_next_name(dir, "", NULL, 0, &n2, err, sizeof err) == 0, "空 lineage 应可用");
|
||
CHECK(n2 && strlen(n2) == ARCHIVE_ROOT_LEN, "空 lineage 也应给出根名");
|
||
free(n2);
|
||
|
||
puts("分支名：来源 + _ + 3 位");
|
||
char *b = NULL;
|
||
CHECK(archive_next_name(dir, "a1b2c3d4", NULL, 0, &b, err, sizeof err) == 0,
|
||
"分支生成失败: %s", err);
|
||
CHECK(b && strncmp(b, "a1b2c3d4_", 9) == 0, "应以来源加下划线开头：%s", b);
|
||
CHECK(b && strlen(b) == 8 + 1 + ARCHIVE_BRANCH_LEN, "分支长度不对：%s", b);
|
||
CHECK(b && is_alnum_lower(b + 9, ARCHIVE_BRANCH_LEN), "后缀字符集不对：%s", b);
|
||
free(b);
|
||
|
||
/* 多次生成应当不同（随机性） */
|
||
char *x1 = NULL, *x2 = NULL;
|
||
archive_next_name(dir, "a1b2c3d4", NULL, 0, &x1, err, sizeof err);
|
||
archive_next_name(dir, "a1b2c3d4", NULL, 0, &x2, err, sizeof err);
|
||
CHECK(x1 && x2 && strcmp(x1, x2) != 0, "两次生成不应相同：%s / %s", x1, x2);
|
||
free(x1); free(x2);
|
||
|
||
puts("长度上限：接不下就重新起根");
|
||
/* 边界：来源长度 + 1 + 3 + ".json" <= 240 */
|
||
size_t maxlineage = ARCHIVE_NAME_MAX - 1 - ARCHIVE_BRANCH_LEN - strlen(".json");
|
||
char *ok = xmalloc(maxlineage + 1);
|
||
memset(ok, 'a', maxlineage);
|
||
ok[maxlineage] = '\0';
|
||
|
||
char *edge = NULL;
|
||
CHECK(archive_next_name(dir, ok, NULL, 0, &edge, err, sizeof err) == 0, "边界生成失败");
|
||
CHECK(edge && strncmp(edge, "aaa", 3) == 0, "刚好装得下时应继续接分支");
|
||
CHECK(edge && strlen(edge) + strlen(".json") <= ARCHIVE_NAME_MAX,
|
||
"完整文件名不应超上限：%zu", edge ? strlen(edge) + 5 : 0);
|
||
free(edge);
|
||
|
||
/* 再多一位就装不下了 */
|
||
char *too = xmalloc(maxlineage + 2);
|
||
memset(too, 'a', maxlineage + 1);
|
||
too[maxlineage + 1] = '\0';
|
||
char *fresh = NULL;
|
||
CHECK(archive_next_name(dir, too, NULL, 0, &fresh, err, sizeof err) == 0, "超限生成失败");
|
||
CHECK(fresh && strlen(fresh) == ARCHIVE_ROOT_LEN,
|
||
"超限时应退回 %d 位根名，实际 %s（%zu 位）",
|
||
ARCHIVE_ROOT_LEN, fresh ? fresh : "(null)", fresh ? strlen(fresh) : 0);
|
||
free(fresh);
|
||
puts("前缀：临时存档");
|
||
char *tp = NULL;
|
||
CHECK(archive_next_name(dir, "a1b2c3d4", "tmp_", 0, &tp, err, sizeof err) == 0,
|
||
"带前缀生成失败");
|
||
CHECK(tp && strncmp(tp, "tmp_a1b2c3d4_", 13) == 0,
|
||
"前缀应在最前面、其后才是分支名：%s", tp ? tp : "(null)");
|
||
free(tp);
|
||
|
||
char *tr = NULL;
|
||
CHECK(archive_next_name(dir, NULL, "tmp_", 0, &tr, err, sizeof err) == 0, "带前缀建根失败");
|
||
CHECK(tr && strncmp(tr, "tmp_", 4) == 0 &&
|
||
strlen(tr) == 4 + ARCHIVE_ROOT_LEN,
|
||
"新根也应带前缀：%s", tr ? tr : "(null)");
|
||
free(tr);
|
||
|
||
/* 前缀要算进长度上限：原本刚好装得下的，加了 tmp_ 就该退回新根 */
|
||
char *edgep = NULL;
|
||
CHECK(archive_next_name(dir, ok, "tmp_", 0, &edgep, err, sizeof err) == 0, "边界失败");
|
||
CHECK(edgep && strlen(edgep) == 4 + ARCHIVE_ROOT_LEN,
|
||
"加了前缀装不下时应退回根名，实际 %s（%zu 位）",
|
||
edgep ? edgep : "(null)", edgep ? strlen(edgep) : 0);
|
||
free(edgep);
|
||
|
||
free(ok); free(too);
|
||
|
||
puts("导出与载入往返");
|
||
JsonValue *msgs = json_new_array();
|
||
JsonValue *m = json_new_object();
|
||
json_object_set_string(m, "role", "user");
|
||
json_object_set_string(m, "content", "你好，暗号是紫水晶");
|
||
json_array_push(msgs, m);
|
||
|
||
char *path = NULL, *name = NULL;
|
||
time_t t = 1700000000;
|
||
CHECK(archive_export(msgs, dir, "parentname", NULL, 0, t, t + 5,
|
||
"test-model", "http://example/v1",
|
||
&path, &name, err, sizeof err) == 0,
|
||
"导出失败: %s", err);
|
||
CHECK(name && strncmp(name, "parentname_", 11) == 0,
|
||
"导出名应挂在来源下：%s", name ? name : "(null)");
|
||
CHECK(path && access(path, R_OK) == 0, "文件应存在：%s", path ? path : "");
|
||
|
||
char *created = NULL, *saved = NULL, *model = NULL, *iname = NULL, *iparent = NULL;
|
||
JsonValue *back = archive_load(path, &created, &saved, &model, &iname, &iparent,
|
||
err, sizeof err);
|
||
CHECK(back != NULL, "载入失败: %s", err);
|
||
CHECK(back && json_array_size(back) == 1, "消息条数应为 1");
|
||
CHECK(iname && name && strcmp(iname, name) == 0, "内容里的 name 应与文件名一致");
|
||
CHECK(iparent && strcmp(iparent, "parentname") == 0,
|
||
"内容里的 parent 应记下来源：%s", iparent ? iparent : "(null)");
|
||
CHECK(model && strcmp(model, "test-model") == 0, "model 应往返");
|
||
CHECK(created && strstr(created, "2023-") != NULL,
|
||
"created 应可读：%s", created ? created : "(null)");
|
||
/* saved 是"这份存档写出时的时间"，与 created（这段对话的开始时间）是两回事 */
|
||
CHECK(saved && strstr(saved, "2023-") != NULL,
|
||
"saved 应可读：%s", saved ? saved : "(null)");
|
||
|
||
json_free(back); json_free(msgs);
|
||
free(created); free(saved); free(model); free(iname); free(iparent);
|
||
|
||
puts("列举");
|
||
ArchiveEntry *list = NULL;
|
||
size_t n = archive_list(dir, &list);
|
||
CHECK(n == 1, "应有 1 份存档，实际 %zu", n);
|
||
if (n == 1) {
|
||
CHECK(strncmp(list[0].name, "parentname_", 11) == 0, "名字应保留");
|
||
CHECK(list[0].count == 1, "条数应为 1，实际 %ld", list[0].count);
|
||
CHECK(list[0].created != NULL, "时间应从内容里读出来");
|
||
}
|
||
archive_list_free(list, n);
|
||
|
||
/* 非 ligent 存档（没有 format 标记）应被忽略 */
|
||
char *jd = xasprintf("%s/junk.json", dir);
|
||
FILE *j2 = fopen(jd, "w");
|
||
if (j2) { fputs("{\"hello\":1}", j2); fclose(j2); }
|
||
n = archive_list(dir, &list);
|
||
CHECK(n == 1, "非 ligent 存档应被忽略，实际 %zu", n);
|
||
archive_list_free(list, n);
|
||
unlink(jd); free(jd);
|
||
|
||
unlink(path); free(path); free(name);
|
||
rmdir(dir);
|
||
printf("\n%d 项检查，%d 项失败\n", checks, failures);
|
||
return failures ? 1 : 0;
|
||
}
|