Python UnitTestGeneratorの例

コード例 #1

ファイル: test_data_acquisition_management_service.py プロジェクト: Bobfrat/coi-services

        with self.assertRaises(NotFound) as cm:
            self.data_acquisition_mgmt_service.register_external_data_set('bad')

        ex = cm.exception
        self.assertEqual(ex.message, 'External Data Set bad does not exist')
        self.mock_read.assert_called_once_with('bad', '')

#    def test_register_instrument(self):
#        self.mock_read.return_value = self.instrument
#        self.mock_create.return_value = ('111', 'bla')
#        self.mock_create.return_value = ('222', 'bla')
#        self.mock_create_association.return_value = ['333', 1]
#
#        self.data_acquisition_mgmt_service.register_instrument('111')



utg = UnitTestGenerator(TestDataAcquisitionManagement,
                        DataAcquisitionManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.ExternalDataProvider, "external_data_provider", {})
utg.add_resource_unittests(RT.DataSource, "data_source", {})
utg.add_resource_unittests(RT.DataSourceModel, "data_source_model", {})
utg.add_resource_unittests(RT.DataSourceAgent, "data_source_agent", {})
utg.add_resource_unittests(RT.DataSourceAgentInstance, "data_source_agent_instance", {})
utg.add_resource_unittests(RT.ExternalDataset, "external_dataset", {})
utg.add_resource_unittests(RT.ExternalDatasetAgent, "external_dataset_agent", {})
utg.add_resource_unittests(RT.ExternalDatasetAgentInstance, "external_dataset_agent_instance", {})

コード例 #2

ファイル: test_data_process_management_service.py プロジェクト: mbarry02/coi-services

    @unittest.skip("not working")
    def test_read_data_process(self):
        # setup
        self.resource_registry.find_associations.return_value = ([self.transform_id], "don't care")
        self.transform_object.data_process_definition_id = self.data_proc_def_id
        self.transform_object.in_subscription_id = self.in_product_A
        self.transform_object.out_data_product_id = self.out_product_A
        self.transform_management_service.read_transform.return_value = self.transform_object

        # test call
        dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process(self.data_process_id)

        # verify results
        self.assertEqual(dpd_id, self.data_proc_def_id)
        self.assertEqual(in_id, self.in_product_A)
        self.assertEqual(out_id, self.out_product_A)
        self.resource_registry.find_associations.assert_called_once_with(
            self.data_process_id, PRED.hasTransform, "", None, False
        )
        self.transform_management_service.read_transform.assert_called_once_with(self.transform_id)


utg = UnitTestGenerator(Test_DataProcessManagementService_Unit, DataProcessManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.TransformFunction, "transform_function", {})
utg.add_resource_unittests(RT.DataProcessDefinition, "data_process_definition", {})
# utg.add_resource_unittests(RT.DataProcess, "data_process", {})
# utg.add_resource_unittests(RT.DataProcess, "data_process2", {})

コード例 #3

ファイル: test_data_product_management_service.py プロジェクト: sfoley/coi-services

        dp_obj = IonObject(RT.DataProduct,
                           name='DP_X',
                           description='some existing dp')
        self.clients.resource_registry.find_resources.return_value = ([dp_obj],
                                                                      [])

        # test call
        result = self.data_product_management_service.find_data_products()

        # check results
        self.assertEqual(result, [dp_obj])
        self.clients.resource_registry.find_resources.assert_called_once_with(
            RT.DataProduct, None, None, False)


utg = UnitTestGenerator(TestDataProductManagementServiceUnit,
                        DataProductManagementService)

#utg.test_all_in_one(True)

utg.add_resource_unittests(RT.DataProduct, "data_product", {})
utg.add_resource_unittests(RT.DataProductCollection, "data_product_collection",
                           {})

#remove some tests that don't work
delattr(TestDataProductManagementServiceUnit,
        "test_data_product_create_d14a028")
delattr(TestDataProductManagementServiceUnit,
        "test_data_product_collection_create_d14a028")
delattr(TestDataProductManagementServiceUnit,
        "test_data_product_collection_create_bad_noname_d14a028")
