-
Notifications
You must be signed in to change notification settings - Fork 100
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implemented cChocoFeature to provide an interface for DSC to configure choco client features.
- Loading branch information
1 parent
c7d4976
commit 5b57ef4
Showing
6 changed files
with
287 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,2 +1,3 @@ | ||
*.mof | ||
!*.schema.mof | ||
!*.schema.mof | ||
TestsResults.xml |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,148 @@ | ||
# Copyright (c) 2017 Chocolatey Software, Inc. | ||
# | ||
# 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. | ||
|
||
<# | ||
.Description | ||
Returns the configuration for cChocoFeature. | ||
.Example | ||
Get-TargetResource -FeatureName allowGlobalConfirmation -Ensure 'Present' | ||
#> | ||
function Get-TargetResource | ||
{ | ||
[CmdletBinding()] | ||
[OutputType([System.Collections.Hashtable])] | ||
param | ||
( | ||
[parameter(Mandatory = $true)] | ||
[System.String] | ||
$FeatureName, | ||
|
||
[ValidateSet('Present','Absent')] | ||
[System.String] | ||
$Ensure='Present' | ||
) | ||
|
||
Write-Verbose "Starting cChocoFeature Get-TargetResource - Feature Name: $FeatureName, Ensure: $Ensure" | ||
|
||
$returnValue = @{ | ||
FeatureName = $FeatureName | ||
Ensure = $Ensure | ||
} | ||
|
||
$returnValue | ||
|
||
} | ||
|
||
<# | ||
.Description | ||
Performs the set for the cChocoFeature resource. | ||
.Example | ||
Get-TargetResource -FeatureName allowGlobalConfirmation -Ensure 'Present' | ||
#> | ||
function Set-TargetResource | ||
{ | ||
[CmdletBinding(SupportsShouldProcess=$true)] | ||
param | ||
( | ||
[parameter(Mandatory = $true)] | ||
[System.String] | ||
$FeatureName, | ||
|
||
[ValidateSet('Present','Absent')] | ||
[string] | ||
$Ensure='Present' | ||
) | ||
|
||
|
||
Write-Verbose "Starting cChocoFeature Set-TargetResource - Feature Name: $FeatureName, Ensure: $Ensure." | ||
|
||
if ($pscmdlet.ShouldProcess("Choco feature $FeatureName will be ensured $Ensure.")) | ||
{ | ||
if ($Ensure -eq 'Present') | ||
{ | ||
Write-Verbose "Enabling choco feature $FeatureName." | ||
choco feature enable -n $FeatureName | ||
} | ||
else | ||
{ | ||
Write-Verbose "Disabling choco feature $FeatureName." | ||
choco feature disable -n $FeatureName | ||
} | ||
} | ||
|
||
} | ||
|
||
<# | ||
.Description | ||
Performs the test for cChocoFeature. | ||
.Example | ||
Test-TargetResource -FeatureName allowGlobalConfirmation -Ensure 'Present' | ||
#> | ||
function Test-TargetResource | ||
{ | ||
[CmdletBinding()] | ||
[OutputType([System.Boolean])] | ||
param | ||
( | ||
[parameter(Mandatory = $true)] | ||
[System.String] | ||
$FeatureName, | ||
|
||
[ValidateSet('Present','Absent')] | ||
[System.String] | ||
$Ensure='Present' | ||
) | ||
|
||
Write-Verbose "Starting cChocoFeature Test-TargetResource - Feature Name: $FeatureName, Ensure: $Ensure." | ||
|
||
$result = $false | ||
$feature = Get-ChocoFeature -FeatureName $FeatureName | Where-Object {$_.State -eq "Enabled"} | ||
|
||
if (($Ensure -eq 'Present' -and ([bool]$feature)) -or ($Ensure -eq 'Absent' -and !([bool]$feature))) | ||
{ | ||
Write-Verbose "Test-TargetResource is true, $FeatureName is $Ensure." | ||
$result = $true | ||
} | ||
else | ||
{ | ||
Write-Verbose "Test-TargetResource is false, $FeatureName is not $Ensure." | ||
} | ||
|
||
return $result | ||
|
||
} | ||
|
||
<# | ||
.Description | ||
Query chocolatey features. | ||
#> | ||
function Get-ChocoFeature | ||
{ | ||
[OutputType([PSCustomObject])] | ||
param( | ||
[string] | ||
$FeatureName | ||
) | ||
choco feature -r | ConvertFrom-Csv -Delimiter "|" -Header Name, State, Description | Where-Object {$_.Name -eq $FeatureName} | ||
} | ||
|
||
|
||
|
||
|
||
Export-ModuleMember -Function *-TargetResource | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
|
||
[ClassVersion("1.0.0.0"), FriendlyName("cChocoFeature")] | ||
class cChocoFeature : OMI_BaseResource | ||
{ | ||
[Key] String FeatureName; | ||
[Write,ValueMap{"Present", "Absent"},Values{"Present", "Absent"}] String Ensure; | ||
}; | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,41 @@ | ||
# Copyright (c) 2017 Chocolatey Software, Inc. | ||
# | ||
# 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. | ||
|
||
|
||
configuration ChocoFeatures { | ||
|
||
Import-DscResource -ModuleName cChoco | ||
|
||
Node 'localhost' { | ||
|
||
cChocoFeature allowGlobalConfirmation { | ||
|
||
FeatureName = "allowGlobalConfirmation" | ||
Ensure = 'Present' | ||
|
||
} | ||
|
||
cChocoFeature powershellHost { | ||
|
||
FeatureName = "powershellHost" | ||
Ensure = 'Absent' | ||
} | ||
} | ||
|
||
} | ||
|
||
|
||
$config = ChocoFeatures | ||
|
||
Start-DscConfiguration -Path $config.psparentpath -Wait -Verbose -Force |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,87 @@ | ||
# Copyright (c) 2017 Chocolatey Software, Inc. | ||
# | ||
# 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. | ||
|
||
|
||
$ResourceName = ((Split-Path $MyInvocation.MyCommand.Path -Leaf) -split '_')[0] | ||
$ResourceFile = (Get-DscResource -Name $ResourceName).Path | ||
|
||
$TestsPath = (split-path -path $MyInvocation.MyCommand.Path -Parent) | ||
$ResourceFile = Get-ChildItem -Recurse $TestsPath\.. -File | Where-Object {$_.name -eq "$ResourceName.psm1"} | ||
|
||
Import-Module -Name $ResourceFile.FullName | ||
|
||
|
||
#---------------------------------# | ||
# Pester tests for cChocoInstall # | ||
#---------------------------------# | ||
Describe "Testing cChocoFeature" { | ||
|
||
Context "Test-TargetResource" { | ||
|
||
mock -ModuleName cChocoFeature -CommandName Get-ChocoFeature -MockWith { | ||
@([pscustomobject]@{ | ||
Name = "allowGlobalConfirmation" | ||
State = "Enabled" | ||
Description = "blah" | ||
}, | ||
[pscustomobject]@{ | ||
Name = "powershellhost" | ||
State = "Disabled" | ||
Description = "blah" | ||
} )| Where-Object { $_.Name -eq $FeatureName } | ||
} -Verifiable | ||
|
||
|
||
it 'Test-TargetResource returns true when Present and Enabled.' { | ||
Test-TargetResource -FeatureName 'allowGlobalConfirmation' -Ensure 'Present' | should be $true | ||
} | ||
|
||
it 'Test-TargetResource returns false when Present and Disabled' { | ||
Test-TargetResource -FeatureName 'powershellhost' -Ensure 'Present' | should be $false | ||
} | ||
|
||
it 'Test-TargetResource returns false when Absent and Enabled' { | ||
Test-TargetResource -FeatureName 'allowGlobalConfirmation' -Ensure 'Absent' | Should be $false | ||
} | ||
|
||
it 'Test-TargetResource returns true when Absent and Disabled' { | ||
Test-TargetResource -FeatureName 'powershellhost' -Ensure 'Absent' | should be $true | ||
} | ||
|
||
} | ||
|
||
Context "Set-TargetResource" { | ||
|
||
InModuleScope -ModuleName cChocoFeature -ScriptBlock { | ||
function choco {} | ||
mock choco {} | ||
} | ||
|
||
Set-TargetResource -FeatureName "TestFeature" -Ensure "Present" | ||
|
||
it "Present - Should have called choco, with enable" { | ||
Assert-MockCalled -CommandName choco -ModuleName cChocoFeature -ParameterFilter { | ||
$args -contains "enable" | ||
} | ||
} | ||
|
||
Set-TargetResource -FeatureName "TestFeature" -Ensure "Absent" | ||
|
||
it "Absent - Should have called choco, with disable" { | ||
Assert-MockCalled -CommandName choco -ModuleName cChocoFeature -ParameterFilter { | ||
$args -contains "disable" | ||
} | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters