146 lines
4.1 KiB
JavaScript
146 lines
4.1 KiB
JavaScript
"use strict";
|
|
|
|
const fs = require("fs");
|
|
const path = require("path");
|
|
const crypto = require("crypto");
|
|
const expect = require("chai").expect;
|
|
const util = require("../util");
|
|
const Helper = require("../../src/helper");
|
|
const storage = require("../../src/plugins/storage");
|
|
const link = require("../../src/plugins/irc-events/link.js");
|
|
|
|
describe("Image storage", function () {
|
|
// Increase timeout due to unpredictable I/O on CI services
|
|
this.timeout(util.isRunningOnCI() ? 25000 : 5000);
|
|
this.slow(300);
|
|
|
|
const testImagePath = path.resolve(__dirname, "../../client/img/logo-grey-bg-120x120px.png");
|
|
const correctImageHash = crypto
|
|
.createHash("sha256")
|
|
.update(fs.readFileSync(testImagePath))
|
|
.digest("hex");
|
|
const correctImageURL = `storage/${correctImageHash.substring(
|
|
0,
|
|
2
|
|
)}/${correctImageHash.substring(2, 4)}/${correctImageHash.substring(4)}.png`;
|
|
|
|
const testSvgPath = path.resolve(__dirname, "../../client/img/logo-grey-bg.svg");
|
|
const correctSvgHash = crypto
|
|
.createHash("sha256")
|
|
.update(fs.readFileSync(testSvgPath))
|
|
.digest("hex");
|
|
const correctSvgURL = `storage/${correctSvgHash.substring(0, 2)}/${correctSvgHash.substring(
|
|
2,
|
|
4
|
|
)}/${correctSvgHash.substring(4)}.svg`;
|
|
|
|
before(function (done) {
|
|
this.app = util.createWebserver();
|
|
this.app.get("/real-test-image.png", function (req, res) {
|
|
res.sendFile(testImagePath);
|
|
});
|
|
this.app.get("/logo.svg", function (req, res) {
|
|
res.sendFile(testSvgPath);
|
|
});
|
|
this.connection = this.app.listen(0, () => {
|
|
this.port = this.connection.address().port;
|
|
done();
|
|
});
|
|
});
|
|
|
|
after(function (done) {
|
|
this.connection.close(done);
|
|
});
|
|
|
|
after(function (done) {
|
|
// After storage tests run, remove the remaining empty
|
|
// storage folder so we return to the clean state
|
|
const dir = Helper.getStoragePath();
|
|
fs.rmdir(dir, done);
|
|
});
|
|
|
|
beforeEach(function () {
|
|
this.irc = util.createClient();
|
|
this.network = util.createNetwork();
|
|
|
|
Helper.config.prefetchStorage = true;
|
|
});
|
|
|
|
afterEach(function () {
|
|
Helper.config.prefetchStorage = false;
|
|
});
|
|
|
|
it("should store the thumbnail", function (done) {
|
|
const port = this.port;
|
|
const message = this.irc.createMessage({
|
|
text: "http://localhost:" + port + "/thumb",
|
|
});
|
|
|
|
link(this.irc, this.network.channels[0], message);
|
|
|
|
this.app.get("/thumb", function (req, res) {
|
|
res.send(
|
|
"<title>Google</title><meta property='og:image' content='http://localhost:" +
|
|
port +
|
|
"/real-test-image.png'>"
|
|
);
|
|
});
|
|
|
|
this.irc.once("msg:preview", function (data) {
|
|
expect(data.preview.head).to.equal("Google");
|
|
expect(data.preview.link).to.equal("http://localhost:" + port + "/thumb");
|
|
expect(data.preview.thumb).to.equal(correctImageURL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should store the image", function (done) {
|
|
const port = this.port;
|
|
const message = this.irc.createMessage({
|
|
text: "http://localhost:" + port + "/real-test-image.png",
|
|
});
|
|
|
|
link(this.irc, this.network.channels[0], message);
|
|
|
|
this.irc.once("msg:preview", function (data) {
|
|
expect(data.preview.type).to.equal("image");
|
|
expect(data.preview.link).to.equal("http://localhost:" + port + "/real-test-image.png");
|
|
expect(data.preview.thumb).to.equal(correctImageURL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should lookup correct extension type", function (done) {
|
|
const port = this.port;
|
|
const message = this.irc.createMessage({
|
|
text: "http://localhost:" + port + "/svg-preview",
|
|
});
|
|
|
|
this.app.get("/svg-preview", function (req, res) {
|
|
res.send(
|
|
"<title>test title</title><meta property='og:image' content='http://localhost:" +
|
|
port +
|
|
"/logo.svg'>"
|
|
);
|
|
});
|
|
|
|
link(this.irc, this.network.channels[0], message);
|
|
|
|
this.irc.once("msg:preview", function (data) {
|
|
expect(data.preview.type).to.equal("link");
|
|
expect(data.preview.link).to.equal("http://localhost:" + port + "/svg-preview");
|
|
expect(data.preview.thumb).to.equal(correctSvgURL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should clear storage folder", function () {
|
|
const dir = Helper.getStoragePath();
|
|
|
|
expect(fs.readdirSync(dir)).to.not.be.empty;
|
|
storage.emptyDir();
|
|
expect(fs.readdirSync(dir)).to.be.empty;
|
|
expect(fs.existsSync(dir)).to.be.true;
|
|
});
|
|
});
|