#define PYTESTCAPI_NEED_INTERNAL_API

#include "parts.h"

#include "util.h"

#include "clinic/file.c.h"

static PyObject *

_testcapi_pyfile_newstdprinter_impl(PyObject *module, int fd)

{

return PyFile_NewStdPrinter(fd);

}

static PyObject *

_testcapi_py_fopen_impl(PyObject *module, PyObject *path, const char *mode,

Py_ssize_t mode_length)

{

NULLABLE(path);

FILE *fp = Py_fopen(path, mode);

if (fp == NULL) {

return NULL;

}

char buffer[256];

size_t size = fread(buffer, 1, Py_ARRAY_LENGTH(buffer), fp);

Py_fclose(fp);

return PyBytes_FromStringAndSize(buffer, size);

}

static PyObject *

_testcapi_py_universalnewlinefgets_impl(PyObject *module, PyObject *file,

int size)

{

FILE *fp = Py_fopen(file, "rb");

if (fp == NULL) {

return NULL;

}

char *buf = (char *)PyMem_Malloc(size);

if (buf == NULL) {

Py_fclose(fp);

return PyErr_NoMemory();

}

char *result = Py_UniversalNewlineFgets(buf, size, fp, NULL);

if (result == NULL) {

PyMem_Free(buf);

Py_fclose(fp);

Py_RETURN_NONE;

}

PyObject *line = PyBytes_FromString(result);

PyMem_Free(buf);

Py_fclose(fp);

return line;

}

static PyMethodDef test_methods[] = {

_TESTCAPI_PYFILE_NEWSTDPRINTER_METHODDEF

_TESTCAPI_PY_FOPEN_METHODDEF

_TESTCAPI_PY_UNIVERSALNEWLINEFGETS_METHODDEF

{NULL},

};

_PyTestCapi_Init_File(PyObject *m)

{

return PyModule_AddFunctions(m, test_methods);

}