示例#1
0
    def test_get_endpoint_error(self, cache, m):
        url = "http://example.com/test/endpoint/"
        json_data = {'data': [], 'Success': False}
        m.register_uri('GET', url, text=json.dumps(json_data), status_code=400)

        cache.get.return_value = None  # No cached value

        # Act
        result = FleetUpManager.get_endpoint(url)

        # Assert
        self.assertTrue(cache.get.called)
        self.assertFalse(cache.set.called)
        self.assertIsNone(result)
    def test_get_endpoint(self, cache, m):
        url = "http://example.com/test/endpoint/"
        json_data = {'data': "123456", 'CachedUntilUTC': '/Date(1493896236163)/', 'Success': True}
        m.register_uri('GET', url,
                       text=json.dumps(json_data))

        cache.get.return_value = None  # No cached value

        # Act
        result = FleetUpManager.get_endpoint(url)

        # Assert
        self.assertTrue(cache.get.called)
        self.assertTrue(cache.set.called)
        args, kwargs = cache.set.call_args
        self.assertDictEqual(json_data, args[1])

        self.assertDictEqual(json_data, result)