test_handlers_clusters.py 文件源码

def test_cluster_upgrade_retrieve(self):
        """
        Verify retrieving a cluster upgrade.
        """
        with mock.patch('cherrypy.engine.publish') as _publish:
            manager = mock.MagicMock(StoreHandlerManager)
            _publish.return_value = [manager]

            test_cluster_upgrade = make_new(CLUSTER_UPGRADE)
            # Verify if the cluster upgrade exists the data is returned
            manager.get.return_value = test_cluster_upgrade
            body = self.simulate_request('/api/v0/cluster/development/upgrade')
            self.assertEqual(falcon.HTTP_200, self.srmock.status)
            self.assertEqual(json.loads(test_cluster_upgrade.to_json()), json.loads(body[0]))

            # Verify no cluster upgrade returns the proper result
            manager.reset_mock()
            manager.get.side_effect = (
                test_cluster_upgrade,
                Exception)

            body = self.simulate_request('/api/v0/cluster/development/upgrade')
            self.assertEqual(falcon.HTTP_204, self.srmock.status)
            self.assertEqual([], body)  # Empty data