def put(self, mine_guid, variance_guid): parser = CustomReqparser() # Arguments required by MineDocument parser.add_argument('document_name', type=str, required=True) parser.add_argument('document_manager_guid', type=str, required=True) parser.add_argument('variance_document_category_code', type=str, required=True) variance = Variance.find_by_variance_guid(variance_guid) if not variance: raise NotFound('Unable to fetch variance.') data = parser.parse_args() document_name = data.get('document_name') document_manager_guid = data.get('document_manager_guid') # Register new file upload mine_doc = MineDocument(mine_guid=mine_guid, document_manager_guid=document_manager_guid, document_name=document_name) if not mine_doc: raise BadRequest('Unable to register uploaded file as document') # Associate Variance & MineDocument to create Variance Document # Add fields specific to Variance Documents mine_doc.save() variance_doc = VarianceDocumentXref( mine_document_guid=mine_doc.mine_document_guid, variance_id=variance.variance_id, variance_document_category_code=data.get('variance_document_category_code')) variance.documents.append(variance_doc) variance.save() return variance
def put(self, mine_guid, variance_guid): parser = CustomReqparser() # Arguments required by MineDocument parser.add_argument('document_name', type=str, required=True) parser.add_argument('document_manager_guid', type=str, required=True) variance = Variance.find_by_variance_guid(variance_guid) if not variance: raise NotFound('Unable to fetch variance.') data = parser.parse_args() document_name = data.get('document_name') document_manager_guid = data.get('document_manager_guid') # Register new file upload mine_doc = MineDocument(mine_guid=mine_guid, document_manager_guid=document_manager_guid, document_name=document_name) if not mine_doc: raise BadRequest('Unable to register uploaded file as document') variance.mine_documents.append(mine_doc) variance.save() return variance
def test_variance_model_find_by_variance_guid_invalid(db_session): with pytest.raises(AssertionError) as e: Variance.find_by_variance_guid('abc123') assert str(e.value) == INVALID_VARIANCE_GUID
def test_variance_model_find_by_variance_guid_fail(db_session): variance = Variance.find_by_variance_guid(uuid.uuid4()) assert variance is None
def test_variance_model_find_by_variance_guid(db_session): init_variance_guid = VarianceFactory().variance_guid variance = Variance.find_by_variance_guid(init_variance_guid) assert variance.variance_guid == init_variance_guid