From 99cbe967054d2778a5ca7a98a3b9f062a8084904 Mon Sep 17 00:00:00 2001
From: Wes Carroll <wcarroll@oceanbeachlife.com>
Date: Fri, 29 Mar 2019 17:49:09 -0400
Subject: [PATCH] Update Add-ZertoPeerSite.Tests.ps1

---
 Tests/Public/Add-ZertoPeerSite.Tests.ps1 | 62 ++++++++++++++++--------
 1 file changed, 42 insertions(+), 20 deletions(-)

diff --git a/Tests/Public/Add-ZertoPeerSite.Tests.ps1 b/Tests/Public/Add-ZertoPeerSite.Tests.ps1
index 2d8dbc5..28a1e43 100644
--- a/Tests/Public/Add-ZertoPeerSite.Tests.ps1
+++ b/Tests/Public/Add-ZertoPeerSite.Tests.ps1
@@ -1,23 +1,45 @@
-$moduleFileName = "ZertoApiWrapper.psm1"
-$filePath = (Split-Path -Parent $MyInvocation.MyCommand.Path) -replace 'Tests', 'ZertoApiWrapper'
-$fileName = (Split-Path -Leaf $MyInvocation.MyCommand.Path ) -replace '.Tests.', '.'
-$commandName = $fileName -replace '.ps1', ''
-$modulePath = $filePath -replace "Public", ""
-Import-Module $modulePath\$moduleFileName -Force
+#Requires -Modules Pester
+$here = (Split-Path -Parent $MyInvocation.MyCommand.Path).Replace("Tests", "ZertoApiWrapper")
+$sut = (Split-Path -Leaf $MyInvocation.MyCommand.Path).Replace(".Tests.", ".")
+$file = Get-ChildItem "$here\$sut"
+#$moduleFileName = "ZertoApiWrapper.psd1"
+#$filePath = (Split-Path -Parent $MyInvocation.MyCommand.Path) -replace 'Tests', 'ZertoApiWrapper'
+#$fileName = (Split-Path -Leaf $MyInvocation.MyCommand.Path ) -replace '.Tests.', '.'
+#$commandName = $fileName -replace '.ps1', ''
+#Get-Module -Name ZertoApiWrapper | Remove-Module -Force
+#Import-Module D:\GitHub\ZertoApiWrapper\ZertoApiWrapper\ZertoApiWrapper.psd1 -force
 
-$userName = "zerto\build"
-$password = ConvertTo-SecureString -String "ZertoBuild" -AsPlainText -Force
-$credential = New-Object -TypeName System.Management.Automation.PSCredential($userName, $password)
+Describe $file.BaseName -Tag 'Unit' {
+    Context "Basic Features" {
+        Mock -ModuleName ZertoApiWrapper Invoke-ZertoRestRequest {
+            return "9a49f42e-2bbd-4bf8-b571-77908a2e5e98.928a122b-1763-4664-ad37-cc00bb883f2f"
+        }
+        It "is valid Powershell (Has no script errors)" {
+            $contents = Get-Content -Path $file -ErrorAction Stop
+            $errors = $null
+            $null = [System.Management.Automation.PSParser]::Tokenize($contents, [ref]$errors)
+            $errors | Should -HaveCount 0
+        }
 
-# $credential = Import-Clixml -Path C:\ZertoScripts\Creds.xml
-$zertoServer = "192.168.1.100"
-$zertoPort = "7669"
-
-Describe "$commandName" {
-    it "file should exist" {
-        "$filePath\$fileName" | should exist
+        it "Should be return a string value" {
+            $results = Add-ZertoPeerSite -targetHost '192.168.1.100' -targetPort '9081'
+            $results | should not benullorempty
+            $results.gettype().name | should -be "String"
+        }
+        it "Should not take a non-int as a port" {
+            {Add-ZertoPeerSite -targetHost '192.168.1.100' -targetPort 'string'} | should -Throw
+        }
+        it "Should not take a non-ip address as a 'TargetHost'" {
+            {Add-ZertoPeerSite -targetHost 'MyZVMHost' -targetPort '9081'} | should -Throw
+            {Add-ZertoPeerSite -targetHost '192.168.1.266' -targetPort '9081'} | should -Throw
+            {Add-ZertoPeerSite -targetHost '192.168.1' -targetPort '9081'} | should -Throw
+        }
+        it "Should not require a target port to be defined" {
+            Add-ZertoPeerSite -targetHost '192.168.1.100'
+        }
+        it "Should support '-whatif'" {
+            {Add-ZertoPeerSite -targetHost '192.168.1.100' -WhatIf | out-null} | Should -Not -Throw
+        }
     }
-    it "module should have a function called $commandName" {
-        get-command $commandName | should be $true
-    }
-}
\ No newline at end of file
+}
+