SetUsageLocation.ps1

#requires -Modules AzureAD

$UsageLocation = Get-AutomationVariable -Name "UsageLocation" -ErrorAction Stop

$servicePrincipalConnection = Get-AutomationConnection -Name "AzureRunAsConnection" -ErrorAction Stop

Connect-AzureAD -TenantId $servicePrincipalConnection.TenantId -ApplicationId $servicePrincipalConnection.ApplicationId `
-CertificateThumbprint $servicePrincipalConnection.CertificateThumbprint

Get-AzureADUser | Where-Object{$_.UsageLocation -ne $UsageLocation} | ForEach-Object {

    Write-Output "Set UsageLocation $UsageLocation for User $($_.UserPrincipalName)"

    Set-AzureADUser -ObjectID $_.ObjectID -UsageLocation $UsageLocation -Verbose
}