diff --git a/src/generate.ts b/src/generate.ts index aa3c47b..3ac9462 100644 --- a/src/generate.ts +++ b/src/generate.ts @@ -250,7 +250,7 @@ ${Object.keys(m.tables) }) .join('\n\n')} -export default interface $Model { +export interface $Model { ${Object.keys(m.tables) .map( (tableName) => @@ -265,5 +265,7 @@ ${Object.keys(m.synonyms) ) .join('\n')} } + +export default $Model; `; }; diff --git a/test/index.ts b/test/index.ts index b9d34f5..2bc0a36 100644 --- a/test/index.ts +++ b/test/index.ts @@ -36,11 +36,14 @@ test( {}, ` -export default interface $Model { +export interface $Model { -}`, +} + +export default $Model; +`, ); const testTable: Partial = { @@ -376,7 +379,7 @@ test( }; } - export default interface $Model { + export interface $Model { parent: Parent; other: Other; test: Test; @@ -384,5 +387,7 @@ test( // Synonyms test_tag: TestTag; } + + export default $Model; `, );