def test_socketcomm_close(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.close() comm._conn.shutdown.assert_called_with() comm._conn.close.assert_called_with()
def test_socketcomm_address(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.getpeername.return_value = "127.0.0.1", 1234 eq_(comm.address, ("127.0.0.1", 1234)) comm._conn.getpeername.assert_called_with()
def test_serialcomm_read_raw_timeout(): with pytest.raises(IOError): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.recv = mock.MagicMock(side_effect=[b"a", b"b", b""]) _ = comm.read_raw(-1)
def test_socketcomm_address(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.getpeername.return_value = "127.0.0.1", 1234 assert comm.address == ("127.0.0.1", 1234) comm._conn.getpeername.assert_called_with()
def test_socketcomm_close(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.close() comm._conn.shutdown.assert_called_with(socket.SHUT_RDWR) comm._conn.close.assert_called_with()
def test_socketcomm_flush_input(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.read = mock.MagicMock() comm.flush_input() comm.read.assert_called_with(-1)
def test_serialcomm_read_raw_timeout(): with pytest.raises(IOError): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.recv = mock.MagicMock(side_effect=[b"a", b"b", b""]) _ = comm.read_raw(-1)
def test_loopbackcomm_read_raw_2char_terminator(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.recv = mock.MagicMock(side_effect=[b"a", b"b", b"c", b"\r", b"\n"]) comm._terminator = "\r\n" eq_(comm.read_raw(), b"abc") comm._conn.recv.assert_has_calls([mock.call(1)] * 5) assert comm._conn.recv.call_count == 5
def test_socketcomm_read_raw(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.recv = mock.MagicMock(side_effect=[b"a", b"b", b"c", b"\n"]) eq_(comm.read_raw(), b"abc") comm._conn.recv.assert_has_calls([mock.call(1)]*4) assert comm._conn.recv.call_count == 4 comm._conn.recv = mock.MagicMock() comm.read_raw(10) comm._conn.recv.assert_called_with(10)
def test_socketcomm_timeout(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.gettimeout.return_value = 1.234 unit_eq(comm.timeout, 1.234 * pq.second) comm._conn.gettimeout.assert_called_with() comm.timeout = 10 comm._conn.settimeout.assert_called_with(10) comm.timeout = 1000 * pq.millisecond comm._conn.settimeout.assert_called_with(1)
def test_socketcomm_timeout(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.gettimeout.return_value = 1.234 unit_eq(comm.timeout, 1.234 * pq.second) comm._conn.gettimeout.assert_called_with() comm.timeout = 10 comm._conn.settimeout.assert_called_with(10) comm.timeout = 1000 * pq.millisecond comm._conn.settimeout.assert_called_with(1)
def test_socketcomm_terminator(): comm = SocketCommunicator(socket.socket()) # Default terminator should be \n eq_(comm.terminator, "\n") comm.terminator = b"*" eq_(comm.terminator, "*") eq_(comm._terminator, "*") comm.terminator = u"\r" eq_(comm.terminator, u"\r") eq_(comm._terminator, u"\r") comm.terminator = "\r\n" eq_(comm.terminator, "\r\n") eq_(comm._terminator, "\r\n")
def test_socketcomm_terminator(): comm = SocketCommunicator(socket.socket()) # Default terminator should be \n eq_(comm.terminator, "\n") comm.terminator = b"*" eq_(comm.terminator, "*") eq_(comm._terminator, "*") comm.terminator = u"\r" # pylint: disable=redefined-variable-type eq_(comm.terminator, u"\r") eq_(comm._terminator, u"\r") comm.terminator = "\r\n" eq_(comm.terminator, "\r\n") eq_(comm._terminator, "\r\n")
def test_socketcomm_terminator(): comm = SocketCommunicator(socket.socket()) # Default terminator should be \n assert comm.terminator == "\n" comm.terminator = b"*" assert comm.terminator == "*" assert comm._terminator == "*" comm.terminator = u"\r" assert comm.terminator == u"\r" assert comm._terminator == u"\r" comm.terminator = "\r\n" assert comm.terminator == "\r\n" assert comm._terminator == "\r\n"
def test_socketcomm_terminator(): comm = SocketCommunicator(socket.socket()) # Default terminator should be \n assert comm.terminator == "\n" comm.terminator = b"*" assert comm.terminator == "*" assert comm._terminator == "*" comm.terminator = u"\r" assert comm.terminator == u"\r" assert comm._terminator == u"\r" comm.terminator = "\r\n" assert comm.terminator == "\r\n" assert comm._terminator == "\r\n"
def test_socketcomm_query(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.read = mock.MagicMock(return_value="answer") comm.sendcmd = mock.MagicMock() assert comm._query("mock") == "answer" comm.sendcmd.assert_called_with("mock") comm.read.assert_called_with(-1) comm._query("mock", size=10) comm.read.assert_called_with(10)
def test_socketcomm_flush_input(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.read = mock.MagicMock() comm.flush_input() comm.read.assert_called_with(-1)
def test_socketcomm_query(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.read = mock.MagicMock(return_value="answer") comm.sendcmd = mock.MagicMock() eq_(comm._query("mock"), "answer") comm.sendcmd.assert_called_with("mock") comm.read.assert_called_with(-1) comm._query("mock", size=10) comm.read.assert_called_with(10)
def open_tcpip(cls, host, port): """ Opens an instrument, connecting via TCP/IP to a given host and TCP port. :param str host: Name or IP address of the instrument. :param int port: TCP port on which the insturment is listening. :rtype: `Instrument` :return: Object representing the connected instrument. .. seealso:: `~socket.socket.connect` for description of `host` and `port` parameters in the TCP/IP address family. """ conn = socket.socket() conn.connect((host, port)) return cls(SocketCommunicator(conn))
def test_socketcomm_read_raw(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._conn.recv = mock.MagicMock(side_effect=[b"a", b"b", b"c", b"\n"]) assert comm.read_raw() == b"abc" comm._conn.recv.assert_has_calls([mock.call(1)] * 4) assert comm._conn.recv.call_count == 4 comm._conn.recv = mock.MagicMock() comm.read_raw(10) comm._conn.recv.assert_called_with(10)
def test_socketcomm_write_raw(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.write_raw(b"mock") comm._conn.sendall.assert_called_with(b"mock")
def test_socketcomm_address_setting(): with pytest.raises(NotImplementedError): comm = SocketCommunicator(socket.socket()) comm.address = "foobar"
def test_socketcomm_init_wrong_filelike(): _ = SocketCommunicator("derp")
def test_socketcomm_init_wrong_filelike(): with pytest.raises(TypeError): _ = SocketCommunicator("derp")
def test_socketcomm_sendcmd(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._sendcmd("mock") comm._conn.sendall.assert_called_with(b"mock\n")
def test_socketcomm_address_setting(): comm = SocketCommunicator(socket.socket()) comm.address = "foobar"
def test_socketcomm_init(): socket_object = socket.socket() comm = SocketCommunicator(socket_object) assert isinstance(comm._conn, socket.socket) is True assert comm._conn == socket_object
def test_socketcomm_tell(): with pytest.raises(NotImplementedError): comm = SocketCommunicator(socket.socket()) comm.tell()
def test_socketcomm_tell(): with pytest.raises(NotImplementedError): comm = SocketCommunicator(socket.socket()) comm.tell()
def test_socketcomm_tell(): comm = SocketCommunicator(socket.socket()) comm.tell()
def test_socketcomm_address_setting(): with pytest.raises(NotImplementedError): comm = SocketCommunicator(socket.socket()) comm.address = "foobar"
def test_socketcomm_write_raw(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm.write_raw(b"mock") comm._conn.sendall.assert_called_with(b"mock")
def test_socketcomm_sendcmd(): comm = SocketCommunicator(socket.socket()) comm._conn = mock.MagicMock() comm._sendcmd("mock") comm._conn.sendall.assert_called_with(b"mock\n")
def test_socketcomm_seek(): comm = SocketCommunicator(socket.socket()) comm.seek(1)