class TestGetObject:
def setup(self):
self.test_cluster = ClusterFixture()
self.cluster = self.test_cluster.obj
self.cluster.save()
def teardown(self):
self.test_cluster.destroy()
def test_get_objects(self):
obj = get_object(self.cluster.__class__,
self.cluster.id, self.cluster.owner)
assert obj.get_dict(True) == obj.get_dict(True)
@pytest.mark.parametrize('bad_user', [
'None',
'',
None,
{},
])
def test_get_object_malformed_user(self, bad_user):
with pytest.raises(BackendError, match='Missing namespace for class'):
get_object(self.cluster.__class__, self.cluster.id, bad_user)
def test_cluster_list_run_get_state(self, monkeypatch):
clusters_to_remove = []
for _ in range(10):
test_cluster = ClusterFixture()
clusters_to_remove.append(test_cluster)
c = test_cluster.obj
c.save()
def fake_get_state(self):
self.metadata = {'executed': True}
self.save()
return config.get('CLUSTER_UNKNOWN_STATE')
monkeypatch.setattr(self.obj.__class__, 'get_state', fake_get_state)
response = self.client.get(
url_for('api.cluster_list'),
headers=self.auth_header,
content_type='application/json',
)
assert response.status_code == 200
obj = self.obj.__class__.load(self.namespace, self.obj.id)
assert obj.metadata, 'get_state wasn\'t executed for cluster {}'.format(obj)
assert obj.metadata['executed'], 'get_state wasn\'t executed'
for c in clusters_to_remove:
c.destroy()
def setup(self):
self.test_cluster = ClusterFixture()
self.cluster = self.test_cluster.obj
self.cluster.save()
def get_object(self):
obj = ClusterFixture()
return obj