def test_plan_get_not_found(self, PlanHandler, resp_mock, request_mock): hand_get = PlanHandler.return_value.get hand_get.side_effect = exception.ResourceNotFound(name='plan', id='test_id') plan.PlanController('test_id').get() hand_get.assert_called_with('test_id') self.assertEqual(404, resp_mock.status)
def test_plan_delete_ok(self, PlanHandler, resp_mock, request_mock): hand_delete = PlanHandler.return_value.delete hand_delete.return_value = None obj = plan.PlanController('test_id') obj.delete() hand_delete.assert_called_with('test_id') self.assertEqual(202, resp_mock.status)
def test_plan_put_empty_yaml(self, PlanHandler, resp_mock, request_mock): request_mock.content_type = 'application/x-yaml' request_mock.body = '{}' hand_update = PlanHandler.return_value.update hand_update.return_value = fakes.FakePlan() plan.PlanController('test_id').put() self.assertEqual(400, resp_mock.status)
def test_plan_delete_not_found(self, PlanHandler, resp_mock, request_mock): hand_delete = PlanHandler.return_value.delete hand_delete.side_effect = exception.ResourceNotFound(name='plan', plan_id='test_id') obj = plan.PlanController('test_id') obj.delete() hand_delete.assert_called_with('test_id') self.assertEqual(404, resp_mock.status)
def test_plan_put_version_not_found_yaml(self, PlanHandler, resp_mock, request_mock): data = 'name: ex_plan1\ndescription: yaml plan1.' request_mock.body = data request_mock.content_type = 'application/x-yaml' hand_update = PlanHandler.return_value.update hand_update.return_value = fakes.FakePlan() plan.PlanController('test_id').put() self.assertEqual(400, resp_mock.status)
def test_plan_put_version_not_found_json(self, PlanHandler, resp_mock, request_mock): data = {'name': 'foo'} request_mock.body = json.dumps(data) request_mock.content_type = 'application/json' hand_update = PlanHandler.return_value.update hand_update.return_value = fakes.FakePlan() plan.PlanController('test_id').put() self.assertEqual(400, resp_mock.status)
def test_plan_put_invalid_yaml(self, PlanHandler, resp_mock, request_mock, mock_policy): mock_policy.return_value = True request_mock.content_type = 'application/x-yaml' request_mock.body = 'invalid yaml file' hand_update = PlanHandler.return_value.update hand_update.return_value = fakes.FakePlan() plan.PlanController('test_id').put() self.assertEqual(400, resp_mock.status)
def test_plan_put_ok_yaml(self, PlanHandler, resp_mock, request_mock): data = 'version: 1\nname: ex_plan1\ndescription: dsc1.' request_mock.body = data request_mock.content_type = 'application/x-yaml' hand_update = PlanHandler.return_value.update hand_update.return_value = fakes.FakePlan() plan.PlanController('test_id').put() hand_update.assert_called_with('test_id', {'name': 'ex_plan1', 'description': u'dsc1.'}) self.assertEqual(200, resp_mock.status)
def test_plan_get_not_found_yaml(self, PlanHandler, resp_mock, request_mock, mock_policy): mock_policy.return_value = True request_mock.accept = 'application/x-yaml' hand_get = PlanHandler.return_value.get hand_get.side_effect = exception.ResourceNotFound(name='plan', id='test_id') plan.PlanController('test_id').get() hand_get.assert_called_with('test_id') self.assertEqual(404, resp_mock.status)
def test_plan_put_ok_json(self, PlanHandler, resp_mock, request_mock, mock_policy): mock_policy.return_value = True data = {'name': 'foo', 'version': '1'} request_mock.body = json.dumps(data) request_mock.content_type = 'application/json' hand_update = PlanHandler.return_value.update hand_update.return_value = fakes.FakePlan() plan.PlanController('test_id').put() hand_update.assert_called_with('test_id', {'name': 'foo'}) self.assertEqual(200, resp_mock.status)
def test_plan_put_not_found(self, PlanHandler, resp_mock, request_mock): data = 'version: 1\nname: ex_plan1\ndescription: dsc1.' request_mock.body = data request_mock.content_type = 'application/x-yaml' hand_update = PlanHandler.return_value.update hand_update.side_effect = exception.ResourceNotFound( name='plan', id='test_id') plan.PlanController('test_id').put() hand_update.assert_called_with('test_id', {'name': 'ex_plan1', 'description': u'dsc1.'}) self.assertEqual(404, resp_mock.status)
def test_plan_get(self, PlanHandler, resp_mock, request_mock): hand_get = PlanHandler.return_value.get fake_plan = fakes.FakePlan() hand_get.return_value = fake_plan cont = plan.PlanController('test_id') resp = cont.get() self.assertIsNotNone(resp) resp_yml = yaml.load(resp) self.assertEqual(fake_plan.raw_content['name'], resp_yml['name']) hand_get.assert_called_with('test_id') self.assertEqual(200, resp_mock.status)
def test_plan_delete_dberror(self, PlanHandler, resp_mock, request_mock): hand_get = PlanHandler.return_value.get fake_plan = fakes.FakePlan() hand_get.return_value = fake_plan hand_delete = PlanHandler.return_value.delete hand_delete.side_effect = db_exc.DBError() obj = plan.PlanController('test_id') obj.delete() hand_delete.assert_called_with('test_id') self.assertEqual(409, resp_mock.status)
def test_plan_put_bad_names(self, PlanHandler, resp_mock, request_mock): request_mock.content_type = 'application/x-yaml' hand_update = PlanHandler.return_value.update hand_update.return_value = fakes.FakePlan() pc = plan.PlanController('test_id') request_mock.body = 'version: 1\nname: %s' % ('a' * 101) pc.put() self.assertEqual(400, resp_mock.status) request_mock.body = 'version: 1\nname: %s' % ('name with spaces') pc.put() self.assertEqual(400, resp_mock.status)
def test_plan_get_json(self, PlanHandler, resp_mock, request_mock): hand_get = PlanHandler.return_value.get fake_plan = fakes.FakePlan() hand_get.return_value = fake_plan cont = plan.PlanController('test_id') resp = cont.get() self.assertIsNotNone(resp) resp_json = json.loads(resp) self.assertEqual(fake_plan.name, resp_json['name']) self.assertEqual(fake_plan.project_id, resp_json['project_id']) self.assertEqual(fake_plan.uuid, resp_json['uuid']) hand_get.assert_called_with('test_id') self.assertEqual(200, resp_mock.status)
def test_plan_get_yaml(self, PlanHandler, resp_mock, request_mock, mock_policy): mock_policy.return_value = True request_mock.accept = 'application/x-yaml' hand_get = PlanHandler.return_value.get fake_plan = fakes.FakePlan() hand_get.return_value = fake_plan cont = plan.PlanController('test_id') resp = cont.get() self.assertIsNotNone(resp) resp_yml = yaml.safe_load(resp) self.assertEqual(fake_plan.raw_content['name'], resp_yml['name']) hand_get.assert_called_with('test_id') self.assertEqual(200, resp_mock.status)
def test_plan_get(self, resp_mock, request_mock): plan_obj = plan.PlanController('test_id') plan_obj.get() self.assertEqual(200, resp_mock.status)
def test_plan_put(self, resp_mock, request_mock): obj = plan.PlanController('test_id') obj.put(None) self.assertEqual(501, resp_mock.status)
def test_plan_delete(self, resp_mock, request_mock): obj = plan.PlanController('test_id') obj.delete() self.assertEqual(501, resp_mock.status)