delattr(TestDataProductManagementServiceUnit,

コード例 #4

ファイル: test_instrument_management_service.py プロジェクト: Bobfrat/coi-services

        self.instrument_mgmt_service = InstrumentManagementService()
        self.instrument_mgmt_service.clients = mock_clients
        
        # must call this manually
        self.instrument_mgmt_service.on_init()

        self.addCleanup(delattr, self, "instrument_mgmt_service")
        self.addCleanup(delattr, self, "mock_ionobj")
        #self.resource_impl_cleanup()

        log.debug("setUp complete")

    #def resource_impl_cleanup(self):
        #pass


utg = UnitTestGenerator(TestInstrumentManagement,
                        InstrumentManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.InstrumentAgentInstance, "instrument_agent_instance", {})
utg.add_resource_unittests(RT.InstrumentAgent, "instrument_agent", {"driver_module": "potato"})
utg.add_resource_unittests(RT.InstrumentDevice, "instrument_device", {"serial_number": "123", "firmware_version": "x"})
utg.add_resource_unittests(RT.InstrumentModel, "instrument_model")
utg.add_resource_unittests(RT.PlatformAgentInstance, "platform_agent_instance", {})
utg.add_resource_unittests(RT.PlatformAgent, "platform_agent", {"description": "the big donut"})
utg.add_resource_unittests(RT.PlatformDevice, "platform_device", {"serial_number": "2345"})
utg.add_resource_unittests(RT.PlatformModel, "platform_model", {"description": "desc"})
utg.add_resource_unittests(RT.SensorDevice, "sensor_device", {"serial_number": "123"})
utg.add_resource_unittests(RT.SensorModel, "sensor_model")

コード例 #5

ファイル: test_observatory_management_service.py プロジェクト: Bobfrat/coi-services

#from pyon.core.exception import BadRequest, Conflict, Inconsistent, NotFound
#import unittest
from pyon.util.unit_test import PyonTestCase


@attr('UNIT', group='sa')
class TestObservatoryManagement(PyonTestCase):

    def setUp(self):
        self.mock_ionobj = self._create_IonObject_mock('ion.services.sa.observatory.observatory_management_service.IonObject')
        
        #self.mock_ionobj = IonObject
        mock_clients = self._create_service_mock('observatory_management')

        self.observatory_mgmt_service = ObservatoryManagementService()
        self.observatory_mgmt_service.clients = mock_clients
        
        # must call this manually
        self.observatory_mgmt_service.on_init()

utg = UnitTestGenerator(TestObservatoryManagement,
                        ObservatoryManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.Deployment, "deployment")
utg.add_resource_unittests(RT.Observatory, "observatory")
utg.add_resource_unittests(RT.Subsite, "subsite")
utg.add_resource_unittests(RT.PlatformSite, "platform_site")
utg.add_resource_unittests(RT.InstrumentSite, "instrument_site")

コード例 #6

ファイル: test_data_product_management_service.py プロジェクト: mbarry02/coi-services

        # Data Product
        dp_obj = IonObject(RT.DataProduct,
                           name='DP_X',
                           description='some existing dp')
        self.clients.resource_registry.find_resources.return_value = ([dp_obj], [])

        # test call
        result = self.data_product_management_service.find_data_products()

        # check results
        self.assertEqual(result, [dp_obj])
        self.clients.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, None, False)



utg = UnitTestGenerator(TestDataProductManagementServiceUnit,
                        DataProductManagementService)

#utg.test_all_in_one(True)

utg.add_resource_unittests(RT.DataProduct, "data_product", {})
utg.add_resource_unittests(RT.DataProductCollection, "data_product_collection", {})

#remove some tests that don't work
delattr(TestDataProductManagementServiceUnit, "test_data_product_create_d14a028")
delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_d14a028")
delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_bad_dupname_d14a028")
delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_bad_noname_d14a028")
delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_bad_wrongtype_d14a028")

コード例 #7

ファイル: test_data_process_management_service.py プロジェクト: sfoley/coi-services

            self.transform_id
        ], "don't care")
        self.transform_object.data_process_definition_id = self.data_proc_def_id
        self.transform_object.in_subscription_id = self.in_product_A
        self.transform_object.out_data_product_id = self.out_product_A
        self.transform_management_service.read_transform.return_value = self.transform_object

        # test call
        dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process(
            self.data_process_id)

        # verify results
        self.assertEqual(dpd_id, self.data_proc_def_id)
        self.assertEqual(in_id, self.in_product_A)
        self.assertEqual(out_id, self.out_product_A)
        self.resource_registry.find_associations.assert_called_once_with(
            self.data_process_id, PRED.hasTransform, '', None, False)
        self.transform_management_service.read_transform.assert_called_once_with(
            self.transform_id)


utg = UnitTestGenerator(Test_DataProcessManagementService_Unit,
                        DataProcessManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.TransformFunction, "transform_function", {})
utg.add_resource_unittests(RT.DataProcessDefinition, "data_process_definition",
                           {})
#utg.add_resource_unittests(RT.DataProcess, "data_process", {})

コード例 #8

ファイル: test_data_process_management_service.py プロジェクト: MatthewArrott/coi-services

                                                                          PRED.hasInputProduct,
                                                                          self.in_product_A,
                                                                          None)
    @unittest.skip('not working')
    def test_read_data_process(self):
        # setup
        self.resource_registry.find_associations.return_value = ([self.transform_id], "don't care")
        self.transform_object.data_process_definition_id = self.data_proc_def_id
        self.transform_object.in_subscription_id = self.in_product_A
        self.transform_object.out_data_product_id = self.out_product_A
        self.transform_management_service.read_transform.return_value = self.transform_object
         
        # test call
        dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process(self.data_process_id)

        # verify results
        self.assertEqual(dpd_id, self.data_proc_def_id)
        self.assertEqual(in_id, self.in_product_A)
        self.assertEqual(out_id, self.out_product_A)
        self.resource_registry.find_associations.assert_called_once_with(self.data_process_id, PRED.hasTransform, '', None, False)
        self.transform_management_service.read_transform.assert_called_once_with(self.transform_id)



