feat(job): reset greenkeeper on a repositorty

couchdb/repositories/_design/by_full_name/views/by_full_name/map.js

@@ -0,0 +1,5 @@
+function (doc) {
+  if (doc.type !== 'repository') return
+
+  emit(doc.fullName.toLowerCase())
+}

jobs/reset.js

@@ -0,0 +1,83 @@
+const _ = require('lodash')
+const dbs = require('../lib/dbs')
+const githubQueue = require('../lib/github-queue')
+const { createDocs } = require('../lib/repository-docs')
+
+module.exports = async function ({ repositoryFullName }) {
+  // find the repository in the database
+  const { repositories, installations } = await dbs()
+  const repoDoc = _.get(
+    await repositories.query('by_full_name', {
+      key: repositoryFullName,
+      include_docs: true
+    }),
+    'rows[0].doc'
+  )
+
+  if (!repoDoc) {
+    const error = new Error(`The repository ${repositoryFullName} does not exist in the database`)
+    error.status = 404
+    throw error
+  }
+
+  // delete all prdocs
+  const prdocs = await repositories.allDocs({
+    include_docs: true,
+    startkey: `${repoDoc.id}:pr:`,
+    endkey: `${repoDoc.id}:pr:\ufff0`,
+    inclusive_end: true
+  })
+
+  const deletePrDocs = prdocs.rows.map(row => repositories.remove(row.doc))
+  await Promise.all(deletePrDocs)
+
+  // delete all greenkeeper branches in the repository
+  const branches = await repositories.allDocs({
+    include_docs: true,
+    startkey: `${repoDoc.id}:branch:`,
+    endkey: `${repoDoc.id}:branch:\ufff0`,
+    inclusive_end: true
+  })
+  const [owner, repo] = repositoryFullName.split('/')
+  const accountId = String(repoDoc.accountId)
+  const accountDoc = await installations.get(accountId)
+  const installationId = accountDoc.installation
+  const ghqueue = githubQueue(installationId)
+  for (let row of branches.rows) {
+    const branch = row.doc
+    try {
+      await ghqueue.write(github => github.gitdata.deleteReference({
+        owner,
+        repo,
+        ref: `heads/${branch.head}`
+      }))
+    } catch (e) {
+      if (branch.head === 'greenkeeper/initial' || branch.head === 'greenkeeper-initial') {
+        throw e
+      }
+    }
+  }
+
+  const deleteBranchDocs = branches.rows.map(row => repositories.remove(row.doc))
+  await Promise.all(deleteBranchDocs)
+
+  // get the current repository state from github
+  // to get the newest repo settings (e.g. user enabled issues in the mean time)
+  const githubRepository = await ghqueue.read(github => github.repos.get({ owner, repo }))
+
+  await repositories.remove(repoDoc)
+  await repositories.bulkDocs(createDocs({
+    repositories: [githubRepository],
+    accountId
+  }))
+
+  // enqueue create initial branch job
+  const newRepoDoc = await repositories.get(githubRepository.id)
+  return {
+    data: {
+      name: 'create-initial-branch',
+      repositoryId: newRepoDoc._id,
+      accountId
+    }
+  }
+}

test/jobs/reset.js

