def test_controller_detail_put_ok(self):
gc_id = '1'
controller_data = {'description': 'New description'}
request = self.factory.put('/controllers/' + gc_id,
controller_data,
format='json')
response = controller_detail(request, gc_id)
self.assertEqual(response.status_code, status.HTTP_200_OK)
# Check description has been updated
request = self.factory.get('/controllers/' + gc_id)
response = controller_detail(request, gc_id)
controller = json.loads(response.content)
self.assertEqual(controller['description'], 'New description')
def test_create_controller_ok(self):
with open('test_data/test.py', 'r') as fp:
metadata = {
'class_name': 'TestClass',
'enabled': 'False',
'description': 'test controller',
'type': 'get'
}
request = self.factory.post('/controllers/data/', {
'file': fp,
'metadata': json.dumps(metadata)
})
response = ControllerData.as_view()(request)
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
global_controller = json.loads(response.content)
self.assertEqual(global_controller['id'], 2)
self.assertEqual(global_controller['enabled'], 'False')
self.assertEqual(global_controller['controller_name'], 'test.py')
# check the global controller has been created
gc_id = '2'
request = self.factory.get('/controllers/' + gc_id)
response = controller_detail(request, gc_id)
self.assertEqual(response.status_code, status.HTTP_200_OK)
gc_data = json.loads(response.content)
self.assertEqual(gc_data['controller_name'], 'test.py')
def test_controller_detail_get_ok(self):
gc_id = '1'
request = self.factory.get('/controllers/' + gc_id)
response = controller_detail(request, gc_id)
global_controller = json.loads(response.content)
self.assertEqual(global_controller['class_name'],
"MinTenantSLOGlobalSpareBWShare")
def test_controller_detail_delete_ok(self):
gc_id = '1'
request = self.factory.delete('/controllers/' + gc_id)
response = controller_detail(request, gc_id)
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
# Verify controller is deleted
request = self.factory.get('/controllers')
response = controller_list(request)
self.assertEqual(response.status_code, status.HTTP_200_OK)
global_controllers = json.loads(response.content)
self.assertEqual(len(global_controllers), 0)
def test_update_controller_with_put_ok(self):
gc_id = '1'
with open('test_data/test.py', 'r') as fp:
request = self.factory.put('/controllers/' + gc_id + '/data',
{'file': fp})
response = ControllerData.as_view()(request, gc_id)
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
# check the global controller has been updated
request = self.factory.get('/controllers/' + gc_id)
response = controller_detail(request, gc_id)
self.assertEqual(response.status_code, status.HTTP_200_OK)
gc_data = json.loads(response.content)
self.assertEqual(gc_data['controller_name'], 'test.py')
def test_controller_detail_with_method_not_allowed(self):
gc_id = '1'
request = self.factory.post('/controllers/' + gc_id)
response = controller_detail(request, gc_id)
self.assertEqual(response.status_code,
status.HTTP_405_METHOD_NOT_ALLOWED)