def setUp(self): self.uow = MagicMock() self.repo = UnmappedVendorPointRecordRepository(self.uow)
class TestUnmappedVendorPointRecordRepository(unittest.TestCase): def setUp(self): self.uow = MagicMock() self.repo = UnmappedVendorPointRecordRepository(self.uow) def test_insert_insert_unmapped_vendor_point_records(self): records = [Mock(), Mock(), Mock()] table_mock = self.uow.tables.unmapped_vendor_point_records self.repo.insert_unmapped_vendor_point_records(records) # table.insert() table_mock.insert.assert_called_with(records) self.uow.run.assert_called_with(table_mock.insert.return_value) def test_get_all_vendor_points(self): table_mock = self.uow.tables.unmapped_vendor_point_records rv = self.repo.get_all_vendor_points() # table.pluck() table_mock.pluck.assert_called_with('vendor_point') return_value = table_mock.pluck.return_value # table.pluck().distinct() return_value.distinct.assert_called_with() return_value = return_value.distinct.return_value self.uow.run_list.assert_called_with(return_value) return_value = self.uow.run_list.return_value self.assertEqual(rv, return_value) def test_get_all_for_vendor_point_records_for_johnson(self): vendor_point = {'johnson_fqr': Mock(), 'johnson_site_id': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records rv = self.repo.get_all_vendor_point_records_for_johnson(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['johnson_fqr'], vendor_point['johnson_site_id']], index='johnson') return_value = table_mock.get_all.return_value self.uow.run_list.assert_called_with(return_value) return_value = self.uow.run_list.return_value self.assertEqual(rv, return_value) def test_get_all_vendor_point_records_for_fieldserver(self): vendor_point = {'fieldserver_offset': Mock(), 'fieldserver_site_id': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records rv = self.repo.get_all_vendor_point_records_for_fieldserver(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['fieldserver_offset'], vendor_point['fieldserver_site_id']], index='fieldserver') return_value = table_mock.get_all.return_value self.uow.run_list.assert_called_with(return_value) return_value = self.uow.run_list.return_value self.assertEqual(rv, return_value) def test_get_all_vendor_point_records_for_invensys(self): vendor_point = {'invensys_point_name': Mock(), 'invensys_equipment_name': Mock(), 'invensys_site_name': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records rv = self.repo.get_all_vendor_point_records_for_invensys(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['invensys_point_name'], vendor_point['invensys_equipment_name'], vendor_point['invensys_site_name']], index='invensys') return_value = table_mock.get_all.return_value self.uow.run_list.assert_called_with(return_value) return_value = self.uow.run_list.return_value self.assertEqual(rv, return_value) def test_get_all_vendor_point_records_for_siemens(self): vendor_point = {'siemens_meter_name': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records rv = self.repo.get_all_vendor_point_records_for_siemens(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['siemens_meter_name']], index='siemens') return_value = table_mock.get_all.return_value self.uow.run_list.assert_called_with(return_value) return_value = self.uow.run_list.return_value self.assertEqual(rv, return_value) def test_delete_all_for_johnson_point(self): vendor_point = {'johnson_fqr': Mock(), 'johnson_site_id': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records self.repo.delete_all_for_johnson_point(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['johnson_fqr'], vendor_point['johnson_site_id']], index='johnson') return_value = table_mock.get_all.return_value # table.get_all().delete() return_value.delete.assert_called_with() return_value = return_value.delete.return_value self.uow.run.assert_called_with(return_value) def test_delete_all_for_fieldserver_point(self): vendor_point = {'fieldserver_offset': Mock(), 'fieldserver_site_id': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records self.repo.delete_all_for_fieldserver_point(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['fieldserver_offset'], vendor_point['fieldserver_site_id']], index='fieldserver') return_value = table_mock.get_all.return_value # table.get_all().delete() return_value.delete.assert_called_with() return_value = return_value.delete.return_value self.uow.run.assert_called_with(return_value) def test_delete_all_for_invensys_point(self): vendor_point = {'invensys_point_name': Mock(), 'invensys_equipment_name': Mock(), 'invensys_site_name': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records self.repo.delete_all_for_invensys_point(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['invensys_point_name'], vendor_point['invensys_equipment_name'], vendor_point['invensys_site_name']], index='invensys') return_value = table_mock.get_all.return_value # table.get_all().delete() return_value.delete.assert_called_with() return_value = return_value.delete.return_value self.uow.run.assert_called_with(return_value) def test_delete_all_for_siemens_point(self): vendor_point = {'siemens_meter_name': Mock()} table_mock = self.uow.tables.unmapped_vendor_point_records self.repo.delete_all_for_siemens_point(vendor_point) # table.get_all() table_mock.get_all.assert_called_with([vendor_point['siemens_meter_name']], index='siemens') return_value = table_mock.get_all.return_value # table.get_all().delete() return_value.delete.assert_called_with() return_value = return_value.delete.return_value self.uow.run.assert_called_with(return_value)