Python CrushMapViewSet示例

示例#1

    def setUp(self):
        self.request = mock.Mock()

        with mock.patch('calamari_rest.views.v2.RPCViewSet'):
            self.cmvs = CrushMapViewSet()
            self.cmvs.client = mock.MagicMock()
            self.cmvs.client.update.return_value = {}

示例#2

class TestCrushMap(TestCase):

    def setUp(self):
        self.request = mock.Mock()

        with mock.patch('calamari_rest.views.v2.RPCViewSet'):
            self.cmvs = CrushMapViewSet()
            self.cmvs.client = mock.MagicMock()
            self.cmvs.client.update.return_value = {}

    def test_retrieve_crushmap(self):
        self.request.method = 'GET'
        response = self.cmvs.retrieve(self.request, 12345)
        self.assertEqual(response.status_code, 200)

    def test_replace_crushmap_passes(self):
        self.request.method = 'POST'
        self.request.DATA = {}
        response = self.cmvs.replace(self.request, 12345)
        self.assertEqual(response.status_code, 200)