Skip to content

Commit

Permalink
fix replica integration tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@Rodriguespn
Rodriguespn committed Nov 14, 2024
1 parent 2816180 commit 0969a76
Show file tree
Hide file tree
Showing 3 changed files with 103 additions and 98 deletions.
160 changes: 80 additions & 80 deletions integration-tests/tests/replicas/singlestore.test.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,9 @@ const users = singlestoreTable('users', {

describe('[select] read replicas singlestore', () => {
it('primary select', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand All @@ -35,9 +35,9 @@ describe('[select] read replicas singlestore', () => {
});

it('random replica select', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const randomMockReplica = vi.fn().mockReturnValueOnce(read1).mockReturnValueOnce(read2);

Expand All @@ -64,8 +64,8 @@ describe('[select] read replicas singlestore', () => {
});

it('single read replica select', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -84,8 +84,8 @@ describe('[select] read replicas singlestore', () => {
});

it('single read replica select + primary select', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -105,9 +105,9 @@ describe('[select] read replicas singlestore', () => {
});

it('always first read select', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2], (replicas) => {
return replicas[0]!;
Expand All @@ -134,9 +134,9 @@ describe('[select] read replicas singlestore', () => {

describe('[selectDistinct] read replicas singlestore', () => {
it('primary selectDistinct', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand All @@ -153,9 +153,9 @@ describe('[selectDistinct] read replicas singlestore', () => {
});

it('random replica selectDistinct', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const randomMockReplica = vi.fn().mockReturnValueOnce(read1).mockReturnValueOnce(read2);

Expand All @@ -181,8 +181,8 @@ describe('[selectDistinct] read replicas singlestore', () => {
});

it('single read replica selectDistinct', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -201,8 +201,8 @@ describe('[selectDistinct] read replicas singlestore', () => {
});

it('single read replica selectDistinct + primary selectDistinct', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -222,9 +222,9 @@ describe('[selectDistinct] read replicas singlestore', () => {
});

it('always first read selectDistinct', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2], (replicas) => {
return replicas[0]!;
Expand All @@ -250,9 +250,9 @@ describe('[selectDistinct] read replicas singlestore', () => {

describe('[with] read replicas singlestore', () => {
it('primary with', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand All @@ -273,9 +273,9 @@ describe('[with] read replicas singlestore', () => {
});

it('random replica with', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const randomMockReplica = vi.fn().mockReturnValueOnce(read1).mockReturnValueOnce(read2);

Expand All @@ -299,8 +299,8 @@ describe('[with] read replicas singlestore', () => {
});

it('single read replica with', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -317,8 +317,8 @@ describe('[with] read replicas singlestore', () => {
});

it('single read replica with + primary with', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -336,9 +336,9 @@ describe('[with] read replicas singlestore', () => {
});

it('always first read with', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2], (replicas) => {
return replicas[0]!;
Expand Down Expand Up @@ -367,9 +367,9 @@ describe('[with] read replicas singlestore', () => {

describe('[update] replicas singlestore', () => {
it('primary update', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand Down Expand Up @@ -402,9 +402,9 @@ describe('[update] replicas singlestore', () => {

describe('[delete] replicas singlestore', () => {
it('primary delete', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand Down Expand Up @@ -438,9 +438,9 @@ describe('[delete] replicas singlestore', () => {

describe('[insert] replicas singlestore', () => {
it('primary insert', () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand Down Expand Up @@ -473,9 +473,9 @@ describe('[insert] replicas singlestore', () => {

describe('[execute] replicas singlestore', () => {
it('primary execute', async () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand Down Expand Up @@ -515,9 +515,9 @@ describe('[execute] replicas singlestore', () => {

describe('[transaction] replicas singlestore', () => {
it('primary transaction', async () => {
const primaryDb = drizzle({} as any);
const read1 = drizzle({} as any);
const read2 = drizzle({} as any);
const primaryDb = drizzle.mock();
const read1 = drizzle.mock();
const read2 = drizzle.mock();

const db = withReplicas(primaryDb, [read1, read2]);

Expand Down Expand Up @@ -558,9 +558,9 @@ describe('[transaction] replicas singlestore', () => {

describe('[findFirst] read replicas singlestore', () => {
it('primary findFirst', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const read2 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });
const read2 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1, read2]);

Expand All @@ -578,9 +578,9 @@ describe('[findFirst] read replicas singlestore', () => {
});

it('random replica findFirst', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const read2 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });
const read2 = drizzle.mock({ schema: { usersTable } });

const randomMockReplica = vi.fn().mockReturnValueOnce(read1).mockReturnValueOnce(read2);

Expand All @@ -607,8 +607,8 @@ describe('[findFirst] read replicas singlestore', () => {
});

it('single read replica findFirst', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -625,8 +625,8 @@ describe('[findFirst] read replicas singlestore', () => {
});

it('single read replica findFirst + primary findFirst', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -644,9 +644,9 @@ describe('[findFirst] read replicas singlestore', () => {
});

it('always first read findFirst', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const read2 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });
const read2 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1, read2], (replicas) => {
return replicas[0]!;
Expand All @@ -670,9 +670,9 @@ describe('[findFirst] read replicas singlestore', () => {

describe('[findMany] read replicas singlestore', () => {
it('primary findMany', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const read2 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });
const read2 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1, read2]);

Expand All @@ -691,9 +691,9 @@ describe('[findMany] read replicas singlestore', () => {
});

it('random replica findMany', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const read2 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });
const read2 = drizzle.mock({ schema: { usersTable } });

const randomMockReplica = vi.fn().mockReturnValueOnce(read1).mockReturnValueOnce(read2);

Expand Down Expand Up @@ -724,8 +724,8 @@ describe('[findMany] read replicas singlestore', () => {
});

it('single read replica findMany', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -748,8 +748,8 @@ describe('[findMany] read replicas singlestore', () => {
});

it('single read replica findMany + primary findMany', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1]);

Expand All @@ -774,9 +774,9 @@ describe('[findMany] read replicas singlestore', () => {
});

it('always first read findMany', () => {
const primaryDb = drizzle({} as any, { schema: { usersTable } });
const read1 = drizzle({} as any, { schema: { usersTable } });
const read2 = drizzle({} as any, { schema: { usersTable } });
const primaryDb = drizzle.mock({ schema: { usersTable } });
const read1 = drizzle.mock({ schema: { usersTable } });
const read2 = drizzle.mock({ schema: { usersTable } });

const db = withReplicas(primaryDb, [read1, read2], (replicas) => {
return replicas[0]!;
Expand Down
Loading

0 comments on commit 0969a76

Please sign in to comment.