mirror of
https://github.com/placeholder-soft/libs.git
synced 2026-01-13 07:20:03 +08:00
test: create dataset
This commit is contained in:
1
packages/bigquery/src/lib/__fixtures__/.gitignore
vendored
Normal file
1
packages/bigquery/src/lib/__fixtures__/.gitignore
vendored
Normal file
@@ -0,0 +1 @@
|
||||
*.json
|
||||
100
packages/bigquery/src/lib/__fixtures__/tiktok-models.ts
Normal file
100
packages/bigquery/src/lib/__fixtures__/tiktok-models.ts
Normal 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);
|
||||
}
|
||||
26
packages/bigquery/src/lib/__tests__/bigquery-store.test.ts
Normal file
26
packages/bigquery/src/lib/__tests__/bigquery-store.test.ts
Normal 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();
|
||||
});
|
||||
});
|
||||
@@ -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": [],
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
{
|
||||
"extends": "./tsconfig.json",
|
||||
"compilerOptions": {
|
||||
"module": "commonjs",
|
||||
"outDir": "../../dist/out-tsc",
|
||||
"declaration": true,
|
||||
"types": ["node"]
|
||||
|
||||
@@ -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"
|
||||
]
|
||||
}
|
||||
|
||||
@@ -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"]
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user