84 lines
1.7 KiB
JavaScript
84 lines
1.7 KiB
JavaScript
|
const fetch = require("node-fetch")
|
||
|
|
const { getProdAppID } = require("@budibase/backend-core/db")
|
||
|
|
|
||
|
|
const USER_LOAD_NUMBER = 10000
|
||
|
|
const BATCH_SIZE = 25
|
||
|
|
const SERVER_URL = "http://localhost:4001"
|
||
|
|
const PASSWORD = "test"
|
||
|
|
|
||
|
|
const APP_ID = process.argv[2]
|
||
|
|
const API_KEY = process.argv[3]
|
||
|
|
|
||
|
|
const words = [
|
||
|
|
"test",
|
||
|
|
"testing",
|
||
|
|
"budi",
|
||
|
|
"mail",
|
||
|
|
"age",
|
||
|
|
"risk",
|
||
|
|
"load",
|
||
|
|
"uno",
|
||
|
|
"arm",
|
||
|
|
"leg",
|
||
|
|
"pen",
|
||
|
|
"glass",
|
||
|
|
"box",
|
||
|
|
"chicken",
|
||
|
|
"bottle",
|
||
|
|
]
|
||
|
|
|
||
|
|
if (!APP_ID) {
|
||
|
|
console.error("Must supply app ID as first CLI option!")
|
||
|
|
process.exit(-1)
|
||
|
|
}
|
||
|
|
if (!API_KEY) {
|
||
|
|
console.error("Must supply API key as second CLI option!")
|
||
|
|
process.exit(-1)
|
||
|
|
}
|
||
|
|
|
||
|
|
const WORD_1 = words[Math.floor(Math.random() * words.length)]
|
||
|
|
const WORD_2 = words[Math.floor(Math.random() * words.length)]
|
||
|
|
|
||
|
|
function generateUser(count) {
|
||
|
|
return {
|
||
|
|
password: PASSWORD,
|
||
|
|
email: `${WORD_1}${count}@${WORD_2}.com`,
|
||
|
|
roles: {
|
||
|
|
[getProdAppID(APP_ID)]: "BASIC",
|
||
|
|
},
|
||
|
|
status: "active",
|
||
|
|
forceResetPassword: false,
|
||
|
|
firstName: "John",
|
||
|
|
lastName: "Smith",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
async function run() {
|
||
|
|
for (let i = 0; i < USER_LOAD_NUMBER; i += BATCH_SIZE) {
|
||
|
|
let promises = []
|
||
|
|
for (let j = 0; j < BATCH_SIZE; j++) {
|
||
|
|
promises.push(
|
||
|
|
fetch(`${SERVER_URL}/api/public/v1/users`, {
|
||
|
|
method: "POST",
|
||
|
|
body: JSON.stringify(generateUser(i + j)),
|
||
|
|
headers: {
|
||
|
|
"x-budibase-api-key": API_KEY,
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
},
|
||
|
|
})
|
||
|
|
)
|
||
|
|
}
|
||
|
|
await Promise.all(promises)
|
||
|
|
console.log(`${i + BATCH_SIZE} users have been created.`)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
run()
|
||
|
|
.then(() => {
|
||
|
|
console.log(`Generated ${USER_LOAD_NUMBER} users!`)
|
||
|
|
})
|
||
|
|
.catch(err => {
|
||
|
|
console.error("Failed for reason: ", err)
|
||
|
|
process.exit(-1)
|
||
|
|
})
|