コード例 #1
0
class TestModel(unittest.TestCase):
    def setUp(self):
        self.resource = Model(base, auth)

    def test_init(self):
        assert self.resource.base_uri == base
        assert self.resource.auth == auth

    def test_load(self):
        self.resource.load({
            "abc": "foo",
        })
        assert self.resource.abc == "foo"

    @patch("textmagic.rest.models.base.make_tm_request")
    def test_request(self, mock_tm_request):
        response = Mock(content=b'{"foo": "aaa"}')
        mock_tm_request.return_value = response
        r, content = self.resource.request("GET", base, foo="abc")
        mock_tm_request.assert_called_with("GET", base, auth=auth, foo="abc")

        assert (r, content) == (response,
                                json.loads(b'{"foo": "aaa"}'.decode("utf-8")))

        response = Mock(content=b'{"foo": "aaa"}')
        mock_tm_request.return_value = response
        r, content = self.resource.request("DELETE", base, foo="abc")
        mock_tm_request.assert_called_with("DELETE",
                                           base,
                                           auth=auth,
                                           foo="abc")

        assert (r, content) == (response, {})
コード例 #2
0
class TestModel(unittest.TestCase):

    def setUp(self):
        self.resource = Model(base, auth)

    def test_init(self):
        assert self.resource.base_uri == base
        assert self.resource.auth == auth

    def test_load(self):
        self.resource.load({
            "abc": "foo",
        })
        assert self.resource.abc == "foo"

    @patch("textmagic.rest.models.base.make_tm_request")
    def test_request(self, mock_tm_request):
        response = Mock(content=b'{"foo": "aaa"}')
        mock_tm_request.return_value = response
        r, content = self.resource.request("GET", base, foo="abc")
        mock_tm_request.assert_called_with(
            "GET",
            base,
            auth=auth,
            foo="abc"
        )

        assert (r, content) == (response, json.loads(b'{"foo": "aaa"}'.decode("utf-8")))

        response = Mock(content=b'{"foo": "aaa"}')
        mock_tm_request.return_value = response
        r, content = self.resource.request("DELETE", base, foo="abc")
        mock_tm_request.assert_called_with(
            "DELETE",
            base,
            auth=auth,
            foo="abc"
        )

        assert (r, content) == (response, {})
コード例 #3
0
 def setUp(self):
     self.resource = Model(base, auth)
コード例 #4
0
 def setUp(self):
     self.resource = Model(base, auth)