Graphite-Powershell.Tests.ps1

Import-Module .\Graphite-Powershell.psd1

InModuleScope Graphite-PowerShell {
    Describe "ConvertTo-GraphiteMetric" {
        Context "Metric Transformation - Base Function" {
            $TestMetric = ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec"

            It "Should Return Something" {
                $TestMetric | Should Not BeNullOrEmpty
            }
            It "Should Provide myServer.networkinterface.realtekpciegbefamilycontroller.bytesreceived-sec as Output" {
                $TestMetric | Should MatchExactly "myServer.networkinterface.realtekpciegbefamilycontroller.bytesreceived-sec"
            }
            It "Should Not Contain Left Parentheses" {
                $TestMetric | Should Not Match "\("
            }
            It "Should Not Contain Right Parentheses" {
                $TestMetric | Should Not Match "\)"
            }
            It "Should Not Contain Forward Slash" {
                $TestMetric | Should Not Match "\/"
            }
            It "Should Not Contain Back Slash" {
                $TestMetric | Should Not Match "\\"
            }
            It "Should Contain a Period" {
                $TestMetric | Should Match "\."
            }
        }
        Context "Metric Transformation - Using MetricReplacementHash" {

            $MockHashTable = [ordered]@{
            "^\\\\" = "";
            "\\\\" = "";
            "\/" = "-";
            ":" = ".";
            "\\" = ".";
            "\(" = ".";
            "\)" = "";
            "\]" = "";
            "\[" = "";
            "\%" = "";
            "\s+" = "";
            "\.\." = ".";
            "_" = ""
            }

            It "Should Return Something" {
                ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec" -MetricReplacementHash $MockHashTable | Should Not BeNullOrEmpty
            }
            It "Should Provide myServer.networkinterface.realtekpciegbefamilycontroller.bytesreceived-sec as Output" {
                ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec" -MetricReplacementHash $MockHashTable | Should MatchExactly "myServer.networkinterface.realtekpciegbefamilycontroller.bytesreceived-sec"
            }
            It "Should Not Contain Left Parentheses" {
                ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec" -MetricReplacementHash $MockHashTable | Should Not Match "\("
            }
            It "Should Not Contain Right Parentheses" {
                ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec" -MetricReplacementHash $MockHashTable | Should Not Match "\)"
            }
            It "Should Not Contain Forward Slash" {
                ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec" -MetricReplacementHash $MockHashTable | Should Not Match "\/"
            }
            It "Should Not Contain Back Slash" {
                ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec" -MetricReplacementHash $MockHashTable | Should Not Match "\\"
            }
            It "Should Contain a Period" {
                ConvertTo-GraphiteMetric -MetricToClean "\\myServer\network interface(realtek pcie gbe family controller)\bytes received/sec" -MetricReplacementHash $MockHashTable | Should Match "\."
            }
        }
        Context "Metric Transformation - Remove Underscores" {
            $TestMetric = ConvertTo-GraphiteMetric -MetricToClean "\\myServer\Processor(_Total)\% Processor Time" -RemoveUnderscores

            It "Should Return Something" {
                $TestMetric | Should Not BeNullOrEmpty
            }
            It "Should Return myServer.production.net.Processor.Total.ProcessorTime as Output" {
                $TestMetric | Should MatchExactly "myServer.Processor.Total.ProcessorTime"
            }
            It "Should Not Contain Underscores" {
                $TestMetric | Should Not Match "_"
            }
        }
        Context "Metric Transformation - Provide Nice Output for Physical Disks" {
            $TestMetric = ConvertTo-GraphiteMetric -MetricToClean "\\myServer\physicaldisk(1 e:)\avg. disk write queue length" -RemoveUnderscores -NicePhysicalDisks

            It "Should Return Something" {
                $TestMetric | Should Not BeNullOrEmpty
            }
            It "Should Return myServer.physicaldisk.e-drive.diskwritequeuelength as Output" {
                $TestMetric | Should MatchExactly "myServer.physicaldisk.e-drive.diskwritequeuelength"
            }
        }
        Context "Metric Transformation - Replace HostName" {
            $TestMetric = ConvertTo-GraphiteMetric -MetricToClean "\\$($env:COMPUTERNAME)\physicaldisk(1 e:)\avg. disk write queue length" -RemoveUnderscores -NicePhysicalDisks -HostName "my.new.hostname"

            It "Should Return Something" {
                $TestMetric | Should Not BeNullOrEmpty
            }
            It "Should Return my.new.hostname.physicaldisk.e-drive.diskwritequeuelength as Output" {
                $TestMetric | Should MatchExactly "my.new.hostname.physicaldisk.e-drive.diskwritequeuelength"
            }

            $TestMetric = ConvertTo-GraphiteMetric -MetricToClean "\\$($env:COMPUTERNAME)\physicaldisk(1 e:)\avg. disk write queue length" -RemoveUnderscores -NicePhysicalDisks -HostName "host_with_underscores"

            It "Should Return host_with_underscores.physicaldisk.e-drive.diskwritequeuelength as Output when RemoveUnderscores is enabled and host has underscores" {
                $TestMetric | Should MatchExactly "host_with_underscores.physicaldisk.e-drive.diskwritequeuelength"
            }
        }
    }

    Describe "Import-XMLConfig" {
        Context "Loading a Configuration File" {
            $_config = Import-XMLConfig -ConfigPath "./StatsToGraphiteConfig.xml"

            It "Loaded Configuration File Should Not Be Empty" {
                $_config | Should Not BeNullOrEmpty
            }
            It "Should Have 16 Properties" {
                $_config.Count | Should Be 17
            }
            It "SendUsingUDP should be Boolean" {
                $_config.SendUsingUDP -is [Boolean] | Should Be $true
            }
            It "MSSQLMetricSendIntervalSeconds should be Int32" {
                $_config.MSSQLMetricSendIntervalSeconds -is [Int32] | Should Be $true
            }
            It "MSSQLConnectTimeout should be Int32" {
                $_config.MSSQLConnectTimeout -is [Int32] | Should Be $true
            }
            It "MSSQLQueryTimeout should be Int32" {
                $_config.MSSQLQueryTimeout -is [Int32] | Should Be $true
            }
            It "MetricSendIntervalSeconds should be Int32" {
                $_config.MetricSendIntervalSeconds -is [Int32] | Should Be $true
            }
            It "MetricTimeSpan should be TimeSpan" {
                $_config.MetricTimeSpan -is [TimeSpan] | Should Be $true
            }
            It "MSSQLMetricTimeSpan should be TimeSpan" {
                $_config.MSSQLMetricTimeSpan -is [TimeSpan] | Should Be $true
            }
            It "MetricReplace should be HashTable" {
                $_config.MetricReplace -is [System.Collections.Specialized.OrderedDictionary] | Should Be $true
            }
        }
    }
}