import test from "ava";
import { expect } from "chai";
import { Readable } from "stream";
import { accumulator, accumulatorBy } from "../src";
import { FlushStrategy } from "../src/functions/baseDefinitions";

test.cb("accumulator() rolling", t => {
    t.plan(3);
    let chunkIndex = 0;
    interface TestObject {
        ts: number;
        key: string;
    }
    const source = new Readable({ objectMode: true });
    const firstFlush = [{ ts: 0, key: "a" }, { ts: 1, key: "b" }];
    const secondFlush = [{ ts: 2, key: "d" }, { ts: 3, key: "e" }];
    const thirdFlush = [{ ts: 4, key: "f" }];
    const flushes = [firstFlush, secondFlush, thirdFlush];

    source
        .pipe(accumulator(2, undefined, FlushStrategy.rolling))
        .on("data", (flush: TestObject[]) => {
            t.deepEqual(flush, flushes[chunkIndex]);
            chunkIndex++;
        })
        .on("error", (e: any) => {
            t.end(e);
        })
        .on("end", () => {
            t.end();
        });
    [...firstFlush, ...secondFlush, ...thirdFlush].forEach(item => {
        source.push(item);
    });
    source.push(null);
});

test.cb("accumulator() rolling with key", t => {
    t.plan(2);
    let chunkIndex = 0;
    interface TestObject {
        ts: number;
        key: string;
    }
    const source = new Readable({ objectMode: true });
    const firstFlush = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 2, key: "d" },
    ];
    const secondFlush = [{ ts: 3, key: "e" }];
    const flushes = [firstFlush, secondFlush];

    source
        .pipe(accumulator(3, undefined, FlushStrategy.rolling, "ts"))
        .on("data", (flush: TestObject[]) => {
            t.deepEqual(flush, flushes[chunkIndex]);
            chunkIndex++;
        })
        .on("error", (e: any) => {
            t.end(e);
        })
        .on("end", () => {
            t.end();
        });
    [...firstFlush, ...secondFlush].forEach(item => {
        source.push(item);
    });
    source.push(null);
});

test.cb(
    "accumulator() rolling should emit error and ignore chunk when its missing key",
    t => {
        t.plan(2);
        let index = 0;
        interface TestObject {
            ts: number;
            key: string;
        }
        const source = new Readable({ objectMode: true });
        const accumulatorStream = accumulator(
            3,
            undefined,
            FlushStrategy.rolling,
            "nonExistingKey",
        );
        const input = [{ ts: 0, key: "a" }, { ts: 1, key: "b" }];

        source
            .pipe(accumulatorStream)
            .on("data", (flush: TestObject[]) => {
                // No valid data output
                expect(flush).to.deep.equal([]);
            })
            .on("error", (err: any) => {
                source.pipe(accumulatorStream);
                accumulatorStream.resume();
                expect(err.message).to.equal(
                    `Key is missing in event: (nonExistingKey, ${JSON.stringify(
                        input[index],
                    )})`,
                );
                index++;
                t.pass();
            })
            .on("end", () => {
                t.end();
            });
        input.forEach(item => {
            source.push(item);
        });
        source.push(null);
    },
);

test.cb(
    "accumulator() rolling should emit error, ignore chunk when key is missing and continue processing chunks correctly",
    t => {
        t.plan(3);
        let chunkIndex = 0;
        interface TestObject {
            ts: number;
            key: string;
        }
        const source = new Readable({ objectMode: true });
        const accumulatorStream = accumulator(
            3,
            undefined,
            FlushStrategy.rolling,
            "ts",
        );
        const input = [
            { ts: 0, key: "a" },
            { ts: 1, key: "b" },
            { ts: 2, key: "c" },
            { key: "d" },
            { ts: 3, key: "e" },
        ];
        const firstFlush = [
            { ts: 0, key: "a" },
            { ts: 1, key: "b" },
            { ts: 2, key: "c" },
        ];
        const secondFlush = [{ ts: 3, key: "e" }];
        const flushes = [firstFlush, secondFlush];

        source
            .pipe(accumulatorStream)
            .on("data", (flush: TestObject[]) => {
                t.deepEqual(flush, flushes[chunkIndex]);
                chunkIndex++;
            })
            .on("error", (err: any) => {
                source.pipe(accumulatorStream);
                accumulatorStream.resume();
                expect(err.message).to.equal(
                    `Key is missing in event: (ts, ${JSON.stringify(
                        input[3],
                    )})`,
                );
                t.pass();
            })
            .on("end", () => {
                t.end();
            });
        input.forEach(item => {
            source.push(item);
        });
        source.push(null);
    },
);

