def test_extra_auth_messages_raise_errors(self): p = self.backend() p.messageReceived(messages.readyForQuery('idle')) try: p.messageReceived(messages.authenticationOk()) except AssertionError, e: self.assertEqual( e.message, 'Adding auth message, but authentication complete')
def test_subsequent_sets_override_original_auth_parameter_status(self): p = self.backend() a = messages.authenticationOk() f = messages.parameterStatus('foo', 'bar') b = messages.parameterStatus('baz', 'goo') z = messages.readyForQuery('idle') p.messageReceived(a) p.messageReceived(f) p.messageReceived(b) p.messageReceived(z) self.assertEqual(p.authenticationResponse, [a, f, b, z]) x = messages.parameterStatus('foo', 'bar2') p.messageReceived(x) self.assertEqual(p.authenticationResponse, [a, x, b, z])
def test_saving_authmessage(self): p = self.backend() m = messages.authenticationOk() p.messageReceived(m) self.assertEqual(p.authenticationResponse, [m]) self.assertFalse(p.authenticationComplete)
def receiveAuth(self, p): p.messageReceived(messages.authenticationOk()) p.messageReceived(messages.parameterStatus('foo', 'bar')) p.messageReceived(messages.readyForQuery('idle'))
def test_authenticationOk(self): m = messages.authenticationOk() self.assertEqual(m.serialize(), 'R\x00\x00\x00\x08\x00\x00\x00\x00')