@@ -0,0 +1,239 @@
+const { test } = require('tap')
+const nock = require('nock')
+const worker = require('../../jobs/reset')
+
+const dbs = require('../../lib/dbs')
+const timeToWaitAfterTests = 500
+
+const waitFor = (milliseconds) => {
+  return new Promise((resolve) => {
+    setTimeout(resolve, milliseconds)
+  })
+}
+
+const removeIfExists = async (db, id) => {
+  try {
+    return await db.remove(await db.get(id))
+  } catch (e) {
+    if (e.status !== 404) {
+      throw e
+    }
+  }
+}
+
+nock.disableNetConnect()
+nock.enableNetConnect('localhost')
+
+let githubNock
+
+const githubRepository = {
+  id: 42,
+  full_name: 'finnp/abc',
+  private: false,
+  fork: false,
+  has_issues: true
+}
+
+test('reset repo', async t => {
+  const { repositories, installations } = await dbs()
+
+  t.beforeEach(async () => {
+    githubNock = nock('https://api.github.com')
+      .post('/installations/37/access_tokens')
+      .reply(200, {
+        token: 'secret'
+      })
+      .get('/rate_limit')
+      .reply(200, {})
+
+    await installations.put({
+      _id: '123',
+      installation: 37
+    })
+    await repositories.put({
+      _id: '42',
+      id: '42',
+      accountId: '123',
+      enabled: true,
+      fullName: 'finnp/abc',
+      type: 'repository'
+    })
+    await repositories.put({
+      _id: '42:pr:123',
+      repositoryId: '42',
+      type: 'pr'
+    })
+    await repositories.put({
+      _id: '42:branch:deadbeef',
+      type: 'branch',
+      repositoryId: '42',
+      head: 'greenkeeper-standard-10.0.0',
+      dependency: 'standard',
+      version: '10.0.0',
+      dependencyType: 'dependencies'
+    })
+  })
+
+  t.afterEach(async () => {
+    nock.cleanAll()
+    await removeIfExists(repositories, '42')
+    await installations.remove(await installations.get('123'))
+    await removeIfExists(repositories, '42:pr:123')
+    await removeIfExists(repositories, '42:branch:deadbeef')
+    await removeIfExists(repositories, '42:branch:deadbeef0')
+  })
+
+  t.test('response with error if repo cound not be found', async t => {
+    t.plan(1)
+    nock.cleanAll()
+    try {
+      await worker({
+        repositoryFullName: 'finnp/hello'
+      })
+    } catch (e) {
+      if (e.status !== 404) {
+        throw e
+      }
+      t.equal(e.message, 'The repository finnp/hello does not exist in the database', 'correct error thrown')
+    } finally {
+      await waitFor(timeToWaitAfterTests)
+    }
+  })
+
+  t.test('delete all prdocs of the repo', async t => {
+    t.plan(1)
+    githubNock
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper-standard-10.0.0')
+      .reply(200, {})
+      .get('/repos/finnp/abc')
+      .reply(200, githubRepository)
+    await worker({
+      repositoryFullName: 'finnp/abc'
+    })
+    try {
+      await repositories.get('42:pr:123')
+    } catch (e) {
+      if (e.status !== 404) {
+        throw e
+      }
+      t.ok(true, 'PrDocs successfully deleted')
+    } finally {
+      await waitFor(timeToWaitAfterTests)
+    }
+  })
+
+  t.test('delete all greenkeeper branches', async t => {
+    t.plan(2)
+
+    githubNock
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper-standard-10.0.0')
+      .reply(200, () => {
+        t.ok(true, 'deleted gk branch')
+      })
+      .get('/repos/finnp/abc')
+      .reply(200, githubRepository)
+
+    await worker({
+      repositoryFullName: 'finnp/abc'
+    })
+    t.ok(true, 'Worker ran successfully')
+    await waitFor(timeToWaitAfterTests)
+  })
+
+  t.test('do not mind if some dependency branch could not be deleted', async t => {
+    t.plan(1)
+
+    githubNock
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper-standard-10.0.0')
+      .reply(409)
+      .get('/repos/finnp/abc')
+      .reply(200, githubRepository)
+
+    await worker({
+      repositoryFullName: 'finnp/abc'
+    })
+    t.ok(true, 'Worker ran successfully')
+    await waitFor(timeToWaitAfterTests)
+  })
+
+  t.test('fail if initial branch could not be deleted', async t => {
+    t.plan(1)
+    await repositories.put({
+      _id: '42:branch:deadbeef0',
+      type: 'branch',
+      repositoryId: '42',
+      head: 'greenkeeper/initial'
+    })
+    githubNock
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper-standard-10.0.0')
+      .reply(200)
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper/initial')
+      .reply(409)
+
+    try {
+      await worker({
+        repositoryFullName: 'finnp/abc'
+      })
+    } catch (e) {
+      if (e.code !== 409) {
+        throw e
+      }
+      t.equal(e.code, 409, 'error thrown')
+    } finally {
+      await waitFor(timeToWaitAfterTests)
+    }
+  })
+
+  t.test('delete all branchdocs of the repo', async t => {
+    t.plan(1)
+    githubNock
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper-standard-10.0.0')
+      .reply(200, {})
+      .get('/repos/finnp/abc')
+      .reply(200, githubRepository)
+    await worker({
+      repositoryFullName: 'finnp/abc'
+    })
+    try {
+      await repositories.get('42:branch:deadbeef')
+    } catch (e) {
+      if (e.status !== 404) {
+        throw e
+      }
+      t.ok(true, 'BranchDocs successfully deleted')
+    } finally {
+      await waitFor(timeToWaitAfterTests)
+    }
+  })
+
+  t.test('delete the repodoc and create a fresh one', async t => {
+    t.plan(1)
+    githubNock
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper-standard-10.0.0')
+      .reply(200, {})
+      .get('/repos/finnp/abc')
+      .reply(200, githubRepository)
+    await worker({
+      repositoryFullName: 'finnp/abc'
+    })
+    const freshRepoDoc = await repositories.get('42')
+    t.false(freshRepoDoc.enabled, 'New RepoDoc saved in database')
+    await waitFor(timeToWaitAfterTests)
+  })
+
+  t.test('schedule create inital branch job', async t => {
+    t.plan(2)
+    githubNock
+      .delete('/repos/finnp/abc/git/refs/heads/greenkeeper-standard-10.0.0')
+      .reply(200, {})
+      .get('/repos/finnp/abc')
+      .reply(200, githubRepository)
+    const newJob = await worker({
+      repositoryFullName: 'finnp/abc'
+    })
+    const freshRepoDoc = await repositories.get('42')
+    t.equal(newJob.data.name, 'create-initial-branch', 'create-initial-branch Job enqueued')
+    t.equal(newJob.data.repositoryId, freshRepoDoc._id, 'Job has the correct repository id')
+    await waitFor(timeToWaitAfterTests)
+  })
+})