utg = UnitTestGenerator(Test_DataProcessManagementService_Unit,
                        DataProcessManagementService)

utg.test_all_in_one(True)

コード例 #9

                'bad')

        ex = cm.exception
        self.assertEqual(ex.message, 'External Data Set bad does not exist')
        self.mock_read.assert_called_once_with('bad', '')


#    def test_register_instrument(self):
#        self.mock_read.return_value = self.instrument
#        self.mock_create.return_value = ('111', 'bla')
#        self.mock_create.return_value = ('222', 'bla')
#        self.mock_create_association.return_value = ['333', 1]
#
#        self.data_acquisition_mgmt_service.register_instrument('111')

utg = UnitTestGenerator(TestDataAcquisitionManagement,
                        DataAcquisitionManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.ExternalDataProvider, "external_data_provider",
                           {})
utg.add_resource_unittests(RT.DataSource, "data_source", {})
utg.add_resource_unittests(RT.DataSourceModel, "data_source_model", {})
utg.add_resource_unittests(RT.DataSourceAgent, "data_source_agent", {})
utg.add_resource_unittests(RT.DataSourceAgentInstance,
                           "data_source_agent_instance", {})
utg.add_resource_unittests(RT.ExternalDataset, "external_dataset", {})
utg.add_resource_unittests(RT.ExternalDatasetAgent, "external_dataset_agent",
                           {})
utg.add_resource_unittests(RT.ExternalDatasetAgentInstance,
                           "external_dataset_agent_instance", {})

コード例 #10

ファイル: test_data_process_management_service.py プロジェクト: tapanagupta/coi-services

    @unittest.skip('not working')
    def test_read_data_process(self):
        # setup
        self.resource_registry.find_associations.return_value = ([
            self.transform_id
        ], "don't care")
        self.transform_object.data_process_definition_id = self.data_proc_def_id
        self.transform_object.in_subscription_id = self.in_product_A
        self.transform_object.out_data_product_id = self.out_product_A
        self.transform_management_service.read_transform.return_value = self.transform_object

        # test call
        dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process(
            self.data_process_id)

        # verify results
        self.assertEqual(dpd_id, self.data_proc_def_id)
        self.assertEqual(in_id, self.in_product_A)
        self.assertEqual(out_id, self.out_product_A)
        self.resource_registry.find_associations.assert_called_once_with(
            self.data_process_id, PRED.hasTransform, '', None, False)
        self.transform_management_service.read_transform.assert_called_once_with(
            self.transform_id)


utg = UnitTestGenerator(Test_DataProcessManagementService_Unit,
                        DataProcessManagementService)

utg.test_all_in_one(True)

コード例 #11

        self.instrument_mgmt_service.clients = mock_clients

        # must call this manually
        self.instrument_mgmt_service.on_init()

        self.addCleanup(delattr, self, "instrument_mgmt_service")
        self.addCleanup(delattr, self, "mock_ionobj")
        #self.resource_impl_cleanup()

        log.debug("setUp complete")

    #def resource_impl_cleanup(self):
    #pass


utg = UnitTestGenerator(TestInstrumentManagement, InstrumentManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.InstrumentAgentInstance,
                           "instrument_agent_instance", {})
utg.add_resource_unittests(RT.InstrumentAgent, "instrument_agent",
                           {"driver_module": "potato"})
utg.add_resource_unittests(RT.InstrumentDevice, "instrument_device", {
    "serial_number": "123",
    "firmware_version": "x"
})
utg.add_resource_unittests(RT.InstrumentModel, "instrument_model")
utg.add_resource_unittests(RT.PlatformAgentInstance, "platform_agent_instance",
                           {})
utg.add_resource_unittests(RT.PlatformAgent, "platform_agent",

コード例 #12

from pyon.util.unit_test import PyonTestCase


@attr('UNIT', group='sa')
class TestObservatoryManagement(PyonTestCase):
    def setUp(self):
        self.mock_ionobj = self._create_IonObject_mock(
            'ion.services.sa.observatory.observatory_management_service.IonObject'
        )

        #self.mock_ionobj = IonObject
        mock_clients = self._create_service_mock('observatory_management')

        self.observatory_mgmt_service = ObservatoryManagementService()
        self.observatory_mgmt_service.clients = mock_clients

        # must call this manually
        self.observatory_mgmt_service.on_init()


utg = UnitTestGenerator(TestObservatoryManagement,
                        ObservatoryManagementService)

utg.test_all_in_one(True)

utg.add_resource_unittests(RT.Deployment, "deployment")
utg.add_resource_unittests(RT.Observatory, "observatory")
utg.add_resource_unittests(RT.Subsite, "subsite")
utg.add_resource_unittests(RT.PlatformSite, "platform_site")
utg.add_resource_unittests(RT.InstrumentSite, "instrument_site")