Skip to content

Commit

Permalink
restore sequence
Browse files Browse the repository at this point in the history
  • Loading branch information
@stevensJourney
stevensJourney committed Feb 11, 2025
1 parent 1a1d373 commit f2ad5a7
Showing 1 changed file with 169 additions and 163 deletions.
332 changes: 169 additions & 163 deletions packages/web/tests/stream.test.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,177 +5,183 @@ import { ConnectedDatabaseUtils, generateConnectedDatabase } from './utils/gener

const UPLOAD_TIMEOUT_MS = 3000;

describe('Streaming', () => {
/**
* Declares a test to be executed with different generated db functions
*/
const itWithGenerators = (
name: string,
test: (createConnectedDatabase: () => ReturnType<typeof generateConnectedDatabase>) => Promise<void>
) => {
const funcWithWebWorker = generateConnectedDatabase;
const funcWithoutWebWorker = () =>
generateConnectedDatabase({
powerSyncOptions: {
dbFilename: 'test-stream-connection-no-worker.db',
flags: {
useWebWorker: false
describe(
'Streaming',
{
sequential: true
},
() => {
/**
* Declares a test to be executed with different generated db functions
*/
const itWithGenerators = (
name: string,
test: (createConnectedDatabase: () => ReturnType<typeof generateConnectedDatabase>) => Promise<void>
) => {
const funcWithWebWorker = generateConnectedDatabase;
const funcWithoutWebWorker = () =>
generateConnectedDatabase({
powerSyncOptions: {
dbFilename: 'test-stream-connection-no-worker.db',
flags: {
useWebWorker: false
}
}
}
});
});

it(`${name} - with web worker`, () => test(funcWithWebWorker));
it(`${name} - without web worker`, () => test(funcWithoutWebWorker));
};

beforeAll(() => Logger.useDefaults());

let connectionUtilities: ConnectedDatabaseUtils | null = null;

afterEach(async () => {
await connectionUtilities?.dispose();
connectionUtilities = null;
});

itWithGenerators('PowerSync reconnect on closed stream', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, waitForStream, remote } = connectionUtilities;

expect(powersync.connected).toBe(true);

// Close the stream
const newStream = waitForStream();
remote.streamController?.close();

it(`${name} - with web worker`, () => test(funcWithWebWorker));
it(`${name} - without web worker`, () => test(funcWithoutWebWorker));
};

beforeAll(() => Logger.useDefaults());

let connectionUtilities: ConnectedDatabaseUtils | null = null;

afterEach(async () => {
await connectionUtilities?.dispose();
connectionUtilities = null;
});

itWithGenerators('PowerSync reconnect on closed stream', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, waitForStream, remote } = connectionUtilities;

expect(powersync.connected).toBe(true);

// Close the stream
const newStream = waitForStream();
remote.streamController?.close();

// A new stream should be requested
await newStream;
});

itWithGenerators('PowerSync reconnect multiple connect calls', async (createConnectedDatabase) => {
// This initially performs a connect call
connectionUtilities = await createConnectedDatabase();
const { powersync, waitForStream, remote } = connectionUtilities;
expect(powersync.connected).toBe(true);

// Call connect again, a new stream should be requested
const newStream = waitForStream();
powersync.connect(new TestConnector());

// A new stream should be requested
await newStream;
});

itWithGenerators('Should trigger upload connector when connected', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, uploadSpy } = connectionUtilities;
expect(powersync.connected).toBe(true);

// do something which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);
// It should try and upload
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing the first check
expect(uploadSpy.mock.calls.length).equals(1);
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});

itWithGenerators('Should retry failed uploads when connected', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, uploadSpy } = connectionUtilities;

expect(powersync.connected).toBe(true);

let uploadCounter = 0;
// This test will throw an exception a few times before uploading
const throwCounter = 2;
uploadSpy.mockImplementation(async (db) => {
if (uploadCounter++ < throwCounter) {
throw new Error('No uploads yet');
}
// Now actually do the upload
const tx = await db.getNextCrudTransaction();
await tx?.complete();
// A new stream should be requested
await newStream;
});

// do something which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);

// It should try and upload
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing a check
expect(uploadSpy.mock.calls.length).equals(throwCounter + 1);
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});

itWithGenerators('Should upload after reconnecting', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, connect, uploadSpy } = connectionUtilities;
expect(powersync.connected).toBe(true);

await powersync.disconnect();

// do something (offline) which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);

await connect();

// It should try and upload
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing a check
expect(uploadSpy.mock.calls.length).equals(1);
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});

itWithGenerators('Should update status when uploading', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, uploadSpy } = connectionUtilities;

expect(powersync.connected).toBe(true);

let uploadStartedPromise = new Promise<void>((resolve) => {
itWithGenerators('PowerSync reconnect multiple connect calls', async (createConnectedDatabase) => {
// This initially performs a connect call
connectionUtilities = await createConnectedDatabase();
const { powersync, waitForStream, remote } = connectionUtilities;
expect(powersync.connected).toBe(true);

// Call connect again, a new stream should be requested
const newStream = waitForStream();
powersync.connect(new TestConnector());

// A new stream should be requested
await newStream;
});

itWithGenerators('Should trigger upload connector when connected', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, uploadSpy } = connectionUtilities;
expect(powersync.connected).toBe(true);

// do something which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);
// It should try and upload
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing the first check
expect(uploadSpy.mock.calls.length).equals(1);
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});

itWithGenerators('Should retry failed uploads when connected', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, uploadSpy } = connectionUtilities;

expect(powersync.connected).toBe(true);

let uploadCounter = 0;
// This test will throw an exception a few times before uploading
const throwCounter = 2;
uploadSpy.mockImplementation(async (db) => {
resolve();
if (uploadCounter++ < throwCounter) {
throw new Error('No uploads yet');
}
// Now actually do the upload
const tx = await db.getNextCrudTransaction();
await tx?.complete();
});

// do something which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);

// It should try and upload
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing a check
expect(uploadSpy.mock.calls.length).equals(throwCounter + 1);
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});

itWithGenerators('Should upload after reconnecting', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, connect, uploadSpy } = connectionUtilities;
expect(powersync.connected).toBe(true);

await powersync.disconnect();

// do something (offline) which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);

await connect();

// It should try and upload
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing a check
expect(uploadSpy.mock.calls.length).equals(1);
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});

// do something which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);

await uploadStartedPromise;

expect(powersync.currentStatus.dataFlowStatus.uploading).true;

// Status should update after uploads are completed
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing a check
expect(powersync.currentStatus.dataFlowStatus.uploading).false;
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});
});
itWithGenerators('Should update status when uploading', async (createConnectedDatabase) => {
connectionUtilities = await createConnectedDatabase();
const { powersync, uploadSpy } = connectionUtilities;

expect(powersync.connected).toBe(true);

let uploadStartedPromise = new Promise<void>((resolve) => {
uploadSpy.mockImplementation(async (db) => {
resolve();
// Now actually do the upload
const tx = await db.getNextCrudTransaction();
await tx?.complete();
});
});

// do something which should trigger an upload
await powersync.execute('INSERT INTO users (id, name) VALUES (uuid(), ?)', ['name']);

await uploadStartedPromise;

expect(powersync.currentStatus.dataFlowStatus.uploading).true;

// Status should update after uploads are completed
await vi.waitFor(
() => {
// to-have-been-called seems to not work after failing a check
expect(powersync.currentStatus.dataFlowStatus.uploading).false;
},
{
timeout: UPLOAD_TIMEOUT_MS,
interval: 500
}
);
});
}
);

0 comments on commit f2ad5a7

Please sign in to comment.