Clean up code sun component tests
Added tests for switch component