199 lines
7.2 KiB
C
199 lines
7.2 KiB
C
/* test_attach.c - /include 的暂存、glob 展开与体积/类型判断
|
|
*
|
|
* 交互式挑选 (attach_pick)要真终端才能测, 由 e2e 第 55 段覆盖;
|
|
* 这里测的是不需要人操作的那一半.
|
|
*/
|
|
#include "attach.h"
|
|
#include "image.h"
|
|
#include "json.h"
|
|
#include "util.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.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 void write_file(const char *path, const void *data, size_t n) {
|
|
FILE *f = fopen(path, "wb");
|
|
if (!f) {
|
|
perror(path);
|
|
exit(2);
|
|
}
|
|
if (n && fwrite(data, 1, n, f) != n) {
|
|
perror("fwrite");
|
|
exit(2);
|
|
}
|
|
fclose(f);
|
|
}
|
|
|
|
static const unsigned char PNG[] = {0x89, 'P', 'N', 'G', 0x0D, 0x0A, 0x1A, 0x0A,
|
|
'p', 'a', 'y', 'l', 'o', 'a', 'd'};
|
|
|
|
/* 数一数暂存里某类块有几个 */
|
|
static size_t count_type(const JsonValue *parts, const char *type) {
|
|
if (!parts)
|
|
return 0;
|
|
size_t n = 0;
|
|
for (size_t i = 0; i < json_array_size((JsonValue *)parts); i++) {
|
|
const char *t =
|
|
json_get_string(json_array_get((JsonValue *)parts, i), "type");
|
|
if (t && strcmp(t, type) == 0)
|
|
n++;
|
|
}
|
|
return n;
|
|
}
|
|
|
|
/* 记一下回调被问了几次 —— 它只在遇到图片时该被问到.
|
|
* 答案和计数放在一起, 以免"计数版只实现了一种答案"这种测试自己的错. */
|
|
static int ask_count;
|
|
static int ask_answer;
|
|
static int counting_ask(void *ud) {
|
|
(void)ud;
|
|
ask_count++;
|
|
return ask_answer;
|
|
}
|
|
|
|
int main(void) {
|
|
char tmpl[] = "/tmp/ligent_attach_XXXXXX";
|
|
char *dir = mkdtemp(tmpl);
|
|
if (!dir) {
|
|
perror("mkdtemp");
|
|
return 2;
|
|
}
|
|
|
|
/* 备份目录也指到临时目录, 别污染仓库 */
|
|
image_set_bindir(dir);
|
|
|
|
char *f_c = xasprintf("%s/a.c", dir);
|
|
char *f_txt = xasprintf("%s/b.txt", dir);
|
|
char *f_bin = xasprintf("%s/bin.dat", dir);
|
|
char *f_png = xasprintf("%s/pic.png", dir);
|
|
char *f_big = xasprintf("%s/big.txt", dir);
|
|
|
|
write_file(f_c, "int main(void){return 0;}\n", 26);
|
|
write_file(f_txt, "hello\n", 6);
|
|
write_file(f_bin, "\x00\x01\x02binary", 9);
|
|
write_file(f_png, PNG, sizeof PNG);
|
|
{
|
|
/* 超过单个文件上限 (256 KB) */
|
|
size_t n = ATTACH_MAX_FILE + 1024;
|
|
char *big = xmalloc(n);
|
|
memset(big, 'x', n);
|
|
write_file(f_big, big, n);
|
|
free(big);
|
|
}
|
|
|
|
puts("glob 展开: 命中几个就暂存几个");
|
|
attach_clear();
|
|
size_t skipped = 0;
|
|
char err[256];
|
|
int added = attach_glob(f_c, NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == 1 && attach_count() == 1, "命中一个文件");
|
|
CHECK(skipped == 0, "没有跳过任何东西");
|
|
|
|
JsonValue *parts = attach_take_parts();
|
|
CHECK(parts && json_array_size(parts) == 1, "取到 1 个块");
|
|
JsonValue *p0 = parts ? json_array_get(parts, 0) : NULL;
|
|
CHECK(p0 && strcmp(json_get_string(p0, "type"), "text") == 0,
|
|
"文本文件应当是 text 块");
|
|
const char *txt = p0 ? json_get_string(p0, "text") : NULL;
|
|
CHECK(txt && strstr(txt, "int main") != NULL, "内容带上了");
|
|
CHECK(txt && strncmp(txt, "===== ", 6) == 0, "块开头标了来源路径");
|
|
CHECK(txt && strchr(txt, '\n') != NULL, "结尾有换行");
|
|
/* 只在本地记的来源, 绝不能进块里 (那会跟着发给服务端) */
|
|
CHECK(p0 && json_get_string(p0, "source") == NULL,
|
|
"块里不该有自造的 source 字段");
|
|
json_free(parts);
|
|
CHECK(attach_count() == 0, "取走之后暂存清空");
|
|
|
|
puts("取走即清空, clear 也是");
|
|
attach_glob(f_c, NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(attach_count() == 1, "再附带一次");
|
|
attach_clear();
|
|
CHECK(attach_count() == 0, "attach_clear 清空");
|
|
CHECK(attach_take_parts() == NULL, "清空之后取不到东西");
|
|
|
|
puts("二进制文件跳过");
|
|
attach_clear();
|
|
added = attach_glob(f_bin, NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == 0 && skipped == 1, "二进制被跳过 (added=%d skipped=%zu)",
|
|
added, skipped);
|
|
|
|
puts("超过单个文件上限的跳过");
|
|
attach_clear();
|
|
added = attach_glob(f_big, NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == 0 && skipped == 1, "太大的文本被跳过 (added=%d)", added);
|
|
|
|
puts("图片: 认内容不认扩展名, 并走备份引用");
|
|
attach_clear();
|
|
added = attach_glob(f_png, NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == 1, "png 被收下");
|
|
CHECK(attach_has_image(), "暂存里有图片");
|
|
parts = attach_take_parts();
|
|
CHECK(count_type(parts, "ligent_image") == 1,
|
|
"应当是一个 ligent_image 引用块");
|
|
json_free(parts);
|
|
|
|
puts("模型不收图时跳过图片 (回调只在遇到图片时被问)");
|
|
attach_clear();
|
|
ask_count = 0;
|
|
ask_answer = 1;
|
|
added = attach_glob(f_txt, counting_ask, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == 1, "纯文本照带");
|
|
CHECK(ask_count == 0, "纯文本不该去问'能不能收图' (问了一次就是白探测)");
|
|
attach_clear();
|
|
ask_count = 0;
|
|
ask_answer = 0;
|
|
added = attach_glob(f_png, counting_ask, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == 0 && skipped == 1, "回调说不行就跳过");
|
|
CHECK(ask_count == 1, "遇到图片时问了一次, 实际 %d", ask_count);
|
|
|
|
puts("匹配不到 / 参数非法");
|
|
attach_clear();
|
|
added = attach_glob("/tmp/绝对不存在的模式*.zzz", NULL, NULL, &skipped, err,
|
|
sizeof err);
|
|
CHECK(added == -1 && err[0], "匹配不到返回 -1 并给出原因: %s", err);
|
|
added = attach_glob(NULL, NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == -1, "NULL 模式返回 -1");
|
|
added = attach_glob("", NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == -1, "空模式返回 -1");
|
|
|
|
puts("多个匹配会累加");
|
|
attach_clear();
|
|
char *pat = xasprintf("%s/*.txt", dir);
|
|
added = attach_glob(pat, NULL, NULL, &skipped, err, sizeof err);
|
|
CHECK(added == 1, "只有 b.txt 能带 (big.txt 超限), 实际 %d", added);
|
|
CHECK(skipped == 1, "big.txt 被跳过");
|
|
free(pat);
|
|
attach_clear();
|
|
|
|
free(f_c);
|
|
free(f_txt);
|
|
free(f_bin);
|
|
free(f_png);
|
|
free(f_big);
|
|
image_set_bindir(".");
|
|
char *cmd = xasprintf("rm -rf %s", tmpl);
|
|
if (system(cmd) != 0) {
|
|
}
|
|
free(cmd);
|
|
|
|
printf("\n%d 项检查, %d 项失败\n", checks, failures);
|
|
return failures ? 1 : 0;
|
|
}
|