def test_deriveCreate2CellSecrets(self, mock_drc):
self.circuit._hs_state = 'hs_state'
cell = Created2Cell.make(ID, hdata='\x00'*84)
self.circuit._deriveCreate2CellSecrets(cell, mock.Mock())
self.assertEqual(self.circuit._hs_state, None)
self.assertEqual(self.circuit._crypt_path, ['test'])
mock_drc.assert_called_once_with('hs_state', cell)
def test_deriveCreate2CellSecrets(self, mock_drc):
self.circuit._hs_state = 'hs_state'
cell = Created2Cell.make(ID, hdata='\x00' * 84)
self.circuit._deriveCreate2CellSecrets(cell, mock.Mock())
self.assertEqual(self.circuit._hs_state, None)
self.assertEqual(self.circuit._crypt_path, ['test'])
mock_drc.assert_called_once_with('hs_state', cell)
def test_deriveExtend2CellSecrets_unexpected_cell(self, dec, dcm, conn):
cell = Created2Cell.make(ID, hdata='\x00' * 84)
dec.return_value = (cell, 2)
self.circuit._conn = conn
ret = mock.Mock()
dcm.return_value = ret
self.circuit._crypt_path = []
self.assertRaises(ValueError, self.circuit._deriveExtend2CellSecrets,
cell, mock.Mock())
self.circuit._conn.send.assert_called_once_with(ret)
def test_deriveExtend2CellSecrets_unexpected_cell(self, dec, dcm, conn):
cell = Created2Cell.make(ID, hdata='\x00'*84)
dec.return_value = (cell, 2)
self.circuit._conn = conn
ret = mock.Mock()
dcm.return_value = ret
self.circuit._crypt_path = []
self.assertRaises(ValueError,
self.circuit._deriveExtend2CellSecrets,
cell,
mock.Mock())
self.circuit._conn.send.assert_called_once_with(ret)
