def test_equals(self):
self.assertRegex(psmin.main(["psminifier.py"], file='$a = "b"'),
'^\$[a-z]="b"$')
self.assertRegex(psmin.main(["psminifier.py"], file='$a = "b"'),
'^\$[a-z]="b"$')
self.assertRegex(psmin.main(["psminifier.py"], file='$a = "b"'),
'^\$[a-z]="b"$')
def test_input_output(self):
# Test stdin and stdout
proc = sp.Popen(["python3", "ps_minifier/psminifier.py"],
stdout=sp.PIPE,
stdin=sp.PIPE)
resp = (proc.communicate(input="$a = 2".encode()))[0].decode()
self.assertRegex(resp, "^==RESULT==[\r\n]+\$[a-z]=2[\r\n]*$")
# Test file input and file output
with open("test_ps.txt", "w") as f:
f.write("$a = 1;\n$a = $a * 2;")
psmin.main(
["psminifier.py", "-f", "test_ps.txt", "-o", "test_ps_out.txt"])
with open("test_ps_out.txt", "r") as f:
self.assertRegex(f.read(), "^(\$[a-z])=1;\\1=\\1\*2;[\r\n]*$")
os.remove("test_ps_out.txt")
os.remove("test_ps.txt")
def test_variable_replacement(self):
self.assertRegex(
psmin.main(["psminifier.py"],
file='$a = "hello there!";\n$b = "hi";\n$a="hey";'),
'^({VR})="hello there!";({VR})="hi";\\1="hey";$'.format(
VR=psmin.VAR_REGEX))
self.assertRegex(
psmin.main(["psminifier.py"],
file='$a1 = "hello there!";\n$b = "hi";\n$a1="hey";'),
'^({VR})="hello there!";({VR})="hi";\\1="hey";$'.format(
VR=psmin.VAR_REGEX))
self.assertRegex(
psmin.main(["psminifier.py"],
file='$a1A = "hello there!";\n$b = "hi";\n$a1A="hey";'),
'^({VR})="hello there!";({VR})="hi";\\1="hey";$'.format(
VR=psmin.VAR_REGEX))
self.assertRegex(
psmin.main(["psminifier.py"],
file='$1a = "hello there!";\n$b2 = "hi";\n$1a="hey";'),
'^({VR})="hello there!";({VR})="hi";\\1="hey";$'.format(
VR=psmin.VAR_REGEX))
self.assertRegex(
psmin.main(["psminifier.py"],
file='$a1A = "hello there!";\n$b = "hi";\n$a1A="hey";'),
'^(\$[a-zA-Z])="hello there!";(\$[a-zA-Z])="hi";\\1="hey";$',
msg="The variable should be of length 1.")
self.assertEqual(psmin.main(["psminifier.py"], file="$true"), "$true")
self.assertEqual(psmin.main(["psminifier.py"], file="$false"),
"$false")
self.assertRegex(
psmin.main(["psminifier.py"],
file='$a12 = "y";\n$b3 = "x $a12 z";'),
'^({VR})="y";({VR})="x \\1 z";$'.format(VR=psmin.VAR_REGEX))
self.assertRegex(
psmin.main(["psminifier.py"],
file='$a12 = "y";\n$b3 = "x `$a12 z";'),
'^({VR})="y";({VR})="x `\$a12 z";$'.format(VR=psmin.VAR_REGEX))
def test_string_integrity(self):
self.assertRegex(
psmin.main(["psminifier.py"], file='$a = "hello there!"'),
'^\$[a-z]="hello there!"$')
def test_brackets_whitespace_removal(self):
self.assertEqual(
psmin.main(["psminifier.py"],
file='If ( 1 -eq 1 ) {\n\tWrite-Output "Test"\n}'),
'If(1 -eq 1){Write-Output "Test"}')
def test_maths_symbols(self):
self.assertEqual(
psmin.main(["psminifier.py"], file="Get-Command *-Service;"),
"Get-Command *-Service;")