old_contents: #!/usr/bin/python
  
  from azuremodules import *
  
  import argparse
  import sys
  import time
  import platform
  import os
  
  parser = argparse.ArgumentParser()
  
  parser.add_argument('-p', '--passwd', help='specify password of vm', required=True)
  args = parser.parse_args()
  
  passwd = args.passwd
  
  def RunTest():
      UpdateState("TestRunning")
      RunLog.info("Checking log waagent.log...")
      output = Run("grep -i 'iptables -I INPUT -p udp --dport' /var/log/waagent* | wc -l | tr -d '\n'")
      if not (output == "0") :
          RunLog.info('The log file contains the verbose logs')
          ResultLog.info('PASS')
          UpdateState("TestCompleted")
      else :
          RunLog.error('Verify waagent.log fail, the log file does not contain the verbose logs')
          ResultLog.error('FAIL')
          UpdateState("TestCompleted")
  
  def Restartwaagent():
      distro = platform.dist()
      if (distro[0] == "CoreOS") :
          Run("echo '"+passwd+"' | sudo -S sed -i s/Logs.Verbose=n/Logs.Verbose=y/g  /usr/share/oem/waagent.conf")
      else :
          Run("echo '"+passwd+"' | sudo -S sed -i s/Logs.Verbose=n/Logs.Verbose=y/g  /etc/waagent.conf")
      RunLog.info("Restart waagent service...")
      result = Run("echo '"+passwd+"' | sudo -S find / -name systemctl |wc -l | tr -d '\n'")    
      if (distro[0] == "Ubuntu") or (distro[0] == "debian"):
          Run("echo '"+passwd+"' | sudo -S service walinuxagent restart")
      else :  
          if (result == "0") :
              os.system("echo '"+passwd+"' | sudo -S service waagent restart")
          else :
              os.system("echo '"+passwd+"' | sudo -S systemctl restart waagent")
      time.sleep(60)
  
  Restartwaagent()
  RunTest()
  
new_contents: #!/usr/bin/python
  
  from azuremodules import *
  
  import argparse
  import sys
  import time
  import platform
  import os
  
  parser = argparse.ArgumentParser()
  
  parser.add_argument('-p', '--passwd', help='specify password of vm', required=True)
  args = parser.parse_args()
  
  passwd = args.passwd
  distro = platform.dist()
  
  def RunTest():
      UpdateState("TestRunning")
      if(distro[0] == "CoreOS"):
          versionOutPut = Run("waagent --version")
      else:
          output = Run("ps aux | grep waagent | grep python | grep -v 'ps aux | grep waagent | grep python'")
          if ("python3" in output) :
              versionOutPut = Run("/usr/bin/python3 /usr/sbin/waagent --version")
          else :
              versionOutPut = Run("/usr/sbin/waagent --version")
  
      RunLog.info("Checking log waagent.log...")
      if("2.0." in versionOutPut):
          output = Run("grep -i 'iptables -I INPUT -p udp --dport' /var/log/waagent* | wc -l | tr -d '\n'")
          RunLog.info("agent version is 2.0")
      else:
          output = Run("grep -i 'VERBOSE HTTP response status' /var/log/waagent* | wc -l | tr -d '\n'")
          RunLog.info("agent version > 2.0")
  
      if not (output == "0") :
          RunLog.info('The log file contains the verbose logs')
          ResultLog.info('PASS')
          UpdateState("TestCompleted")
      else :
          RunLog.error('Verify waagent.log fail, the log file does not contain the verbose logs')
          ResultLog.error('FAIL')
          UpdateState("TestCompleted")
  
  def Restartwaagent():
      if (distro[0] == "CoreOS") :
          Run("echo '"+passwd+"' | sudo -S sed -i s/Logs.Verbose=n/Logs.Verbose=y/g  /usr/share/oem/waagent.conf")
      else :
          Run("echo '"+passwd+"' | sudo -S sed -i s/Logs.Verbose=n/Logs.Verbose=y/g  /etc/waagent.conf")
      RunLog.info("Restart waagent service...")
      result = Run("echo '"+passwd+"' | sudo -S find / -name systemctl |wc -l | tr -d '\n'")    
      if (distro[0] == "Ubuntu") or (distro[0] == "debian"):
          Run("echo '"+passwd+"' | sudo -S service walinuxagent restart")
      else :  
          if (result == "0") :
              os.system("echo '"+passwd+"' | sudo -S service waagent restart")
          else :
              os.system("echo '"+passwd+"' | sudo -S systemctl restart waagent")
      time.sleep(60)
  
  Restartwaagent()
  RunTest()
  
subject: update for agent 2.1/2.2