test: create dataset

This commit is contained in:
keyp-dev
2023-03-13 17:42:36 +08:00
parent fa70deb275
commit ff3fc4a2dc
7 changed files with 143 additions and 17 deletions

View File

@@ -0,0 +1 @@
*.json

View File

@@ -0,0 +1,100 @@
import { z, AnyZodObject } from 'zod';
import { convert } from 'zoq';
export const BasePageModel = z.object({
title: z.string(),
url: z.string(),
pathname: z.string(),
nanoid: z.string(),
created_at: z.coerce.date(),
});
export type BasePageModel = z.infer<typeof BasePageModel>;
export function withBasePageModel<Incoming extends AnyZodObject>(
page: Incoming
) {
return BasePageModel.merge(page);
}
export const kTikTokPages = {
sitemap: 'sitemap',
topic: 'topic',
account: 'account',
tag: 'tag',
} as const;
export type TikTokPage = keyof typeof kTikTokPages;
const TikTokSitemapTopic = z.object({
topic: z.string(),
url: z.string(),
});
export type TikTokSitemapTopic = z.infer<typeof TikTokSitemapTopic>;
const TikTokSitemapAccount = z.object({
name: z.string(),
url: z.string(),
});
export type TikTokSitemapAccount = z.infer<typeof TikTokSitemapAccount>;
const TikTokSitemapTag = z.object({
tag: z.string(),
url: z.string(),
});
export type TikTokSitemapTag = z.infer<typeof TikTokSitemapTag>;
const TikTokSitemapPageModel = withBasePageModel(
z.object({
topics: z.array(TikTokSitemapTopic),
accounts: z.array(TikTokSitemapAccount),
tags: z.array(TikTokSitemapTag),
})
);
export type TikTokSitemapPageModel = z.infer<typeof TikTokSitemapPageModel>;
const TikTokVideo = z.object({
author: z.string(),
content: z.string(),
music: z.object({ title: z.string(), url: z.string() }),
player: z.object({ image_url: z.string(), video_url: z.string() }),
stars: z.string(),
comments: z.string(),
shares: z.string().nullish(),
});
export type TikTokVideo = z.infer<typeof TikTokVideo>;
const TikTokVideoPageModel = withBasePageModel(
z.object({
data: z.array(TikTokVideo),
})
);
export type TikTokVideoPageModel = z.infer<typeof TikTokVideoPageModel>;
// export const TikTokPageModels2 = {
// sitemap: TikTokSitemapPageModel,
// topic: TikTokVideoPageModel,
// account: TikTokVideoPageModel,
// tag: TikTokVideoPageModel,
// } as const;
// export type TikTokModelName = keyof typeof TikTokPageModels;
const TikTokPageModels = z.object({
sitemap: TikTokSitemapPageModel,
topic: TikTokVideoPageModel,
account: TikTokVideoPageModel,
tag: TikTokVideoPageModel,
});
export type TikTokPageModels = z.infer<typeof TikTokPageModels>;
const TikTokModelNameEnum = TikTokPageModels.keyof();
export type TikTokModelName = z.infer<typeof TikTokModelNameEnum>;
export function getBigQuerySchemaByTikTok(name: TikTokModelName) {
// return convert(TikTokPageModels[name] as any);
return convert(TikTokPageModels.shape[name] as any);
}

View File

@@ -0,0 +1,26 @@
import path from 'path';
import { BigQueryStore } from '../bigquery-store';
import { getBigQuerySchemaByTikTok } from '../__fixtures__/tiktok-models';
const keyFilename = path.resolve(
__dirname,
'../__fixtures__/shopify-cloud-0-55a469e24366.json'
);
const projectId = 'shopify-cloud-0';
const datasetId = 'tiktok_appstore';
describe('BigQueryStore', () => {
test('create dataset', () => {
const store = new BigQueryStore(
{
projectId,
keyFilename,
},
datasetId,
getBigQuerySchemaByTikTok
);
store.ensureDataset();
});
});

View File

@@ -1,13 +1,7 @@
{
"extends": "../../tsconfig.base.json",
"compilerOptions": {
"module": "commonjs",
"forceConsistentCasingInFileNames": true,
"strict": true,
"noImplicitOverride": true,
"noPropertyAccessFromIndexSignature": true,
"noImplicitReturns": true,
"noFallthroughCasesInSwitch": true
"module": "commonjs"
},
"files": [],
"include": [],

View File

@@ -1,6 +1,7 @@
{
"extends": "./tsconfig.json",
"compilerOptions": {
"module": "commonjs",
"outDir": "../../dist/out-tsc",
"declaration": true,
"types": ["node"]

View File

@@ -7,8 +7,14 @@
},
"include": [
"jest.config.ts",
"src/**/*.test.ts",
"src/**/*.spec.ts",
"src/**/*.d.ts"
"**/*.test.ts",
"**/*.spec.ts",
"**/*.test.tsx",
"**/*.spec.tsx",
"**/*.test.js",
"**/*.spec.js",
"**/*.test.jsx",
"**/*.spec.jsx",
"**/*.d.ts"
]
}

View File

@@ -8,24 +8,22 @@
"emitDecoratorMetadata": true,
"experimentalDecorators": true,
"allowSyntheticDefaultImports": true,
"resolveJsonModule": true,
"allowJs": false,
"esModuleInterop": true,
"preserveConstEnums": true,
"preserveWatchOutput": true,
"importHelpers": true,
"target": "es2015",
"target": "esnext",
"module": "esnext",
"lib": ["es2017", "dom"],
"lib": ["esnext", "dom"],
"skipLibCheck": true,
"skipDefaultLibCheck": true,
"noImplicitAny": true,
"baseUrl": ".",
"paths": {
"@placeholdersoft/bigquery": ["packages/bigquery/src/index.ts"],
"@placeholdersoft/helpers": ["packages/helpers/src/index.ts"],
"@placeholdersoft/typed-env": ["packages/typed-env/src/index.ts"],
"@placeholdersoft/typed-env-cli": ["packages/typed-env-cli/src/index.ts"]
}
},
"typeRoots": ["node_modules/@types", "types"]
},
"exclude": ["node_modules", "tmp"]
}