def test_expected_documents_find_by_mine_guid_after_insert(db_session): mine = MineFactory(mine_expected_documents=BATCH_SIZE) mine_guid = mine.mine_guid org_expected_documents = MineExpectedDocument.find_by_mine_guid( str(mine_guid)) MineExpectedDocumentFactory(mine=mine) new_expected_documents = MineExpectedDocument.find_by_mine_guid( str(mine_guid)) assert len(new_expected_documents) == BATCH_SIZE + 1 assert all(mine_guid == ned.mine_guid for ned in new_expected_documents) assert all(mine_guid == oed.mine_guid for oed in org_expected_documents)
def setup_info(test_client): TEST_DOCUMENT_MANAGER_GUID1 = uuid.uuid4() TEST_DOCUMENT_MANAGER_GUID2 = uuid.uuid4() TEST_EXPECTED_DOCUMENT_GUID = uuid.uuid4() mine_document = MineDocument( mine_guid=TEST_MINE_GUID, document_manager_guid=TEST_DOCUMENT_MANAGER_GUID1, document_name='file.txt', **DUMMY_USER_KWARGS) mine_document.save() orphaned_mine_document = MineDocument( mine_guid=TEST_MINE_GUID, document_manager_guid=TEST_DOCUMENT_MANAGER_GUID2, document_name='file2.txt', **DUMMY_USER_KWARGS) orphaned_mine_document.save() expected_document = MineExpectedDocument( exp_document_guid=TEST_EXPECTED_DOCUMENT_GUID, req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID1), mine_guid=TEST_MINE_GUID, exp_document_name=TEST_EXPECTED_DOCUMENT_NAME1, due_date=datetime.strptime('1984-06-18', '%Y-%m-%d'), received_date=datetime.strptime('1984-06-18', '%Y-%m-%d'), exp_document_status_code=TEST_EXPECTED_DOCUMENT_STATUS_CODE1, **DUMMY_USER_KWARGS) expected_document.mine_documents.append(mine_document) expected_document.save() yield dict( file_upload_1=(io.BytesIO(b'Test File'), 'file1.docx'), file_upload_2=(io.BytesIO(b'Test File'), 'file2.pdf'), document_manager_guid1=str(TEST_DOCUMENT_MANAGER_GUID1), document_manager_guid2=str(TEST_DOCUMENT_MANAGER_GUID2), mine_document=mine_document, orphaned_mine_document=orphaned_mine_document, expected_document=expected_document, ) db.session.delete(mine_document) db.session.commit() db.session.delete(expected_document) db.session.commit()
def test_add_fiscal_due_date_when_current_date_is_fiscal(): fiscal = datetime(datetime.now().year, 3, 31, 00, 00, 00) due_date_type = 'FIS' period = '12' expected_due_date = fiscal + relativedelta(months=12) due_date = MineExpectedDocument._get_due_date_for_expected_document( None, fiscal, due_date_type, period) assert due_date == expected_due_date
def test_add_aniversary_due_date(): current_date = datetime.now() expected_due_date = current_date due_date_type = 'ANV' period = '12' due_date = MineExpectedDocument._get_due_date_for_expected_document( None, current_date, due_date_type, period) assert expected_due_date == due_date
def test_add_fiscal_due_date_with_five_year_period_this_year(): current_date = datetime(datetime.now().year, 2, 1, 00, 00, 00) expected_due_date = datetime(current_date.year + 4, 3, 31, 00, 00, 00) due_date_type = 'FIS' period = '60' due_date = MineExpectedDocument._get_due_date_for_expected_document( None, current_date, due_date_type, period) assert due_date == expected_due_date
def test_add_fiscal_due_date_on_new_year(): new_year = datetime(datetime.now().year, 1, 1, 00, 00, 00) due_date_type = 'FIS' period = '12' expected_due_date = datetime(datetime.now().year, 3, 31, 00, 00, 00) due_date = MineExpectedDocument._get_due_date_for_expected_document( None, new_year, due_date_type, period) assert due_date == expected_due_date
def test_add_fiscal_due_date_on_year_end(): end_of_year = datetime(datetime.now().year - 1, 12, 31, 23, 59, 59) due_date_type = 'FIS' period = '12' expected_due_date = datetime(datetime.now().year, 3, 31, 00, 00, 00) due_date = MineExpectedDocument._get_due_date_for_expected_document( None, end_of_year, due_date_type, period) assert due_date == expected_due_date
def test_add_fiscal_due_date_with_one_year_period(test_client): current_date = datetime(datetime.now().year, 7, 1, 00, 00, 00) expected_due_date = datetime(current_date.year + 1, 3, 31, 00, 00, 00) due_date_type = 'FIS' period = '12' due_date = MineExpectedDocument._get_due_date_for_expected_document( None, current_date, due_date_type, period) assert due_date == expected_due_date
def test_expected_documents_find_by_mine_guid_after_insert(test_client, auth_headers): org_expected_documents = MineExpectedDocument.find_by_mine_guid(TEST_MINE_GUID) expected_document2 = MineExpectedDocument( exp_document_guid=uuid.UUID(TEST_EXPECTED_DOCUMENT_GUID2), mine_guid=uuid.UUID(TEST_MINE_GUID), exp_document_name=TEST_EXPECTED_DOCUMENT_NAME2, exp_document_status_code=TEST_EXPECTED_DOCUMENT_STATUS_CODE1, **DUMMY_USER_KWARGS) expected_document2.save() new_expected_documents = MineExpectedDocument.find_by_mine_guid(TEST_MINE_GUID) assert len(new_expected_documents) == (len(org_expected_documents) + 1) assert all(str(ned.mine_guid) == TEST_MINE_GUID for ned in new_expected_documents) assert all(str(oed.mine_guid) == TEST_MINE_GUID for oed in org_expected_documents)
def test_expected_documents_find_by_mine_guid(test_client, auth_headers): expected_documents = MineExpectedDocument.find_by_mine_guid(TEST_MINE_GUID) assert all( str(ed.mine_guid) == TEST_MINE_GUID for ed in expected_documents)
def setup_data(session): # Clear data clear_data(session) # Insert Region Code for region_code_value, display_order_value in zip( TEST_REGION_CODES, TEST_REGION_CODE_DISPLAY_ORDER): region_code = MineRegionCode(mine_region_code=region_code_value, description=TEST_REGION_DESCRIPTION, display_order=display_order_value, **DUMMY_USER_KWARGS) region_code.save() # Insert Mine Tenure Types for code, description in zip(TEST_MINE_TENURE_TYPE_CODES, TEST_MINE_TENURE_TYPE_DESCRIPTIONS): mine_tenure_type_code = MineTenureTypeCode(mine_tenure_type_code=code, description=description, **DUMMY_USER_KWARGS) mine_tenure_type_code.save() # Insert Mine Disturbance Codes for code, description in zip(TEST_MINE_DISTURBANCE_CODES, TEST_MINE_DISTURBANCE_DESCRIPTIONS): mine_disturbance_code = MineDisturbanceCode(mine_disturbance_code=code, description=description, **DUMMY_USER_KWARGS) mine_disturbance_code.save() # Insert Mine Commodity Codes for code, description in zip(TEST_MINE_COMMODITY_CODES, TEST_MINE_COMMODITY_DESCRIPTIONS): mine_commodity_code = MineCommodityCode(mine_commodity_code=code, description=description, **DUMMY_USER_KWARGS) mine_commodity_code.save() # Test Mine Data mine = Mine(mine_guid=uuid.UUID(TEST_MINE_GUID), mine_no=TEST_MINE_NO, mine_name=TEST_MINE_NAME, mine_region=TEST_REGION_CODE, **DUMMY_USER_KWARGS) mine.save() # Test Mine Type mine_type = MineType(mine_type_guid=uuid.UUID(TEST_MINE_TYPE_GUID), mine_guid=uuid.UUID(TEST_MINE_GUID), mine_tenure_type_code=TEST_MINE_TENURE_TYPE_CODES[0], active_ind=True, **DUMMY_USER_KWARGS) mine_type.save() # Test Mine Type Detail mine_type_detail = MineTypeDetail( mine_type_detail_xref_guid=uuid.UUID(TEST_MINE_TYPE_DETAIL_GUID), mine_type_guid=uuid.UUID(TEST_MINE_TYPE_GUID), mine_disturbance_code=TEST_MINE_DISTURBANCE_CODES[0], active_ind=True, **DUMMY_USER_KWARGS) mine_type_detail.save() # Test Tenure Data tenure = MineralTenureXref( mineral_tenure_xref_guid=uuid.UUID(TEST_TENURE_GUID), mine_guid=uuid.UUID(TEST_MINE_GUID), tenure_number_id=TEST_TENURE_ID, **DUMMY_USER_KWARGS) tenure.save() # Test Location Data mine_location = MineLocation( mine_location_guid=uuid.UUID(TEST_LOCATION_GUID), mine_guid=uuid.UUID(TEST_MINE_GUID), latitude=TEST_LAT_1, longitude=TEST_LONG_1, effective_date=datetime.today(), expiry_date=datetime.today(), **DUMMY_USER_KWARGS) mine_location.save() # Test Person Type Codes for k, v in PARTY_STATUS_CODE.items(): party_code = PartyTypeCode(party_type_code=v, description=v, **DUMMY_USER_KWARGS) party_code.save() session.commit() # Test Party Region Codes for code, description in zip(TEST_SUB_DIVISION_CODES, TEST_SUB_DIVISION_CODE_DESCRIPTIONS): sub_division_code = SubDivisionCode(sub_division_code=code, description=description, **DUMMY_USER_KWARGS) sub_division_code.save() # Test Operation Codes for k, v in MINE_OPERATION_STATUS.items(): mine_operation_status_code = MineOperationStatusCode( mine_operation_status_code=v['value'], description=v['label'], **DUMMY_USER_KWARGS) mine_operation_status_code.save() for k, v in MINE_OPERATION_STATUS_REASON.items(): mine_operation_status_reason_code = MineOperationStatusReasonCode( mine_operation_status_reason_code=v['value'], description=v['label'], **DUMMY_USER_KWARGS) mine_operation_status_reason_code.save() for k, v in MINE_OPERATION_STATUS_SUB_REASON.items(): mine_operation_status_sub_reason_code = MineOperationStatusSubReasonCode( mine_operation_status_sub_reason_code=v['value'], description=v['label'], **DUMMY_USER_KWARGS) mine_operation_status_sub_reason_code.save() session.commit() # Insert Operation Code Xref for status_k, status_v in MINE_OPERATION_STATUS.items(): for reason_k, reason_v in MINE_OPERATION_STATUS_REASON.items(): for sub_k, sub_v in MINE_OPERATION_STATUS_SUB_REASON.items(): mine_status_xref = MineStatusXref( mine_status_xref_guid=uuid.uuid4(), mine_operation_status_code=status_v['value'], mine_operation_status_reason_code=reason_v['value'], mine_operation_status_sub_reason_code=sub_v['value'], **DUMMY_USER_KWARGS) mine_status_xref.save() # Test Person Data person = Party(party_guid=uuid.UUID(TEST_PARTY_PER_GUID_1), first_name=TEST_PARTY_PER_FIRST_NAME_1, party_name=TEST_PARTY_PER_PARTY_NAME_1, email=TEST_PARTY_PER_EMAIL_1, phone_no=TEST_PARTY_PER_PHONE_1, phone_ext=TEST_PARTY_PER_PHONE_EXT_1, party_type_code=TEST_PARTY_TYPE, **DUMMY_USER_KWARGS) person.save(commit=False) person2 = Party(party_guid=uuid.UUID(TEST_PARTY_PER_GUID_2), first_name=TEST_PARTY_PER_FIRST_NAME_2, party_name=TEST_PARTY_PER_PARTY_NAME_2, email=TEST_PARTY_PER_EMAIL_2, phone_no=TEST_PARTY_PER_PHONE_2, phone_ext=TEST_PARTY_PER_PHONE_EXT_2, party_type_code=TEST_PARTY_TYPE, **DUMMY_USER_KWARGS) person2.save(commit=False) person3 = Party(party_guid=uuid.UUID(TEST_PARTY_PER_GUID_3), first_name=TEST_PARTY_PER_FIRST_NAME_3, party_name=TEST_PARTY_PER_PARTY_NAME_3, email=TEST_PARTY_PER_EMAIL_3, phone_no=TEST_PARTY_PER_PHONE_3, phone_ext=TEST_PARTY_PER_PHONE_EXT_3, party_type_code=TEST_PARTY_TYPE, **DUMMY_USER_KWARGS) person3.save(commit=False) party_org = Party(party_guid=uuid.UUID(TEST_PARTY_ORG_GUID), party_name=TEST_PARTY_ORG_NAME, email=TEST_PARTY_ORG_EMAIL, phone_no=TEST_PARTY_ORG_PHONE, phone_ext=TEST_PARTY_ORG_EXT, party_type_code=TEST_ORG_TYPE, **DUMMY_USER_KWARGS) party_org.save() # Test Permit Status Codes for permit_code_value in TEST_PERMIT_STATUS_CODES: permit_code = PermitStatusCode( permit_status_code=permit_code_value, description=TEST_PERMIT_STATUS_CODE_NAME_1, **DUMMY_USER_KWARGS) permit_code.save() # Test Permit Data permit = Permit(permit_guid=TEST_PERMIT_GUID_1, mine_guid=TEST_MINE_GUID, permit_no=TEST_PERMIT_NO_1, permit_status_code=TEST_PERMIT_STATUS_CODE_1, **DUMMY_USER_KWARGS) permit.save() permit_amendment_status_code = PermitAmendmentStatusCode( permit_amendment_status_code=TEST_PERMIT_AMENDMENT_STATUS_CODE, description=TEST_PERMIT_AMENDMENT_STATUS_CODE_NAME, **DUMMY_USER_KWARGS) permit_amendment_status_code.save() permit_amendment_status_code2 = PermitAmendmentStatusCode( permit_amendment_status_code=TEST_PERMIT_AMENDMENT_STATUS_CODE_2, description=TEST_PERMIT_AMENDMENT_STATUS_CODE_NAME, **DUMMY_USER_KWARGS) permit_amendment_status_code2.save() permit_amendment_type_code = PermitAmendmentTypeCode( permit_amendment_type_code=TEST_PERMIT_AMENDMENT_TYPE_CODE, description=TEST_PERMIT_AMENDMENT_TYPE_CODE_NAME, **DUMMY_USER_KWARGS) permit_amendment_type_code.save() permit_amendment_type_code2 = PermitAmendmentTypeCode( permit_amendment_type_code=TEST_PERMIT_AMENDMENT_TYPE_CODE_2, description=TEST_PERMIT_AMENDMENT_TYPE_CODE_NAME, **DUMMY_USER_KWARGS) permit_amendment_type_code2.save() required_document_due_date_type1 = RequiredDocumentDueDateType( req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[0], req_document_due_date_description= TEST_REQUIRED_REPORT_DUE_DATE_DESCRIPTION[0], **DUMMY_USER_KWARGS) required_document_due_date_type1.save() required_document_due_date_type2 = RequiredDocumentDueDateType( req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[1], req_document_due_date_description= TEST_REQUIRED_REPORT_DUE_DATE_DESCRIPTION[1], **DUMMY_USER_KWARGS) required_document_due_date_type2.save() required_document_category1 = RequiredDocumentCategory( req_document_category=TEST_REQUIRED_REPORT_CATEGORY_TAILINGS) required_document_category1.save() required_document_category2 = RequiredDocumentCategory( req_document_category='OTH') required_document_category2.save() required_document1 = RequiredDocument( req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID1), req_document_name=TEST_REQUIRED_REPORT_NAME1, req_document_category=required_document_category1. req_document_category, req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[0], req_document_due_date_period_months=12, **DUMMY_USER_KWARGS) required_document1.save() required_document_sub_category = RequiredDocumentSubCategory( req_document_sub_category_code=TEST_REQUIRED_REPORT_SUB_CATEGORY_1) required_document_sub_category.save() required_document2 = RequiredDocument( req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID2), req_document_name=TEST_REQUIRED_REPORT_NAME2, req_document_category=required_document_category1. req_document_category, req_document_sub_category_code=required_document_sub_category. req_document_sub_category_code, req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[0], req_document_due_date_period_months=12, **DUMMY_USER_KWARGS) required_document2.save() required_document3 = RequiredDocument( req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID3), req_document_name=TEST_REQUIRED_REPORT_NAME3, req_document_category=required_document_category2. req_document_category, req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[1], req_document_due_date_period_months=12, **DUMMY_USER_KWARGS) required_document3.save() expected_document_status1 = ExpectedDocumentStatus( exp_document_status_code=TEST_EXPECTED_DOCUMENT_STATUS_CODE1, description="Not Received", display_order=10, **DUMMY_USER_KWARGS) expected_document_status1.save() expected_document_status2 = ExpectedDocumentStatus( exp_document_status_code=TEST_EXPECTED_DOCUMENT_STATUS_CODE2, description="Pending Review", display_order=20, **DUMMY_USER_KWARGS) expected_document_status2.save() expected_document1 = MineExpectedDocument( exp_document_guid=uuid.UUID(TEST_EXPECTED_DOCUMENT_GUID1), req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID1), mine_guid=TEST_MINE_GUID, exp_document_name=TEST_EXPECTED_DOCUMENT_NAME1, due_date=datetime.strptime('1984-06-18', '%Y-%m-%d'), received_date=datetime.strptime('1984-06-18', '%Y-%m-%d'), exp_document_status_code=expected_document_status1. exp_document_status_code, **DUMMY_USER_KWARGS) expected_document1.save() mine_tsf1 = MineTailingsStorageFacility( mine_tailings_storage_facility_guid= TEST_TAILINGS_STORAGE_FACILITY_GUID1, mine_guid=TEST_MINE_GUID, mine_tailings_storage_facility_name= TEST_TAILINGS_STORAGE_FACILITY_NAME1, **DUMMY_USER_KWARGS) mine_tsf1.save() mpat1 = MinePartyAppointmentType( mine_party_appt_type_code=TEST_MINE_PARTY_APPT_TYPE_CODE1, description=TEST_MINE_PARTY_APPT_TYPE_DESCRIPTION1, grouping_level=2, **DUMMY_USER_KWARGS) mpat1.save() mpat2 = MinePartyAppointmentType( mine_party_appt_type_code=TEST_MINE_PARTY_APPT_TYPE_CODE2, description=TEST_MINE_PARTY_APPT_TYPE_DESCRIPTION2, grouping_level=2, **DUMMY_USER_KWARGS) mpat2.save() mpat3 = MinePartyAppointmentType(mine_party_appt_type_code='EOR', description='Engineer of Record', grouping_level=1, **DUMMY_USER_KWARGS) mpat3.save() mpat4 = MinePartyAppointmentType(mine_party_appt_type_code='PMT', description='Engineer of Record', grouping_level=1, **DUMMY_USER_KWARGS) mpat4.save() mpat5 = MinePartyAppointmentType(mine_party_appt_type_code='MMG', description='Mine Manager', grouping_level=1, **DUMMY_USER_KWARGS) mpat5.save() mine_doc1 = MineDocument( mine_guid=uuid.UUID(TEST_MINE_GUID), document_name=TEST_MINE_DOCUMENT_NAME1, document_manager_guid=TEST_DOCUMENT_MANAGER_FILE_GUID, **DUMMY_USER_KWARGS) mine_doc1.mine_expected_document.append(expected_document1) mine_doc1.save() application_status_code_1 = ApplicationStatusCode( application_status_code='RIP', description='In Review', display_order=10, **DUMMY_USER_KWARGS) application_status_code_1.save() application_status_code_2 = ApplicationStatusCode( application_status_code='APR', description='Approved', display_order=20, **DUMMY_USER_KWARGS) application_status_code_2.save()
def setup_data(session): # Clear data clear_data(session) # Insert Region Code for region_code_value, display_order_value in zip( TEST_REGION_CODES, TEST_REGION_CODE_DISPLAY_ORDER): region_code = MineRegionCode(mine_region_code=region_code_value, description=TEST_REGION_DESCRIPTION, display_order=display_order_value, **DUMMY_USER_KWARGS) region_code.save() # Insert Mine Tenure Types for code, description in zip(TEST_MINE_TENURE_TYPE_CODES, TEST_MINE_TENURE_TYPE_DESCRIPTIONS): mine_tenure_type_code = MineTenureTypeCode(mine_tenure_type_code=code, description=description, **DUMMY_USER_KWARGS) mine_tenure_type_code.save() # Insert Mine Disturbance Codes for code, description in zip(TEST_MINE_DISTURBANCE_CODES, TEST_MINE_DISTURBANCE_DESCRIPTIONS): mine_disturbance_code = MineDisturbanceCode(mine_disturbance_code=code, description=description, **DUMMY_USER_KWARGS) mine_disturbance_code.save() # Insert Mine Commodity Codes for code, description in zip(TEST_MINE_COMMODITY_CODES, TEST_MINE_COMMODITY_DESCRIPTIONS): mine_commodity_code = MineCommodityCode(mine_commodity_code=code, description=description, **DUMMY_USER_KWARGS) mine_commodity_code.save() # Test Mine Data mine_identity = MineIdentity(mine_guid=uuid.UUID(TEST_MINE_GUID), **DUMMY_USER_KWARGS) mine_detail = MineDetail(mine_detail_guid=uuid.UUID(TEST_MINE_DETAIL_GUID), mine_guid=uuid.UUID(TEST_MINE_GUID), mine_no=TEST_MINE_NO, mine_name=TEST_MINE_NAME, mine_region=TEST_REGION_CODE, **DUMMY_USER_KWARGS) mine_identity.save() mine_detail.save() # Test Mine Type mine_type = MineType(mine_type_guid=uuid.UUID(TEST_MINE_TYPE_GUID), mine_guid=uuid.UUID(TEST_MINE_GUID), mine_tenure_type_code=TEST_MINE_TENURE_TYPE_CODES[0], active_ind=True, **DUMMY_USER_KWARGS) mine_type.save() # Test Mine Type Detail mine_type_detail = MineTypeDetail( mine_type_detail_xref_guid=uuid.UUID(TEST_MINE_TYPE_DETAIL_GUID), mine_type_guid=uuid.UUID(TEST_MINE_TYPE_GUID), mine_disturbance_code=TEST_MINE_DISTURBANCE_CODES[0], active_ind=True, **DUMMY_USER_KWARGS) mine_type_detail.save() # Test Tenure Data tenure = MineralTenureXref( mineral_tenure_xref_guid=uuid.UUID(TEST_TENURE_GUID), mine_guid=uuid.UUID(TEST_MINE_GUID), tenure_number_id=TEST_TENURE_ID, **DUMMY_USER_KWARGS) tenure.save() # Test Location Data mine_location = MineLocation( mine_location_guid=uuid.UUID(TEST_LOCATION_GUID), mine_guid=uuid.UUID(TEST_MINE_GUID), latitude=TEST_LAT_1, longitude=TEST_LONG_1, effective_date=datetime.today(), expiry_date=datetime.today(), **DUMMY_USER_KWARGS) mine_location.save() # Test Person Type Codes for k, v in PARTY_STATUS_CODE.items(): party_code = PartyTypeCode(party_type_code=v, description=v, **DUMMY_USER_KWARGS) party_code.save() session.commit() # Test Operation Codes for k, v in MINE_OPERATION_STATUS.items(): mine_operation_status_code = MineOperationStatusCode( mine_operation_status_code=v['value'], description=v['label'], **DUMMY_USER_KWARGS) mine_operation_status_code.save() for k, v in MINE_OPERATION_STATUS_REASON.items(): mine_operation_status_reason_code = MineOperationStatusReasonCode( mine_operation_status_reason_code=v['value'], description=v['label'], **DUMMY_USER_KWARGS) mine_operation_status_reason_code.save() for k, v in MINE_OPERATION_STATUS_SUB_REASON.items(): mine_operation_status_sub_reason_code = MineOperationStatusSubReasonCode( mine_operation_status_sub_reason_code=v['value'], description=v['label'], **DUMMY_USER_KWARGS) mine_operation_status_sub_reason_code.save() session.commit() # Test Person Data person = Party(party_guid=uuid.UUID(TEST_PARTY_PER_GUID_1), first_name=TEST_PARTY_PER_FIRST_NAME_1, party_name=TEST_PARTY_PER_PARTY_NAME_1, email=TEST_PARTY_PER_EMAIL_1, phone_no=TEST_PARTY_PER_PHONE_1, phone_ext=TEST_PARTY_PER_PHONE_EXT_1, party_type_code=TEST_PARTY_TYPE, **DUMMY_USER_KWARGS) person.save(commit=False) person2 = Party(party_guid=uuid.UUID(TEST_PARTY_PER_GUID_2), first_name=TEST_PARTY_PER_FIRST_NAME_2, party_name=TEST_PARTY_PER_PARTY_NAME_2, email=TEST_PARTY_PER_EMAIL_2, phone_no=TEST_PARTY_PER_PHONE_2, phone_ext=TEST_PARTY_PER_PHONE_EXT_2, party_type_code=TEST_PARTY_TYPE, **DUMMY_USER_KWARGS) person2.save(commit=False) person3 = Party(party_guid=uuid.UUID(TEST_PARTY_PER_GUID_3), first_name=TEST_PARTY_PER_FIRST_NAME_3, party_name=TEST_PARTY_PER_PARTY_NAME_3, email=TEST_PARTY_PER_EMAIL_3, phone_no=TEST_PARTY_PER_PHONE_3, phone_ext=TEST_PARTY_PER_PHONE_EXT_3, party_type_code=TEST_PARTY_TYPE, **DUMMY_USER_KWARGS) person3.save(commit=False) party_org = Party(party_guid=uuid.UUID(TEST_PARTY_ORG_GUID), party_name=TEST_PARTY_ORG_NAME, email=TEST_PARTY_ORG_EMAIL, phone_no=TEST_PARTY_ORG_PHONE, phone_ext=TEST_PARTY_ORG_EXT, party_type_code=TEST_ORG_TYPE, **DUMMY_USER_KWARGS) party_org.save() # Test Manager Data manager = MgrAppointment(mgr_appointment_guid=uuid.UUID(TEST_MANAGER_GUID), party_guid=uuid.UUID(TEST_PARTY_PER_GUID_1), mine_guid=uuid.UUID(TEST_MINE_GUID), effective_date=datetime.today() - timedelta(days=10), **DUMMY_USER_KWARGS) manager.save() # Test Permit Status Codes for permit_code_value in TEST_PERMIT_STATUS_CODES: permit_code = PermitStatusCode( permit_status_code=permit_code_value, description=TEST_PERMIT_STATUS_CODE_NAME_1, **DUMMY_USER_KWARGS) permit_code.save() # Test Permit Data permit = Permit(permit_guid=TEST_PERMIT_GUID_1, mine_guid=TEST_MINE_GUID, permit_no=TEST_PERMIT_NO_1, permit_status_code=TEST_PERMIT_STATUS_CODE_1, received_date=datetime.today(), issue_date=datetime.today(), **DUMMY_USER_KWARGS) permit.save() # Test Permittee Data permittee = Permittee(permittee_guid=uuid.UUID(TEST_PERMITTEE_GUID), permit_guid=uuid.UUID(TEST_PERMIT_GUID_1), party_guid=uuid.UUID(TEST_PARTY_PER_GUID_1), **DUMMY_USER_KWARGS) permittee.save() required_document_due_date_type1 = RequiredDocumentDueDateType( req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[0], req_document_due_date_description= TEST_REQUIRED_REPORT_DUE_DATE_DESCRIPTION[0], **DUMMY_USER_KWARGS) required_document_due_date_type1.save() required_document_due_date_type2 = RequiredDocumentDueDateType( req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[1], req_document_due_date_description= TEST_REQUIRED_REPORT_DUE_DATE_DESCRIPTION[1], **DUMMY_USER_KWARGS) required_document_due_date_type2.save() required_document_category1 = RequiredDocumentCategory( req_document_category_guid=TEST_REQUIRED_REPORT_CATEGORY_TAILINGS_GUID, req_document_category=TEST_REQUIRED_REPORT_CATEGORY_TAILINGS) required_document_category1.save() required_document_category2 = RequiredDocumentCategory( req_document_category_guid=TEST_REQUIRED_REPORT_CATEGORY_OTHER_GUID, req_document_category=TEST_REQUIRED_REPORT_CATEGORY_OTHER) required_document_category2.save() required_document1 = RequiredDocument( req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID1), req_document_name=TEST_REQUIRED_REPORT_NAME1, req_document_category_guid=TEST_REQUIRED_REPORT_CATEGORY_TAILINGS_GUID, req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[0], req_document_due_date_period_months=12, **DUMMY_USER_KWARGS) required_document1.save() required_document2 = RequiredDocument( req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID2), req_document_name=TEST_REQUIRED_REPORT_NAME2, req_document_category_guid=TEST_REQUIRED_REPORT_CATEGORY_TAILINGS_GUID, req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[0], req_document_due_date_period_months=12, **DUMMY_USER_KWARGS) required_document2.save() required_document3 = RequiredDocument( req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID3), req_document_name=TEST_REQUIRED_REPORT_NAME3, req_document_category_guid=TEST_REQUIRED_REPORT_CATEGORY_OTHER_GUID, req_document_due_date_type=TEST_REQUIRED_REPORT_DUE_DATE_TYPE[1], req_document_due_date_period_months=12, **DUMMY_USER_KWARGS) required_document3.save() expected_document1 = MineExpectedDocument( exp_document_guid=uuid.UUID(TEST_EXPECTED_DOCUMENT_GUID1), req_document_guid=uuid.UUID(TEST_REQUIRED_REPORT_GUID1), mine_guid=TEST_MINE_GUID, exp_document_name=TEST_EXPECTED_DOCUMENT_NAME1, due_date=datetime.strptime('1984-06-18', '%Y-%m-%d'), received_date=datetime.strptime('1984-06-18', '%Y-%m-%d'), **DUMMY_USER_KWARGS) expected_document1.save() mine_tsf1 = MineTailingsStorageFacility( mine_tailings_storage_facility_guid= TEST_TAILINGS_STORAGE_FACILITY_GUID1, mine_guid=TEST_MINE_GUID, mine_tailings_storage_facility_name= TEST_TAILINGS_STORAGE_FACILITY_NAME1, **DUMMY_USER_KWARGS) mine_tsf1.save() mpat1 = MinePartyAppointmentType( mine_party_appt_type_code=TEST_MINE_PARTY_APPT_TYPE_CODE1, description=TEST_MINE_PARTY_APPT_TYPE_DESCRIPTION1, **DUMMY_USER_KWARGS) mpat1.save() mpat2 = MinePartyAppointmentType( mine_party_appt_type_code=TEST_MINE_PARTY_APPT_TYPE_CODE2, description=TEST_MINE_PARTY_APPT_TYPE_DESCRIPTION2, **DUMMY_USER_KWARGS) mpat2.save() mpat3 = MinePartyAppointmentType(mine_party_appt_type_code='EOR', description='Engineer of Record', **DUMMY_USER_KWARGS) mpat3.save() mpa = MinePartyAppointment( mine_party_appt_guid=TEST_MINE_PARTY_APPT_GUID, mine_guid=uuid.UUID(TEST_MINE_GUID), party_guid=uuid.UUID(TEST_PARTY_PER_GUID_1), mine_party_appt_type_code=TEST_MINE_PARTY_APPT_TYPE_CODE1, **DUMMY_USER_KWARGS) mpa.save() mine_doc1 = MineDocument( mine_guid=uuid.UUID(TEST_MINE_GUID), document_name=TEST_MINE_DOCUMENT_NAME1, document_manager_guid=TEST_DOCUMENT_MANAGER_FILE_GUID, **DUMMY_USER_KWARGS) mine_doc1.mine_expected_document.append(expected_document1) mine_doc1.save()
def test_expected_documents_find_by_exp_document_guid(db_session): exp_doc = MineExpectedDocumentFactory.create_batch(size=BATCH_SIZE)[0] expected_document = MineExpectedDocument.find_by_exp_document_guid( str(exp_doc.exp_document_guid)) assert expected_document.exp_document_guid == exp_doc.exp_document_guid
def test_expected_documents_find_by_mine_guid(db_session): mine_guid = MineFactory(mine_expected_documents=BATCH_SIZE).mine_guid expected_documents = MineExpectedDocument.find_by_mine_guid(str(mine_guid)) assert len(expected_documents) == BATCH_SIZE assert all(mine_guid == ed.mine_guid for ed in expected_documents)
def test_expected_documents_find_by_exp_document_guid(test_client): expected_document = MineExpectedDocument.find_by_exp_document_guid( TEST_EXPECTED_DOCUMENT_GUID1) assert str( expected_document.exp_document_guid) == TEST_EXPECTED_DOCUMENT_GUID1