test(beahave): Added one more step

* Card ID: CCT-1921
* Added check that productid.json contains wanted repo_id
  and product_id
* Make black more happy

Signed-off-by: Jiri Hnidek <jhnidek@redhat.com>

Author: jirihnidek

features/productid.feature

@@ -7,3 +7,6 @@ Feature: Management of productid certificate by libdnf5 plugin productid
       | never-enabled-content-37060 |
     When rpm "slow-eagle" is installed from RPM repository
     Then productid certificate "37060.pem" is installed in "/etc/pki/product"
+    And product database contains
+      | repo_id                     | product_id |
+      | never-enabled-content-37060 | 37060      |

features/steps/test_impl_productid.py

@@ -4,6 +4,7 @@
 import os
 import json
 
+
 def run(cmd, shell=True, cwd=None):
     """
     Run a command.
@@ -35,7 +36,7 @@ def run_in_context(context, cmd, can_fail=False, expected_exit_code=None, **run_
     :return: None
     """
     if getattr(context, "faketime", None) is not None:
-        cmd = 'NO_FAKE_STAT=1 ' + context.faketime + cmd
+        cmd = "NO_FAKE_STAT=1 " + context.faketime + cmd
 
     if getattr(context, "fake_kernel_release", None) is not None:
         cmd = context.fake_kernel_release + cmd
@@ -45,19 +46,23 @@ def run_in_context(context, cmd, can_fail=False, expected_exit_code=None, **run_
 
     context.cmd = cmd
 
-    if hasattr(context.scenario, "working_dir") and 'cwd' not in run_args:
-        run_args['cwd'] = context.scenario.working_dir
+    if hasattr(context.scenario, "working_dir") and "cwd" not in run_args:
+        run_args["cwd"] = context.scenario.working_dir
 
     context.cmd_exitcode, context.cmd_stdout, context.cmd_stderr = run(cmd, **run_args)
 
     if not can_fail and context.cmd_exitcode != 0:
-        raise AssertionError('Running command "%s" failed: %s' % (cmd, context.cmd_exitcode))
+        raise AssertionError(
+            'Running command "%s" failed: %s' % (cmd, context.cmd_exitcode)
+        )
     elif expected_exit_code is not None and expected_exit_code != context.cmd_exitcode:
         raise AssertionError(
-            'Running command "%s" had unexpected exit code: %s' % (cmd, context.cmd_exitcode)
+            'Running command "%s" had unexpected exit code: %s'
+            % (cmd, context.cmd_exitcode)
         )
 
-@given('system is registered against candlepin server')
+
+@given("system is registered against candlepin server")
 def step_impl(context):
     """
     Check if the system is registered against the candlepin server. If not, register it.
@@ -71,7 +76,8 @@ def step_impl(context):
         cmd = "rhc connect --username admin --password admin --organization donaldduck"
         run_in_context(context, cmd, can_fail=False)
 
-@given('repositories are enabled')
+
+@given("repositories are enabled")
 def step_impl(context):
     """
     Enable repositories for the system.
@@ -82,6 +88,7 @@ def step_impl(context):
         cmd = f"dnf5 config-manager setopt {row['repo_id']}.enabled=1"
         run_in_context(context, cmd, can_fail=False)
 
+
 @when('rpm "{rpm_name}" is installed from RPM repository')
 def step_impl(context, rpm_name):
     """
@@ -93,7 +100,10 @@ def step_impl(context, rpm_name):
     cmd = f"dnf5 install -y {rpm_name}"
     run_in_context(context, cmd, can_fail=False)
 
-@then('productid certificate "{product_cert_name}" is installed in "{product_cert_dir_path}"')
+
+@then(
+    'productid certificate "{product_cert_name}" is installed in "{product_cert_dir_path}"'
+)
 def step_impl(context, product_cert_name, product_cert_dir_path):
     """
     Check if the productid certificate is installed in the specified directory.
@@ -103,4 +113,29 @@ def step_impl(context, product_cert_name, product_cert_dir_path):
     :return: None
     """
     product_cert_path = os.path.join(product_cert_dir_path, product_cert_name)
-    assert os.path.exists(product_cert_path)
+    assert os.path.exists(product_cert_path)
+
+
+PRODUCTDB_PATH = "/var/lib/rhsm/productid.json"
+
+
+@then("product database contains")
+def step_impl(contex):
+    """
+    Check if the product database contains the expected products and repositories.
+    :param contex: behave context
+    :return: None
+    """
+    productdb_content = None
+    assert os.path.exists(PRODUCTDB_PATH)
+    with open(PRODUCTDB_PATH) as f:
+        productdb_content = json.loads(f.read())
+
+    assert productdb_content is not None
+    assert isinstance(productdb_content, dict)
+    for row in contex.table:
+        product_id = row["product_id"]
+        repo_id = row["repo_id"]
+        assert product_id in productdb_content
+        repo_list = productdb_content[product_id]
+        assert repo_id in repo_list