test.cb("accumulator() sliding", t => {
    t.plan(4);
    let chunkIndex = 0;
    interface TestObject {
        ts: number;
        key: string;
    }
    const source = new Readable({ objectMode: true });
    const input = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 4, key: "d" },
    ];
    const firstFlush = [{ ts: 0, key: "a" }];
    const secondFlush = [{ ts: 0, key: "a" }, { ts: 1, key: "b" }];
    const thirdFlush = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
    ];
    const fourthFlush = [
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 4, key: "d" },
    ];

    const flushes = [firstFlush, secondFlush, thirdFlush, fourthFlush];
    source
        .pipe(accumulator(3, undefined, FlushStrategy.sliding))
        .on("data", (flush: TestObject[]) => {
            t.deepEqual(flush, flushes[chunkIndex]);
            chunkIndex++;
        })
        .on("error", (e: any) => {
            t.end(e);
        })
        .on("end", () => {
            t.end();
        });
    input.forEach(item => {
        source.push(item);
    });
    source.push(null);
});

test.cb("accumulator() sliding with key", t => {
    t.plan(6);
    let chunkIndex = 0;
    interface TestObject {
        ts: number;
        key: string;
    }
    const source = new Readable({ objectMode: true });
    const input = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 3, key: "d" },
        { ts: 5, key: "f" },
        { ts: 6, key: "g" },
    ];
    const firstFlush = [{ ts: 0, key: "a" }];
    const secondFlush = [{ ts: 0, key: "a" }, { ts: 1, key: "b" }];
    const thirdFlush = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
    ];
    const fourthFlush = [
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 3, key: "d" },
    ];
    const fifthFlush = [{ ts: 3, key: "d" }, { ts: 5, key: "f" }];
    const sixthFlush = [{ ts: 5, key: "f" }, { ts: 6, key: "g" }];

    const flushes = [
        firstFlush,
        secondFlush,
        thirdFlush,
        fourthFlush,
        fifthFlush,
        sixthFlush,
    ];
    source
        .pipe(accumulator(3, undefined, FlushStrategy.sliding, "ts"))
        .on("data", (flush: TestObject[]) => {
            t.deepEqual(flush, flushes[chunkIndex]);
            chunkIndex++;
        })
        .on("error", (e: any) => {
            t.end(e);
        })
        .on("end", () => {
            t.end();
        });
    input.forEach(item => {
        source.push(item);
    });
    source.push(null);
});

test.cb(
    "accumulator() sliding should emit error and ignore chunk when key is missing",
    t => {
        t.plan(2);
        let index = 0;
        interface TestObject {
            ts: number;
            key: string;
        }
        const source = new Readable({ objectMode: true });
        const accumulatorStream = accumulator(
            3,
            undefined,
            FlushStrategy.sliding,
            "nonExistingKey",
        );
        const input = [{ ts: 0, key: "a" }, { ts: 1, key: "b" }];

        source
            .pipe(accumulatorStream)
            .on("data", (flush: TestObject[]) => {
                expect(flush).to.deep.equal([]);
            })
            .on("error", (err: any) => {
                source.pipe(accumulatorStream);
                accumulatorStream.resume();
                expect(err.message).to.equal(
                    `Key is missing in event: (nonExistingKey, ${JSON.stringify(
                        input[index],
                    )})`,
                );
                index++;
                t.pass();
            })
            .on("end", () => {
                t.end();
            });
        input.forEach(item => {
            source.push(item);
        });
        source.push(null);
    },
);

test.cb(
    "accumulator() sliding should emit error, ignore chunk when key is missing and continue processing chunks correctly",
    t => {
        t.plan(6);
        let chunkIndex = 0;
        interface TestObject {
            ts: number;
            key: string;
        }
        const source = new Readable({ objectMode: true });
        const accumulatorStream = accumulator(
            3,
            undefined,
            FlushStrategy.sliding,
            "ts",
        );
        const input = [
            { ts: 0, key: "a" },
            { key: "b" },
            { ts: 2, key: "c" },
            { ts: 3, key: "d" },
            { ts: 5, key: "f" },
            { ts: 6, key: "g" },
        ];
        const firstFlush = [{ ts: 0, key: "a" }];
        const secondFlush = [{ ts: 0, key: "a" }, { ts: 2, key: "c" }];
        const thirdFlush = [{ ts: 2, key: "c" }, { ts: 3, key: "d" }];
        const fourthFlush = [{ ts: 3, key: "d" }, { ts: 5, key: "f" }];
        const fifthFlush = [{ ts: 5, key: "f" }, { ts: 6, key: "g" }];

        const flushes = [
            firstFlush,
            secondFlush,
            thirdFlush,
            fourthFlush,
            fifthFlush,
        ];
        source
            .pipe(accumulatorStream)
            .on("data", (flush: TestObject[]) => {
                t.deepEqual(flush, flushes[chunkIndex]);
                chunkIndex++;
            })
            .on("error", (err: any) => {
                source.pipe(accumulatorStream);
                accumulatorStream.resume();
                expect(err.message).to.equal(
                    `Key is missing in event: (ts, ${JSON.stringify(
                        input[1],
                    )})`,
                );
                t.pass();
            })
            .on("end", () => {
                t.end();
            });
        input.forEach(item => {
            source.push(item);
        });
        source.push(null);
    },
);

