def test_update_pg_hba(self): file_contents = """some pg hba data here\n""" pg_hba_info = '# standby master host ip addresses\nhost\tall\tall\t192.168.2.1/32\ttrust\nhost\tall\tall\t192.168.1.1/32\ttrust\n' data_dirs = ['/tmp/d1', '/tmp/d2'] expected = [file_contents + pg_hba_info, file_contents + pg_hba_info] m = MagicMock() m.return_value.__enter__.return_value.read.side_effect = [file_contents, file_contents] with patch('__builtin__.open', m, create=True): self.assertEqual(expected, update_pg_hba(pg_hba_info, data_dirs))
def test_update_pg_hba_duplicate(self): file_contents = """some pg hba data here\n""" duplicate_entry = """# standby coordinator host ip addresses\nhost\tall\tall\t192.168.2.1/32\ttrust\nhost\tall\tall\t192.168.1.1/32\ttrust\n""" pg_hba_info = '# standby coordinator host ip addresses\nhost\tall\tall\t192.168.2.1/32\ttrust\nhost\tall\tall\t192.168.1.1/32\ttrust\n' data_dirs = ['/tmp/d1', '/tmp/d2'] expected = [file_contents + pg_hba_info] m = MagicMock() m.return_value.__enter__.return_value.read.side_effect = [ file_contents, file_contents + duplicate_entry ] with patch('builtins.open', m, create=True): res = update_pg_hba(pg_hba_info, data_dirs) self.assertEqual(expected, res)