diff --git a/src/bot.js b/src/bot.js index d684f23..aaece1d 100644 --- a/src/bot.js +++ b/src/bot.js @@ -82,7 +82,7 @@ function printGuilds() { const debouncePrintGuilds = utils.debounce(printGuilds, 2000); export const debounceLogin = utils.debounce(client.login, 60000); -(async () => { +export const login = async function() { try { await client.login(token); // await db.check_connection(); @@ -90,4 +90,11 @@ export const debounceLogin = utils.debounce(client.login, 60000); console.error('Login failed:', err); process.exit(1); } -})(); \ No newline at end of file +} + +function main() +{ + login(); +} + +main(); \ No newline at end of file diff --git a/tests/bot.test.js b/tests/bot.test.js index a48bb57..634acf4 100644 --- a/tests/bot.test.js +++ b/tests/bot.test.js @@ -65,6 +65,7 @@ const {webhookHelper} = require("../src/helpers/webhookHelper.js"); const {utils} = require("../src/helpers/utils.js"); let {handleMessageCreate, client} = require("../src/bot.js"); +const {login} = require("../src/bot"); describe('bot', () => { beforeEach(() => { @@ -290,9 +291,11 @@ describe('bot', () => { }) }) - test('calls client.login with correct argument', () => { - // Act + test('login calls client.login with correct argument', async() => { + // Arrange client.login = jest.fn().mockResolvedValue(); + // Act + await login(); // Assert expect(client.login).toHaveBeenCalledTimes(1); expect(client.login).toHaveBeenCalledWith(process.env.FLUXER_BOT_TOKEN)