130 lines
4.6 KiB
TypeScript
130 lines
4.6 KiB
TypeScript
import type { ClientState } from "../test-definition";
|
|
|
|
export class AssertableState {
|
|
public readonly files: Map<string, string>;
|
|
public readonly clientFiles: Map<string, string>[];
|
|
|
|
public constructor(state: ClientState) {
|
|
this.files = state.files;
|
|
this.clientFiles = state.clientFiles;
|
|
}
|
|
|
|
public assertFileCount(expected: number): this {
|
|
if (this.files.size !== expected) {
|
|
const keys = Array.from(this.files.keys()).join(", ");
|
|
throw new Error(
|
|
`Expected ${expected} file(s), got ${this.files.size}: [${keys}]`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertFileExists(path: string): this {
|
|
if (!this.files.has(path)) {
|
|
const keys = Array.from(this.files.keys()).join(", ");
|
|
throw new Error(`Expected "${path}" to exist. Files: [${keys}]`);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertFileNotExists(path: string): this {
|
|
if (this.files.has(path)) {
|
|
const keys = Array.from(this.files.keys()).join(", ");
|
|
throw new Error(
|
|
`Expected "${path}" not to exist. Files: [${keys}]`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertContent(path: string, expected: string): this {
|
|
this.assertFileExists(path);
|
|
const actual = this.files.get(path) ?? "";
|
|
if (actual !== expected) {
|
|
throw new Error(
|
|
`Expected "${path}" to have content "${expected}", got: "${actual}"`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertContains(path: string, ...substrings: string[]): this {
|
|
this.assertFileExists(path);
|
|
const content = this.files.get(path) ?? "";
|
|
const missing = substrings.filter((s) => !content.includes(s));
|
|
if (missing.length > 0) {
|
|
throw new Error(
|
|
`Expected "${path}" to contain ${missing.map((s) => `"${s}"`).join(", ")}. Content: "${content}"`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertContainsAny(path: string, ...substrings: string[]): this {
|
|
this.assertFileExists(path);
|
|
const content = this.files.get(path) ?? "";
|
|
const found = substrings.some((s) => content.includes(s));
|
|
if (!found) {
|
|
throw new Error(
|
|
`Expected "${path}" to contain at least one of ${substrings.map((s) => `"${s}"`).join(", ")}. Content: "${content}"`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertAnyFileContains(...substrings: string[]): this {
|
|
const allContent = Array.from(this.files.values()).join("\n");
|
|
const missing = substrings.filter((s) => !allContent.includes(s));
|
|
if (missing.length > 0) {
|
|
const dump = Array.from(this.files.entries())
|
|
.map(([k, v]) => ` ${k}: "${v}"`)
|
|
.join("\n");
|
|
throw new Error(
|
|
`Expected some file to contain ${missing.map((s) => `"${s}"`).join(", ")}.\nFiles:\n${dump}`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertNoFileContains(...substrings: string[]): this {
|
|
const offenders: { path: string; substring: string }[] = [];
|
|
for (const [path, content] of this.files) {
|
|
for (const s of substrings) {
|
|
if (content.includes(s)) {
|
|
offenders.push({ path, substring: s });
|
|
}
|
|
}
|
|
}
|
|
if (offenders.length > 0) {
|
|
const dump = Array.from(this.files.entries())
|
|
.map(([k, v]) => ` ${k}: "${v}"`)
|
|
.join("\n");
|
|
throw new Error(
|
|
`Expected no file to contain ${substrings.map((s) => `"${s}"`).join(", ")}, but found ${offenders.map((o) => `"${o.substring}" in "${o.path}"`).join(", ")}.\nFiles:\n${dump}`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public assertContentInAtMostOneFile(substring: string): this {
|
|
const matches = Array.from(this.files.entries()).filter(([, content]) =>
|
|
content.includes(substring)
|
|
);
|
|
if (matches.length > 1) {
|
|
const dump = Array.from(this.files.entries())
|
|
.map(([k, v]) => ` ${k}: "${v}"`)
|
|
.join("\n");
|
|
throw new Error(
|
|
`Expected "${substring}" in at most 1 file, found in ${matches.length}: [${matches.map(([p]) => p).join(", ")}].\nFiles:\n${dump}`
|
|
);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
public ifFileExists(path: string, fn: (state: this) => void): this {
|
|
if (this.files.has(path)) {
|
|
fn(this);
|
|
}
|
|
return this;
|
|
}
|
|
}
|