Python DatalakeFile._get_json_from_remoteの例

コード例 #1

 def test_get_json(self):
     datalake_file = DatalakeFile(None, uri=self.uri, type=type)
     mock_func = create_autospec(datalake_file._get_json_from_remote,
                                 return_value=self.json_data)
     datalake_file._get_json_from_remote = mock_func
     data = datalake_file.get_json()
     self.assertEqual(data, self.json_data)
     mock_func.assert_called_once_with()

コード例 #2

 def test_get_json_from_remote(self):
     datalake_file = DatalakeFile(None, uri=self.uri, type=type)
     mock_response = self._build_json_response()
     datalake_file._do_download = MagicMock(return_value=mock_response)
     data = datalake_file._get_json_from_remote()
     self.assertEqual(data, self.json_data)