changing nslookup for assertions and adding hostname method

Author: Dan Lavu

sssd_test_framework/utils/network.py

@@ -40,6 +40,22 @@ def ip(self, name: str | None = None) -> IPUtils:
 
         return inst
 
+    def hostname(self, hostname: str | None = None) -> str:
+        """
+         Get or change hostname.
+
+        :param hostname: Hostname.
+        :type hostname: str
+        :return: Configured hostname.
+        :rtype: str
+        """
+        if hostname is not None:
+            _hostname = self.host.conn.run("hostname").stdout
+            self.__rollback.append(f"hostname {_hostname}")
+            self.host.conn.run(f"hostname {hostname}")
+
+        return self.host.conn.run("hostname").stdout
+
     def tcpdump(self, pcap_path: str, args: list[Any] | None = None) -> SSHKillableProcess:
         """
         Run tcpdump. The packets are captured in ``pcap_path``.
@@ -136,17 +152,17 @@ def dig(self, address: str, server: str = "", reverse: bool = False) -> dict[str
 
         return None if not records else records[0]
 
-    def nslookup(self, args: list[str]) -> ProcessResult:
+    def nslookup(self, args: list[str]) -> int:
         """
         Execute nslookup command with given arguments.
 
         :param args: Arguments to ``nslookup``, defaults to None
         :type args: list[str]
-        :return: SSH Process result
-        :rtype: ProcessResult
+        :return: Return code
+        :rtype: int
         """
 
-        return self.host.conn.exec(["nslookup", *args], raise_on_error=False)
+        return self.host.conn.exec(["nslookup", *args], raise_on_error=False).rc
 
     def teardown(self):
         """