class TestRestore(unittest.TestCase): def setUp(self): self.restore = Restore("batman", "master", "/data", "host=batman port=5432 user=batman") pass def tearDown(self): pass def test_parse_connstring(self): self.assertDictEqual(self.restore.master_connection, { 'host': 'batman', 'port': '5432', 'user': '******' }) @patch('subprocess.call', MagicMock(return_value=0)) def test_run(self): ret = self.restore.run() self.assertEqual(ret, 0) @patch('subprocess.call', MagicMock(return_value=1)) def test_run_fail(self): ret = self.restore.run() self.assertEqual(ret, 1)
class TestRestore(unittest.TestCase): def setUp(self): self.restore = Restore("batman", "master", "/data", "host=batman port=5432 user=batman") pass def tearDown(self): pass def test_parse_connstring(self): self.assertDictEqual(self.restore.master_connection, {'host': 'batman', 'port': '5432', 'user': '******'}) @patch('subprocess.call', MagicMock(return_value=0)) def test_run(self): ret = self.restore.run() self.assertEqual(ret, 0) @patch('subprocess.call', MagicMock(return_value=1)) def test_run_fail(self): ret = self.restore.run() self.assertEqual(ret, 1)