def test_memset_success():
p = MemsetProtocol()
f = protocol.ClientFactory()
f.key = 'a'
f.value = 'v'
f.deferred = d = defer.Deferred()
p.factory = f
f.timeout = 0.2
def stored(key):
assert key == 'a'
d.addCallback(stored)
p.lineReceived('STORED')
return d
def test_memset_failure():
p = MemsetProtocol()
f = protocol.ClientFactory()
f.key = 'a'
f.value = 'v'
f.deferred = d = defer.Deferred()
p.factory = f
f.timeout = 0.2
def failed(f):
assert f.getErrorMessage() == "bad bad bad"
d.addErrback(failed)
p.lineReceived('CLIENT_ERROR bad bad bad')
return d
def test_memset_timeout():
p = MemsetProtocol()
p.transport = mock = MockTransport()
f = protocol.ClientFactory()
f.key = 'a'
f.value = 'v'
f.timeout = 0.2
f.deferred = d = defer.Deferred()
p.factory = f
p.connectionMade()
def failed(f):
assert f.getErrorMessage() == "Timeout"
d.addErrback(failed)
return d
def test_memset_connection():
p = MemsetProtocol()
p.transport = mock = MockTransport()
f = protocol.ClientFactory()
f.key = 'a'
f.deferred = defer.Deferred()
f.value = 'v'
p.factory = f
f.timeout = 0.2
p.connectionMade()
expectedValue = '\r\n'.join(["set a 0 0 1", 'v', ''])
assert p.transport.value() == expectedValue
p.lineReceived('')
