Skip to content

Instantly share code, notes, and snippets.

@markekraus
Created January 16, 2019 02:12
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save markekraus/5939881af81b9b1668c03c95eda55cc1 to your computer and use it in GitHub Desktop.
Save markekraus/5939881af81b9b1668c03c95eda55cc1 to your computer and use it in GitHub Desktop.
Busted Enumerator
class MyEnumerator : System.Collections.IEnumerator {
hidden static [bool]$toggle = $true
hidden static [bool]$toggle2 = $true
MyEnumerator (){}
[bool] MoveNext()
{
if([MyEnumerator]::toggle) {
[MyEnumerator]::toggle = $false
return $true
} else {
[MyEnumerator]::toggle = $true
return $false
}
}
[void] Reset()
{
[MyEnumerator]::toggle = $true
[MyEnumerator]::toggle2 = $true
}
[object] get_Current()
{
if([MyEnumerator]::toggle2)
{
[MyEnumerator]::toggle2 = $false
return [Object]$Null
} else {
[MyEnumerator]::toggle2 = $true
return 'a'
}
}
}
class MyClass : System.Collections.IEnumerable {
[System.Collections.IEnumerator] GetEnumerator()
{
return [MyEnumerator]::New()
}
MyClass(){
[MyEnumerator]::toggle = $true
[MyEnumerator]::toggle2 = $true
}
[int]$Count = 1
[int]$Length = 1
}
$A = [MyClass]::New()
foreach ($i in $A) {
$i
}
foreach ($i in $A) {
$i
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment