Infrastructure/astro
0
Fork 0
mirror of https://github.com/withastro/astro.git synced 2025-02-10 22:38:53 -05:00
Code Issues Activity

feat(test): foreignKeys

Browse source
This commit is contained in:
bholmesdev 2024-02-06 17:27:50 -05:00
parent 981e1e92a2
commit 3c03c31a79
1 changed files with 72 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

81
packages/db/test/unit/reference-queries.test.js
View file

@ -16,7 +16,8 @@ let User = defineCollection({
let SentBox = defineCollection({ let SentBox = defineCollection({
fields: { fields: {
to: field.text({ references: () => User.fields.id }), to: field.text(),
toName: field.text(),
subject: field.text(), subject: field.text(),
body: field.text(), body: field.text(),
}, },
@ -49,11 +50,34 @@ function userChangeQueries(
describe('reference queries', () => { describe('reference queries', () => {
it('adds references with lossless table recreate', async () => { it('adds references with lossless table recreate', async () => {
const SentBoxFinal = collectionSchema.parse(
defineCollection({
fields: {
...SentBox.fields,
to: field.text({ references: () => User.fields.id }),
},
})
);
const { queries } = await userChangeQueries(SentBox, SentBoxFinal);
expect(queries[0]).to.not.be.undefined;
const tempTableName = getTempTableName(queries[0]);
expect(queries).to.deep.equal([
`CREATE TABLE \"${tempTableName}\" (_id INTEGER PRIMARY KEY, \"to\" text NOT NULL REFERENCES \"User\" (\"id\"), \"toName\" text NOT NULL, \"subject\" text NOT NULL, \"body\" text NOT NULL)`,
`INSERT INTO \"${tempTableName}\" (\"_id\", \"to\", \"toName\", \"subject\", \"body\") SELECT \"_id\", \"to\", \"toName\", \"subject\", \"body\" FROM \"User\"`,
'DROP TABLE "User"',
`ALTER TABLE \"${tempTableName}\" RENAME TO \"User\"`,
]);
});
it('removes references with lossless table recreate', async () => {
const SentBoxInitial = collectionSchema.parse( const SentBoxInitial = collectionSchema.parse(
defineCollection({ defineCollection({
fields: { fields: {
...SentBox.fields, ...SentBox.fields,
to: field.text(), to: field.text({ references: () => User.fields.id }),
}, },
}) })
); );
@ -64,34 +88,73 @@ describe('reference queries', () => {
const tempTableName = getTempTableName(queries[0]); const tempTableName = getTempTableName(queries[0]);
expect(queries).to.deep.equal([ expect(queries).to.deep.equal([
`CREATE TABLE \"${tempTableName}\" (_id INTEGER PRIMARY KEY, \"to\" text NOT NULL REFERENCES \"User\" (\"id\"), \"subject\" text NOT NULL, \"body\" text NOT NULL)`, `CREATE TABLE \"${tempTableName}\" (_id INTEGER PRIMARY KEY, \"to\" text NOT NULL, \"toName\" text NOT NULL, \"subject\" text NOT NULL, \"body\" text NOT NULL)`,
`INSERT INTO \"${tempTableName}\" (\"_id\", \"to\", \"subject\", \"body\") SELECT \"_id\", \"to\", \"subject\", \"body\" FROM \"User\"`, `INSERT INTO \"${tempTableName}\" (\"_id\", \"to\", \"toName\", \"subject\", \"body\") SELECT \"_id\", \"to\", \"toName\", \"subject\", \"body\" FROM \"User\"`,
'DROP TABLE "User"', 'DROP TABLE "User"',
`ALTER TABLE \"${tempTableName}\" RENAME TO \"User\"`, `ALTER TABLE \"${tempTableName}\" RENAME TO \"User\"`,
]); ]);
}); });
it('removes references with lossless table recreate', async () => { it('does not use ADD COLUMN when adding optional column with reference', async () => {
const SentBoxFinal = collectionSchema.parse( const SentBoxFinal = collectionSchema.parse(
defineCollection({ defineCollection({
fields: { fields: {
...SentBox.fields, ...SentBox.fields,
to: field.text(), from: field.text({ references: () => User.fields.id, optional: true }),
}, },
}) })
); );
const { queries } = await userChangeQueries(SentBox, SentBoxFinal);
const { queries } = await userChangeQueries(SentBox, SentBoxFinal);
expect(queries[0]).to.not.be.undefined; expect(queries[0]).to.not.be.undefined;
const tempTableName = getTempTableName(queries[0]); const tempTableName = getTempTableName(queries[0]);
expect(queries).to.deep.equal([ expect(queries).to.deep.equal([
`CREATE TABLE \"${tempTableName}\" (_id INTEGER PRIMARY KEY, \"to\" text NOT NULL, \"subject\" text NOT NULL, \"body\" text NOT NULL)`, `CREATE TABLE \"${tempTableName}\" (_id INTEGER PRIMARY KEY, \"to\" text NOT NULL, \"toName\" text NOT NULL, \"subject\" text NOT NULL, \"body\" text NOT NULL, \"from\" text REFERENCES \"User\" (\"id\"))`,
`INSERT INTO \"${tempTableName}\" (\"_id\", \"to\", \"subject\", \"body\") SELECT \"_id\", \"to\", \"subject\", \"body\" FROM \"User\"`, `INSERT INTO \"${tempTableName}\" (\"_id\", \"to\", \"toName\", \"subject\", \"body\") SELECT \"_id\", \"to\", \"toName\", \"subject\", \"body\" FROM \"User\"`,
'DROP TABLE "User"', 'DROP TABLE "User"',
`ALTER TABLE \"${tempTableName}\" RENAME TO \"User\"`, `ALTER TABLE \"${tempTableName}\" RENAME TO \"User\"`,
]); ]);
}); });
it('adds and updates foreign key with lossless table recreate', async () => {
const SentBoxFinal = collectionSchema.parse(
defineCollection({
...SentBox,
foreignKeys: [
{ fields: ['to', 'toName'], references: () => [User.fields.id, User.fields.name] },
],
})
);
const SentBoxWithDifferentFK = collectionSchema.parse(
defineCollection({
...SentBox,
foreignKeys: [{ fields: ['to'], references: () => [User.fields.id] }],
})
);
const expected = (tempTableName) => [
`CREATE TABLE \"${tempTableName}\" (_id INTEGER PRIMARY KEY, \"to\" text NOT NULL, \"toName\" text NOT NULL, \"subject\" text NOT NULL, \"body\" text NOT NULL, FOREIGN KEY (\"to\", \"toName\") REFERENCES \"User\"(\"id\", \"name\"))`,
`INSERT INTO \"${tempTableName}\" (\"_id\", \"to\", \"toName\", \"subject\", \"body\") SELECT \"_id\", \"to\", \"toName\", \"subject\", \"body\" FROM \"User\"`,
'DROP TABLE "User"',
`ALTER TABLE \"${tempTableName}\" RENAME TO \"User\"`,
];
const addedForeignKey = await userChangeQueries(SentBox, SentBoxFinal);
const updatedForeignKey = await userChangeQueries(SentBoxWithDifferentFK, SentBoxFinal);
expect(addedForeignKey.queries[0]).to.not.be.undefined;
expect(updatedForeignKey.queries[0]).to.not.be.undefined;
expect(addedForeignKey.queries).to.deep.equal(
expected(getTempTableName(addedForeignKey.queries[0]))
);
expect(updatedForeignKey.queries).to.deep.equal(
expected(getTempTableName(updatedForeignKey.queries[0]))
);
});
}); });
/** @param {string} query */ /** @param {string} query */