From 02aef65853aa04fc6611b82904d8dfbfe59fdecd Mon Sep 17 00:00:00 2001
From: "d.lepikhova" <d.lepikhova@postgrespro.ru>
Date: Tue, 31 May 2022 11:21:59 +0500
Subject: [PATCH] Fix is_enterprise checking in ptrack_helpers.py

---
 tests/helpers/ptrack_helpers.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/helpers/ptrack_helpers.py b/tests/helpers/ptrack_helpers.py
index 3b14b717..a4ec7c9c 100644
--- a/tests/helpers/ptrack_helpers.py
+++ b/tests/helpers/ptrack_helpers.py
@@ -90,17 +90,17 @@ def dir_files(base_dir):
 def is_enterprise():
     # pg_config --help
     if os.name == 'posix':
-        cmd = [os.environ['PG_CONFIG'], '--help']
+        cmd = [os.environ['PG_CONFIG'], '--pgpro-edition']
 
     elif os.name == 'nt':
-        cmd = [[os.environ['PG_CONFIG']], ['--help']]
+        cmd = [[os.environ['PG_CONFIG']], ['--pgpro-edition']]
 
     p = subprocess.Popen(
         cmd,
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE
     )
-    if b'postgrespro.ru' in p.communicate()[0]:
+    if b'enterprise' in p.communicate()[0]:
         return True
     else:
         return False