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)