def testUpdateSlave(self):
slave = SlaveMock()
self._sut.add_slave(slave)
slave.password = "******"
self._sut.update_slave(slave)
slaves = self._sut.get_slaves()
slave.password = md5(slave.password.encode()).hexdigest()
self.assertListEqual([slave.to_tuple()], slaves)
def testAddSlave(self):
slave1 = SlaveMock()
slave2 = SlaveMock("slave2", password="******", owner=45678)
self._sut.add_slave(slave1)
self._sut.add_slave(slave2)
slaves = self._sut.get_slaves()
slave1.password = md5(slave1.password.encode()).hexdigest()
slave2.password = md5(slave2.password.encode()).hexdigest()
self.assertListEqual([slave1.to_tuple(), slave2.to_tuple()], slaves)
slave3 = SlaveMock("a" * 60) # long name
try:
self._sut.add_slave(slave3)
except ValueError as e:
self.assertEqual(e.args[0],
self._rm.get_string("slave_name_too_long"))
else:
assert False
def testGetSlave(self):
slave = SlaveMock()
self._sut.add_slave(slave)
slave_from_db = self._sut.get_slave(slave.nickname)
slave.password = md5(slave.password.encode()).hexdigest()
self.assertEqual(slave.to_tuple(), slave_from_db)
try:
self._sut.get_slave("missing_slave")
except ValueError:
return
else:
assert False
def testAddSlaveTwoTimes(self):
slave = SlaveMock()
self._sut.add_slave(slave)
try:
self._sut.add_slave(slave)
except ValueError as e:
self.assertEqual(e.args[0],
"Slave already exists") # TODO:replace with rm
else:
assert False
slaves = self._sut.get_slaves()
slave.password = md5(slave.password.encode()).hexdigest()
self.assertListEqual([slave.to_tuple()], slaves)