class GmpCloneScheduleTestCase(unittest.TestCase): def setUp(self): self.connection = MockConnection() self.gmp = Gmp(self.connection) def test_clone(self): self.gmp.clone_schedule('a1') self.connection.send.has_been_called_with('<create_schedule>' '<copy>a1</copy>' '</create_schedule>') def test_missing_id(self): with self.assertRaises(RequiredArgument): self.gmp.clone_schedule('') with self.assertRaises(RequiredArgument): self.gmp.clone_schedule(None)