Unit test

Add test for firewall service