test.cb("accumulatorBy() rolling", t => {
    t.plan(2);
    let chunkIndex = 0;
    interface TestObject {
        ts: number;
        key: string;
    }
    const source = new Readable({ objectMode: true });
    const firstFlush = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 2, key: "d" },
    ];
    const secondFlush = [{ ts: 3, key: "e" }];
    const flushes = [firstFlush, secondFlush];

    source
        .pipe(
            accumulatorBy(
                undefined,
                FlushStrategy.rolling,
                (event: TestObject, bufferChunk: TestObject) => {
                    return bufferChunk.ts + 3 <= event.ts;
                },
            ),
        )
        .on("data", (flush: TestObject[]) => {
            t.deepEqual(flush, flushes[chunkIndex]);
            chunkIndex++;
        })
        .on("error", (e: any) => {
            t.end(e);
        })
        .on("end", () => {
            t.end();
        });
    [...firstFlush, ...secondFlush].forEach(item => {
        source.push(item);
    });
    source.push(null);
});

test.cb(
    "accumulatorBy() rolling should emit error when key iteratee throws",
    t => {
        t.plan(2);
        interface TestObject {
            ts: number;
            key: string;
        }
        const source = new Readable({ objectMode: true });
        const input = [
            { ts: 0, key: "a" },
            { ts: 1, key: "b" },
            { ts: 2, key: "c" },
        ];
        const accumulaterStream = accumulatorBy(
            undefined,
            FlushStrategy.rolling,
            (event: TestObject, bufferChunk: TestObject) => {
                if (event.key !== "a") {
                    throw new Error("Failed mapping");
                }
                return bufferChunk.ts + 3 <= event.ts;
            },
        );
        source
            .pipe(accumulaterStream)
            .on("error", (err: any) => {
                source.pipe(accumulaterStream);
                accumulaterStream.resume();
                expect(err.message).to.equal("Failed mapping");
                t.pass();
            })
            .on("end", () => {
                t.end();
            });

        input.forEach(item => {
            source.push(item);
        });
        source.push(null);
    },
);

test.cb("accumulatorBy() sliding", t => {
    t.plan(6);
    let chunkIndex = 0;
    interface TestObject {
        ts: number;
        key: string;
    }
    const source = new Readable({ objectMode: true });
    const input = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 3, key: "d" },
        { ts: 5, key: "f" },
        { ts: 6, key: "g" },
    ];
    const firstFlush = [{ ts: 0, key: "a" }];
    const secondFlush = [{ ts: 0, key: "a" }, { ts: 1, key: "b" }];
    const thirdFlush = [
        { ts: 0, key: "a" },
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
    ];
    const fourthFlush = [
        { ts: 1, key: "b" },
        { ts: 2, key: "c" },
        { ts: 3, key: "d" },
    ];
    const fifthFlush = [{ ts: 3, key: "d" }, { ts: 5, key: "f" }];
    const sixthFlush = [{ ts: 5, key: "f" }, { ts: 6, key: "g" }];

    const flushes = [
        firstFlush,
        secondFlush,
        thirdFlush,
        fourthFlush,
        fifthFlush,
        sixthFlush,
    ];
    source
        .pipe(
            accumulatorBy(
                undefined,
                FlushStrategy.sliding,
                (event: TestObject, bufferChunk: TestObject) => {
                    return bufferChunk.ts + 3 <= event.ts ? true : false;
                },
            ),
        )
        .on("data", (flush: TestObject[]) => {
            t.deepEqual(flush, flushes[chunkIndex]);
            chunkIndex++;
        })
        .on("error", (e: any) => {
            t.end(e);
        })
        .on("end", () => {
            t.end();
        });
    input.forEach(item => {
        source.push(item);
    });
    source.push(null);
});

test.cb(
    "accumulatorBy() sliding should emit error when key iteratee throws",
    t => {
        t.plan(2);
        interface TestObject {
            ts: number;
            key: string;
        }
        const source = new Readable({ objectMode: true });
        const input = [
            { ts: 0, key: "a" },
            { ts: 1, key: "b" },
            { ts: 2, key: "c" },
        ];
        const accumulaterStream = accumulatorBy(
            undefined,
            FlushStrategy.sliding,
            (event: TestObject, bufferChunk: TestObject) => {
                if (event.key !== "a") {
                    throw new Error("Failed mapping");
                }
                return bufferChunk.ts + 3 <= event.ts ? true : false;
            },
        );
        source
            .pipe(accumulaterStream)
            .on("error", (err: any) => {
                source.pipe(accumulaterStream);
                accumulaterStream.resume();
                expect(err.message).to.equal("Failed mapping");
                t.pass();
            })
            .on("end", () => {
                t.end();
            });

        input.forEach(item => {
            source.push(item);
        });
        source.push(null);
    },
);