Exemple #1
0
 def setUp(self):
     from dtrove.providers.openstack import Provider
     self.provider = Provider()
     # Mocks
     mock_nova = patch('dtrove.providers.openstack.nova_client')
     mock_cinder = patch('dtrove.providers.openstack.cinder_client')
     mock_key = patch('dtrove.providers.openstack.keystone_client')
     mock_provision = patch('dtrove.models.Instance.provision')
     self.MockNova = mock_nova.start()
     self.MockCinder = mock_cinder.start()
     self.MockKeystone = mock_key.start()
     self.MockProvision = mock_provision.start()
     # Add clean ups
     self.addCleanup(mock_nova.stop)
     self.addCleanup(mock_cinder.stop)
     self.addCleanup(mock_key.stop)
     self.addCleanup(mock_provision.stop)
     # An instance to provision
     self.instance = create_instance(server='', save=True)
     # shortcut to mock clients
     self.nova = self.MockNova.Client()
     self.keystone = self.MockKeystone.Client()
Exemple #2
0
class OpenStackProviderTests(DtroveTest):

    def setUp(self):
        from dtrove.providers.openstack import Provider
        self.provider = Provider()
        # Mocks
        mock_nova = patch('dtrove.providers.openstack.nova_client')
        mock_cinder = patch('dtrove.providers.openstack.cinder_client')
        mock_key = patch('dtrove.providers.openstack.keystone_client')
        mock_provision = patch('dtrove.models.Instance.provision')
        self.MockNova = mock_nova.start()
        self.MockCinder = mock_cinder.start()
        self.MockKeystone = mock_key.start()
        self.MockProvision = mock_provision.start()
        # Add clean ups
        self.addCleanup(mock_nova.stop)
        self.addCleanup(mock_cinder.stop)
        self.addCleanup(mock_key.stop)
        self.addCleanup(mock_provision.stop)
        # An instance to provision
        self.instance = create_instance(server='', save=True)
        # shortcut to mock clients
        self.nova = self.MockNova.Client()
        self.keystone = self.MockKeystone.Client()

    def test_nova_client(self):
        self.assertEqual(self.nova, self.provider.nova)
        self.MockNova.Client.assert_called_with(
            username=settings.OS_USERNAME,
            project_id=settings.OS_PROJECT_ID,
            region_name='IAD',
            bypass_url=None,
            auth_token=self.keystone.auth_ref.auth_token,
            auth_url=settings.OS_AUTH_URL,
            api_key=settings.OS_PASSWORD,
        )

    def test_create(self):
        servers = [
            OSServer('127.0.0.1', 'uuid', 'building', 10, {}),
            OSServer('127.0.0.1', 'uuid', 'building', 50, {}),
            OSServer('127.0.0.1', 'uuid', 'active', 90, {}),
        ]

        def side_effect(*args, **kwargs):
            return servers.pop(0)

        self.nova.servers.get.side_effect = side_effect
        self.provider.create(self.instance)
        self.assertEqual('active', self.instance.server_status)
        self.assertEqual(100, self.instance.progress)

    def test_create_error(self):
        server = OSServer('127.0.0.1', 'id', 'error', 10, {'message': 'fail'})
        self.nova.servers.get.return_value = server
        self.assertRaises(Exception, self.provider.create, self.instance)
        self.assertEqual('error', self.instance.server_status)
        self.assertEqual(10, self.instance.progress)
        self.assertEqual('fail', self.instance.message)

    def test_update_status(self):
        # Test update_status on a new instance with no server
        self.provider.update_status(self.instance)
        self.assertEqual('NA', self.instance.server_status)
        self.assertEqual(0, self.instance.progress)

    def test_url(self):
        SC = {
            'foo': [
                {'region': 'IAD', 'publicURL': 'my_url'}
            ]
        }
        self.keystone.auth_ref.service_catalog.get_endpoints.return_value = SC
        url = self.provider.url('foo')
        self.assertEqual('my_url', url)

    def test_create_missing_key(self):
        self.nova.keypairs.get.side_effect = Exception
        self.provider.create_key(self.instance.key)
        self.nova.keypairs.create.assert_called_with(
            name=self.instance.key.name,
            public_key=self.instance.key.public
        )