Exemple #1
0
def new___init__(self,
                 reader,
                 writer,
                 *,
                 address,
                 encoding=None,
                 parser=None,
                 loop=None):
    if loop is not None and sys.version_info >= (3, 8):
        warnings.warn("The loop argument is deprecated", DeprecationWarning)
    if parser is None:
        parser = Reader
    assert callable(parser), ("Parser argument is not callable", parser)
    self._reader = reader
    self._writer = writer
    self._address = address
    self._waiters = deque()
    self._reader.set_parser(
        parser(protocolError=ProtocolError, replyError=ReplyError))
    self._reader_task = asyncio.ensure_future(self._read_data())
    self._close_msg = None
    self._db = 0
    self._closing = False
    self._closed = False
    self._close_state = asyncio.Event()
    self._reader_task.add_done_callback(lambda x: self._close_state.set())
    self._in_transaction = None
    self._transaction_error = None  # XXX: never used?
    self._in_pubsub = 0
    self._pubsub_channels = coerced_keys_dict()
    self._pubsub_patterns = coerced_keys_dict()
    self._encoding = encoding
    self._pipeline_buffer = None
    self.__rsm = builtins.rsm_map
    def test_simple(self):
        d = coerced_keys_dict()
        self.assertEqual(d, {})

        d = coerced_keys_dict({b'a': 'b', b'c': 'd'})
        self.assertIn('a', d)
        self.assertIn(b'a', d)
        self.assertIn('c', d)
        self.assertIn(b'c', d)
        self.assertEqual(d, {b'a': 'b', b'c': 'd'})
    def test_simple(self):
        d = coerced_keys_dict()
        self.assertEqual(d, {})

        d = coerced_keys_dict({b'a': 'b', b'c': 'd'})
        self.assertIn('a', d)
        self.assertIn(b'a', d)
        self.assertIn('c', d)
        self.assertIn(b'c', d)
        self.assertEqual(d, {b'a': 'b', b'c': 'd'})
def test_simple():
    d = coerced_keys_dict()
    assert d == {}

    d = coerced_keys_dict({b"a": "b", b"c": "d"})
    assert "a" in d
    assert b"a" in d
    assert "c" in d
    assert b"c" in d
    assert d == {b"a": "b", b"c": "d"}
Exemple #5
0
def test_simple():
    d = coerced_keys_dict()
    assert d == {}

    d = coerced_keys_dict({b'a': 'b', b'c': 'd'})
    assert 'a' in d
    assert b'a' in d
    assert 'c' in d
    assert b'c' in d
    assert d == {b'a': 'b', b'c': 'd'}
    def test_valid_init(self):
        d = coerced_keys_dict({b'foo': 'bar'})
        self.assertEqual(d, {b'foo': 'bar'})
        self.assertIn('foo', d)
        self.assertIn(b'foo', d)
        self.assertEqual(d['foo'], 'bar')
        self.assertEqual(d[b'foo'], 'bar')

        d = coerced_keys_dict()
        d.update({b'foo': 'bar'})
        self.assertEqual(d, {b'foo': 'bar'})
        self.assertIn('foo', d)
        self.assertIn(b'foo', d)
        self.assertEqual(d['foo'], 'bar')
        self.assertEqual(d[b'foo'], 'bar')
def test_valid_init():
    d = coerced_keys_dict({b"foo": "bar"})
    assert d == {b"foo": "bar"}
    assert "foo" in d
    assert b"foo" in d
    assert d["foo"] == "bar"
    assert d[b"foo"] == "bar"

    d = coerced_keys_dict()
    d.update({b"foo": "bar"})
    assert d == {b"foo": "bar"}
    assert "foo" in d
    assert b"foo" in d
    assert d["foo"] == "bar"
    assert d[b"foo"] == "bar"
    def test_valid_init(self):
        d = coerced_keys_dict({b'foo': 'bar'})
        self.assertEqual(d, {b'foo': 'bar'})
        self.assertIn('foo', d)
        self.assertIn(b'foo', d)
        self.assertEqual(d['foo'], 'bar')
        self.assertEqual(d[b'foo'], 'bar')

        d = coerced_keys_dict()
        d.update({b'foo': 'bar'})
        self.assertEqual(d, {b'foo': 'bar'})
        self.assertIn('foo', d)
        self.assertIn(b'foo', d)
        self.assertEqual(d['foo'], 'bar')
        self.assertEqual(d[b'foo'], 'bar')
Exemple #9
0
def test_valid_init():
    d = coerced_keys_dict({b'foo': 'bar'})
    assert d == {b'foo': 'bar'}
    assert 'foo' in d
    assert b'foo' in d
    assert d['foo'] == 'bar'
    assert d[b'foo'] == 'bar'

    d = coerced_keys_dict()
    d.update({b'foo': 'bar'})
    assert d == {b'foo': 'bar'}
    assert 'foo' in d
    assert b'foo' in d
    assert d['foo'] == 'bar'
    assert d[b'foo'] == 'bar'
    def test_invalid_init(self):
        d = coerced_keys_dict({'foo': 'bar'})
        self.assertEqual(d, {'foo': 'bar'})

        self.assertNotIn('foo', d)
        self.assertNotIn(b'foo', d)
        with self.assertRaises(KeyError):
            d['foo']
        with self.assertRaises(KeyError):
            d[b'foo']

        d = coerced_keys_dict()
        d.update({'foo': 'bar'})
        self.assertEqual(d, {'foo': 'bar'})

        self.assertNotIn('foo', d)
        self.assertNotIn(b'foo', d)
        with self.assertRaises(KeyError):
            d['foo']
        with self.assertRaises(KeyError):
            d[b'foo']
def test_invalid_init():
    d = coerced_keys_dict({"foo": "bar"})
    assert d == {"foo": "bar"}

    assert "foo" not in d
    assert b"foo" not in d
    with pytest.raises(KeyError):
        d["foo"]
    with pytest.raises(KeyError):
        d[b"foo"]

    d = coerced_keys_dict()
    d.update({"foo": "bar"})
    assert d == {"foo": "bar"}

    assert "foo" not in d
    assert b"foo" not in d
    with pytest.raises(KeyError):
        d["foo"]
    with pytest.raises(KeyError):
        d[b"foo"]
    def test_invalid_init(self):
        d = coerced_keys_dict({'foo': 'bar'})
        self.assertEqual(d, {'foo': 'bar'})

        self.assertNotIn('foo', d)
        self.assertNotIn(b'foo', d)
        with self.assertRaises(KeyError):
            d['foo']
        with self.assertRaises(KeyError):
            d[b'foo']

        d = coerced_keys_dict()
        d.update({'foo': 'bar'})
        self.assertEqual(d, {'foo': 'bar'})

        self.assertNotIn('foo', d)
        self.assertNotIn(b'foo', d)
        with self.assertRaises(KeyError):
            d['foo']
        with self.assertRaises(KeyError):
            d[b'foo']
Exemple #13
0
def test_invalid_init():
    d = coerced_keys_dict({'foo': 'bar'})
    assert d == {'foo': 'bar'}

    assert 'foo' not in d
    assert b'foo' not in d
    with pytest.raises(KeyError):
        d['foo']
    with pytest.raises(KeyError):
        d[b'foo']

    d = coerced_keys_dict()
    d.update({'foo': 'bar'})
    assert d == {'foo': 'bar'}

    assert 'foo' not in d
    assert b'foo' not in d
    with pytest.raises(KeyError):
        d['foo']
    with pytest.raises(KeyError):
        d[b'foo']