ISHBootstrap

2.0.0

Private/Set-ISHDatabaseCrawlerRegistration.Tests.ps1

<#
# Copyright (c) 2021 All Rights Reserved by the RWS Group for and on behalf of its affiliates and subsidiaries.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#>

$here = Split-Path -Parent $MyInvocation.MyCommand.Path
$sut = (Split-Path -Leaf $MyInvocation.MyCommand.Path) -replace '\.Tests\.', '.'
. "$here\$sut"
. "$here\New-PSCredential.ps1"
. "$here\New-SqlServerQuerySplat.ps1"

$server = 'Server' + (Get-random)
$database = 'Database' + (Get-random)
$username = 'Username' + (Get-random)
$password = 'Password' + (Get-random)

Describe "Set-ISHDatabaseCrawlerRegistration" {
    Mock "Invoke-ISHMaintenance" {

    }
    Mock "New-SqlServerQuerySplat" {
        @{
            Server     = $server
            Database   = $database
            Credential = New-PSCredential -Username $username -Password $password
        }
    }
    It "Set-ISHDatabaseCrawlerRegistration - More than 2 registrations and 0 valid" {
        Mock "Invoke-SqlServerQuery" {
            if ($Sql -like "*WHERE*") {
                0
            }
            else {
                2
            }
        }
        Set-ISHDatabaseCrawlerRegistration
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1 -ParameterFilter {
            $Crawler -and $UnRegisterAll
        }
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1 -ParameterFilter {
            $Crawler -and $Register
        }
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 2
    }
    It "Set-ISHDatabaseCrawlerRegistration - More than 2 registrations and 1 valid" {
        Mock "Invoke-SqlServerQuery" {
            if ($Sql -like "*WHERE*") {
                1
            }
            else {
                2
            }
        }
        Set-ISHDatabaseCrawlerRegistration
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1 -ParameterFilter {
            $Crawler -and $UnRegisterAll
        }
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1 -ParameterFilter {
            $Crawler -and $Register
        }
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 2
    }
    It "Set-ISHDatabaseCrawlerRegistration - Exactly 1 registration and 0 valid" {
        Mock "Invoke-SqlServerQuery" {
            if ($Sql -like "*WHERE*") {
                0
            }
            else {
                1
            }
        }
        Set-ISHDatabaseCrawlerRegistration
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1 -ParameterFilter {
            $Crawler -and $UnRegisterAll
        }
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1 -ParameterFilter {
            $Crawler -and $Register
        }
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 2
    }
    It "Set-ISHDatabaseCrawlerRegistration - Exactly 1 registration and 1 valid" {
        Mock "Invoke-SqlServerQuery" {
            if ($Sql -like "*WHERE*") {
                1
            }
            else {
                1
            }
        }
        Set-ISHDatabaseCrawlerRegistration
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 0
    }
    It "Set-ISHDatabaseCrawlerRegistration - Exactly 0 registration" {
        Mock "Invoke-SqlServerQuery" {
            if ($Sql -like "*WHERE*") {
                0
            }
            else {
                0
            }
        }
        Set-ISHDatabaseCrawlerRegistration
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1 -ParameterFilter {
            $Crawler -and $Register
        }
        Assert-MockCalled -CommandName "Invoke-ISHMaintenance" -Scope It -Exactly 1
    }
}