Add alias for sqlcmd when installed on linux

PiperOrigin-RevId: 385205457

Author: raymond13513

perfkitbenchmarker/linux_packages/mssql_tools.py

@@ -43,3 +43,6 @@ def AptInstall(vm):
                                                  location=_DEB_FILE_LOCATION))
   vm.RemoteCommand('sudo apt-get update')
   vm.RemoteCommand('sudo ACCEPT_EULA=Y /usr/bin/apt-get -y install mssql-tools')
+  vm.RemoteCommand('echo "alias sqlcmd=\'/opt/mssql-tools/bin/sqlcmd\'"'
+                   ' >> ~/.bashrc')
+  vm.RemoteCommand('source ~/.bashrc')

perfkitbenchmarker/sql_engine_utils.py

@@ -199,7 +199,7 @@ def InstallPackages(self):
 
   def MakeSqlCommand(self, command: str, database_name=''):
     """See base class."""
-    sqlserver_command = '/opt/mssql-tools/bin/sqlcmd -S %s -U %s -P %s ' % (
+    sqlserver_command = 'sqlcmd -S %s -U %s -P %s ' % (
         self.connection_properties.endpoint,
         self.connection_properties.database_username,
         self.connection_properties.database_password)

tests/relational_db_test.py

@@ -184,7 +184,7 @@ def testMakeSqlserverCommand(self):
     db.endpoint = '1.1.1.1'
     self.assertEqual(
         db.client_vm_query_tools.MakeSqlCommand('Select 1'),
-        '/opt/mssql-tools/bin/sqlcmd -S 1.1.1.1 -U root -P perfkitbenchmarker -Q "Select 1"'
+        'sqlcmd -S 1.1.1.1 -U root -P perfkitbenchmarker -Q "Select 1"'
     )
 
   def testMakeSqlserverCommandWithLocalHost(self):
@@ -195,7 +195,7 @@ def testMakeSqlserverCommandWithLocalHost(self):
     db.endpoint = '1.1.1.1'
     self.assertEqual(
         db.server_vm_query_tools.MakeSqlCommand('Select 1'),
-        '/opt/mssql-tools/bin/sqlcmd -S localhost -U root -P perfkitbenchmarker -Q "Select 1"'
+        'sqlcmd -S localhost -U root -P perfkitbenchmarker -Q "Select 1"'
     )
 
   def testInstallMYSQLServer(self):