Added bulkdownload to allow for more tests to pass

Author: david-rocca

src/middleware/schemas/BulkDownloadOrg.json

@@ -1,11 +1,11 @@
 {
   "$schema": "http://json-schema.org/draft-07/schema#",
-  "$id": "https://cve.org/schemas/org/bulkdownload",
+  "$id": "BaseOrg",
   "type": "object",
   "title": "CVE Bulk Download Organization",
   "description": "Schema for a CVE Bulk Download Organization",
   "allOf": [
-    { "$ref": "/schemas/org/base" },
+    { "$ref": "BaseOrg" },
     {
       "properties": {
         "authority": {

src/model/adporg.js

@@ -24,6 +24,6 @@ ADPSchema.statics.validateOrg = function (record) {
   }
   return validateObject
 }
-const CNAOrg = BaseOrg.discriminator('ADPOrg', ADPSchema, options)
+const ADPOrg = BaseOrg.discriminator('ADPOrg', ADPSchema, options)
 
-module.exports = CNAOrg
+module.exports = ADPOrg

src/model/bulkdownloadorg.js

@@ -0,0 +1,29 @@
+const mongoose = require('mongoose')
+const BaseOrg = require('./baseorg')
+const fs = require('fs')
+const Ajv = require('ajv')
+const addFormats = require('ajv-formats')
+const BaseOrgSchema = JSON.parse(fs.readFileSync('src/middleware/schemas/BaseOrg.json'))
+const BulkDownloadOrgSchema = JSON.parse(fs.readFileSync('src/middleware/schemas/BulkDownloadOrg.json'))
+const ajv = new Ajv({ allErrors: false })
+addFormats(ajv)
+ajv.addSchema(BaseOrgSchema)
+
+const validate = ajv.compile(BulkDownloadOrgSchema)
+
+const schema = {}
+
+const options = { discriminatorKey: 'kind' }
+const BulkDownloadSchema = new mongoose.Schema(schema, options)
+BulkDownloadSchema.statics.validateOrg = function (record) {
+  const validateObject = {}
+  validateObject.isValid = validate(record)
+
+  if (!validateObject.isValid) {
+    validateObject.errors = validate.errors
+  }
+  return validateObject
+}
+const BulkDownloadOrg = BaseOrg.discriminator('BulkDownloadOrg', BulkDownloadSchema, options)
+
+module.exports = BulkDownloadOrg

src/repositories/baseOrgRepository.js

@@ -2,6 +2,7 @@ const BaseRepository = require('./baseRepository')
 const BaseOrgModel = require('../model/baseorg')
 const CNAOrgModel = require('../model/cnaorg')
 const ADPOrgModel = require('../model/adporg')
+const BulkDownloadModel = require('../model/bulkdownloadorg')
 const SecretariatOrgModel = require('../model/secretariatorg')
 const CveIdRepository = require('./cveIdRepository')
 const uuid = require('uuid')
@@ -222,6 +223,10 @@ class BaseOrgRepository extends BaseRepository {
       registryObjectRaw.hard_quota = 0
       const adpObjectToSave = new ADPOrgModel(registryObjectRaw)
       registryObject = await adpObjectToSave.save(options)
+    } else if (registryObjectRaw.authority.includes('BULK_DOWNLOAD')) {
+      registryObjectRaw.hard_quota = 0
+      const bulkDownloadObjectToSave = new BulkDownloadModel(registryObjectRaw)
+      registryObject = await bulkDownloadObjectToSave.save(options)
     } else {
       // eslint-disable-next-line no-throw-literal
       throw 'dave you screwed up'
@@ -237,8 +242,9 @@ class BaseOrgRepository extends BaseRepository {
       _.set(legacyObjectRaw, 'policies.id_quota', CONSTANTS.DEFAULT_ID_QUOTA)
     }
     if (
-      legacyObjectRaw.authority.active_roles.length === 1 &&
-      legacyObjectRaw.authority.active_roles[0] === 'ADP'
+      legacyObjectRaw.authority.active_roles.length === 1 && (
+        legacyObjectRaw.authority.active_roles[0] === 'ADP' ||
+      legacyObjectRaw.authority.active_roles[0] === 'BULK_DOWNLOAD')
     ) {
       // ADPs have quota of 0
       _.set(legacyObjectRaw, 'policies.id_quota', 0)