Add test for if/switch conditions

These test for nested block expression with `return`s where we need to
add the profileEnd function in each of the branches to ensure we stop
the profiling no matter which branch is taken when the code gets
executed.

__snapshots__/test.js.snap

@@ -82,6 +82,45 @@ const b = () => {
 };"
 `;
 
+exports[`maintains indentation when for if blocks 1`] = `
+"
+const b = () => {
+	console.profile('b');
+
+	// profile
+
+	if (true) {
+		console.profileEnd();
+
+		return 12;
+	}
+
+	switch (true) {
+		case 'case1':
+			console.profileEnd();
+
+			return 'what';
+		case 'case 2':
+			{
+				console.profileEnd();
+
+				return 'where';
+			}
+		case 'case 3':
+			console.profileEnd();
+
+			return 'who';
+		default:
+			console.profileEnd();
+
+			return 'default';
+	}
+
+	console.profileEnd();
+	return 42;
+};"
+`;
+
 exports[`maintains indentation when used with iterators 1`] = `
 "
 const b = () => {

test.js

@@ -141,3 +141,32 @@ it('maintains indentation when used with iterators', () => {
 	expect(code).toMatchSnapshot();
 });
 
+it('maintains indentation when for if blocks', () => {
+	const example = `
+	const b = () => {
+		// profile
+		
+		if (true) {
+			return 12;
+		}
+
+		switch (true) {
+			case 'case1':
+				return 'what';
+			case 'case 2': {
+				return 'where';
+			}
+			case 'case 3':
+				return 'who';
+			default:
+				return 'default'
+		}
+
+		return 42;
+	};
+	`;
+
+	const code = babel.transform(example, { plugins: [ plugin ] }).code;
+	expect(code).toMatchSnapshot();
+});
+