@@@ remove debugging print

[mLib-python] / t / t-sys.py

### -*-python-*-
###
### Test system-specific functionality
###
### (c) 2019 Straylight/Edgeware
###

###----- Licensing notice ---------------------------------------------------
###
### This file is part of the Python interface to mLib.
###
### mLib/Python is free software: you can redistribute it and/or modify it
### under the terms of the GNU General Public License as published by the
### Free Software Foundation; either version 2 of the License, or (at your
### option) any later version.
###
### mLib/Python is distributed in the hope that it will be useful, but
### WITHOUT ANY WARRANTY; without even the implied warranty of
### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
### General Public License for more details.
###
### You should have received a copy of the GNU General Public License
### along with mLib/Python.  If not, write to the Free Software
### Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
### USA.

import mLib as M
import fcntl as FC
import os as OS
import socket as SK
import sys as SYS
import testutils as T
import unittest as U

## Cygwin leaves cruft in the top bits of inode numbers for some reason.
if SYS.platform.startswith('cygwin'):
  def _hack_inode(ino): return ino&0x0000ffffffffffff
else:
  def _hack_inode(ino): return ino

###--------------------------------------------------------------------------
class TestSys (U.TestCase):

  def _fdid(me, fd):
    st = OS.fstat(fd)
    return st.st_dev, _hack_inode(st.st_ino)

  def _check_same_file(me, fd0, fd1):
    me.assertEqual(me._fdid(fd0), me._fdid(fd1))

  def _make_fd(me):
    fd, other = OS.pipe()
    OS.close(other)
    return fd

  def test_fdflags(me):
    fd = me._make_fd()
    f = OS.fdopen(fd, "r")
    try:
      of, ofd = FC.fcntl(fd, FC.F_GETFL), FC.fcntl(fd, FC.F_GETFD)
      M.fdflags(fd, fbic = OS.O_NONBLOCK, fxor = OS.O_NONBLOCK)
      me.assertEqual(FC.fcntl(fd, FC.F_GETFL), of | OS.O_NONBLOCK)
      M.fdflags(f, fbic = OS.O_NONBLOCK)
      me.assertEqual(FC.fcntl(fd, FC.F_GETFL), of&~OS.O_NONBLOCK)
      M.fdflags(fd, fdbic = FC.FD_CLOEXEC, fdxor = FC.FD_CLOEXEC)
      me.assertEqual(FC.fcntl(fd, FC.F_GETFD), ofd | FC.FD_CLOEXEC)
      M.fdflags(f, fdbic = FC.FD_CLOEXEC)
      me.assertEqual(FC.fcntl(fd, FC.F_GETFD), ofd&~FC.FD_CLOEXEC)
      me.assertRaises(AttributeError, M.fdflags, "bzzt")
      me.assertRaises(OSError, M.fdflags, -1, 1, 1)
    finally:
      f.close()

  def test_fdpass(me):
    sk0, sk1 = SK.socketpair(SK.AF_UNIX, SK.SOCK_STREAM)
    fd = me._make_fd()
    nfd = -1
    try:
      msg = T.bin("some unnecessary message")
      n = M.fdsend(sk0, fd, msg)
      me.assertEqual(n, len(msg))
      nfd, buf = M.fdrecv(sk1, 32)
      me.assertEqual(buf, msg)
      me._check_same_file(fd, nfd)
      OS.close(nfd); nfd = -1
    finally:
      sk0.close(); sk1.close()
      OS.close(fd)
      if nfd != -1: OS.close(nfd)

  def test_mdup(me):
    NFD = 5
    fds = [me._make_fd() for i in T.range(NFD)]
    v = [(fds[i], fds[(i + 1)%5]) for i in T.range(NFD)]
    try:
      id = [me._fdid(fds[i]) for i in T.range(NFD)]
      vv = M.mdup(v)
      me.assertTrue(vv is v)
      for i in T.range(NFD):
        cur, want = v[i]
        me.assertEqual(cur, want)
        me.assertEqual(cur, fds[(i + 1)%NFD])
        me.assertEqual(me._fdid(cur), id[i])
    finally:
      for fd, _ in v: OS.close(fd)

###----- That's all, folks --------------------------------------------------

if __name__ == "__main__": U.main()