Add a seed script to insert a test user. #1

.env.example

@@ -13,3 +13,8 @@ POSTGRES_DB=poke-admin
 
 # Test Variables
 API_URL=http://localhost:3000
+
+# Admin User Variables
+ADMIN_USER_ID=21247c46-90db-469d-a3fc-41febde52bf0
+ADMIN_USERNAME=admin
+ADMIN_PASSWORD=password

api/test/index.js

@@ -1,29 +1,28 @@
 import axios from 'axios'
-import bcrypt from 'bcrypt'
 import chai from 'chai'
-import faker from 'faker'
 import noop from 'lodash/noop'
 import { chaiStruct } from 'chai-struct'
 import createApi from '../create-api'
-import { knex, redis } from '../../database/connections'
+import * as admin from '../../database/seeds/admin'
+import { createKnex, createRedis } from '../../database/connections'
 
 chai.use(chaiStruct)
 
 let server
+let knex
+let redis
 let _knex
 let _redis
 let _trx
 
 before('start root transaction and insert test user into database', done => {
+  knex = createKnex()
+  redis = createRedis()
   knex.transaction(trx => {
     _trx = trx
     ;(async () => {
       try {
-        const { password: unhashed } = TEST_USER
-        const password = await bcrypt.hash(unhashed, 1)
-        await _trx
-          .insert({ ...TEST_USER, password })
-          .into('users')
+        await admin.seed(trx)
         done()
       }
       catch (err) {
@@ -74,10 +73,5 @@ export const client = axios.create({
   baseURL: process.env.API_URL
 })
 
-export const TEST_USER = {
-  userId: faker.random.uuid(),
-  username: 'admin',
-  password: 'password'
-}
-
-export { expect } from 'chai'
+export const { expect } = chai
+export const { user: TEST_USER } = admin

client/lib/authorize.js

@@ -0,0 +1,12 @@
+import noop from 'lodash/noop'
+
+export default function authorize(getInitialProps = noop) {
+  return ({ res, router, session, isServer, ...ctx }) => {
+    if (!session.user) {
+      return isServer
+        ? res.redirect('/login')
+        : router.replace('/login')
+    }
+    return getInitialProps({ res, router, session, isServer, ...ctx })
+  }
+}

client/lib/index.js

@@ -1 +1,2 @@
+export { default as authorize } from './authorize'
 export { default as isServer } from './is-server'

client/pages/index.jsx

@@ -1,4 +1,5 @@
-import { Consumer } from '../lib'
+import { authorize } from '../lib'
+import { Consumer } from '../services'
 
 export default function Index() {
   return (
@@ -11,3 +12,5 @@ export default function Index() {
     </Consumer>
   )
 }
+
+Index.getInitialProps = authorize()

database/connections.js

@@ -6,13 +6,13 @@ import camelKeys from './lib/camel-keys'
 
 promisifyAll(RedisClient.prototype)
 
-export const knex = Knex({
+export const createKnex = () => Knex({
   client: 'pg',
   connection: process.env.POSTGRES_URL,
   postProcessResponse: camelKeys,
   wrapIdentifier: (value, wrap) => wrap(snakeCase(value))
 })
 
-export const redis = new RedisClient({
+export const createRedis = () => new RedisClient({
   url: process.env.REDIS_URL
 })

database/seeds/__dev__.js

@@ -0,0 +1,19 @@
+import path from 'path'
+import { createKnex, createRedis } from '../connections'
+
+(async () => {
+  const knex = createKnex()
+  const redis = createRedis()
+  try {
+    await redis.flushallAsync()
+    await knex.seed.run({
+      directory: path.join(__dirname, 'dev/')
+    })
+    await knex.destroy()
+    await redis.quitAsync()
+  }
+  catch (err) {
+    console.error(err)
+    process.exit(1)
+  }
+})()

database/seeds/admin.js

@@ -0,0 +1,16 @@
+import bcrypt from 'bcrypt'
+
+export const user = {
+  userId: process.env.ADMIN_USER_ID,
+  username: process.env.ADMIN_USERNAME,
+  password: process.env.ADMIN_PASSWORD
+}
+
+export async function seed(knex) {
+  const saltRounds = process.env.NODE_ENV === 'production' ? 12 : 1
+  const password = await bcrypt.hash(user.password, saltRounds)
+  await knex.raw('truncate table users cascade')
+  await knex
+    .insert({ ...user, password })
+    .into('users')
+}

database/seeds/dev/index.js

@@ -0,0 +1,26 @@
+import chalk from 'chalk'
+import * as admin from '../admin'
+
+export async function seed(knex) {
+
+  const trySeed = (knex => async (script, description) => {
+    /* eslint-disable no-console */
+    console.log(chalk.white(`Running seed script for ${description}...`))
+    try {
+      await script.seed(knex)
+      console.log(chalk.green('done'))
+    }
+    catch (err) {
+      console.error(err)
+      console.error(chalk.red(`ERROR:Failed to run seed script for ${description}.`))
+      process.exit(1)
+    }
+  })(knex)
+
+  console.log('\n')
+
+  await trySeed(admin, 'admin user')
+  await admin.seed(knex)
+
+  console.log('\n')
+}

package.json

@@ -10,9 +10,10 @@
     "db:up": "pg-bump up",
     "db:down": "pg-bump down",
     "db:cycle": "run-s -s db:down db:up",
+    "db:seed": "node -r dotenv/config -r esm database/seeds/__dev__.js",
     "build": "next build client/",
     "start": "node -r dotenv/config -r esm .",
-    "dev": "run-s -s db:cycle && nodemon .",
+    "dev": "run-s -s db:cycle db:seed && nodemon .",
     "lint": "eslint --fix .",
     "test:api": "mocha --opts api/test/mocha.opts",
     "tdd:api": "run-s -s db:cycle 'test:api -- -w -R min'",
@@ -43,7 +44,7 @@
       "text-summary"
     ],
     "include": [
-      "**/*.js"
+      "api/**/*.js"
     ],
     "exclude": [
       "**/test/**",
@@ -142,6 +143,7 @@
     "bcrypt": "3.0.4",
     "bluebird": "3.5.3",
     "boom": "7.3.0",
+    "chalk": "2.4.2",
     "connect-redis": "3.4.0",
     "dotenv": "6.2.0",
     "esm": "3.2.14",
@@ -175,7 +177,6 @@
     "eslint-plugin-promise": "4.0.1",
     "eslint-plugin-react": "7.12.4",
     "eslint-plugin-standard": "4.0.0",
-    "faker": "4.1.0",
     "husky": "1.3.1",
     "lint-staged": "8.1.5",
     "mocha": "6.0.2",

server/index.js

@@ -1,7 +1,9 @@
 import createServer from './create-server'
-import { knex, redis } from '../database/connections'
+import { createKnex, createRedis } from '../database/connections'
 
 const dev = process.env.NODE_ENV !== 'production'
+const knex = createKnex()
+const redis = createRedis()
 
 ;(async () => {
   const server = await createServer({ dev, knex, redis })