Check for exact match for profile

__snapshots__/test.js.snap

@@ -34,6 +34,19 @@ const b = () => {
 };"
 `;
 
+exports[`does not wrap a function if comment does not exactly match 1`] = `
+"
+const b = () => {
+	// wraps a profile
+	return 42;
+};
+
+const a = () => {
+	// bug in something else called profile
+	return b();
+};"
+`;
+
 exports[`has multiple comments 1`] = `
 "
 const b = () => {

index.js

@@ -1,5 +1,7 @@
 const t = require('babel-types');
 
+const SPACE_REGEX = /\s+/;
+
 const BlockVisitor = function (path, args) {
 	const body = path.node.body || [];
 
@@ -16,9 +18,14 @@ const BlockVisitor = function (path, args) {
 				for (var j = 0, length = allComments.length; j < length; j++) {
 					const cNode = allComments[j];
 
-					if (cNode.type === 'CommentLine' && (cNode.value || '').indexOf('profile') > -1) {
-						args.state.gotProfileComment = true;
-						args.state.path = path;
+					if (cNode.type === 'CommentLine') {
+						const value = (cNode.value || '').replace(SPACE_REGEX, '');
+
+						if (value === 'profile') {
+							args.state.gotProfileComment = true;
+							args.state.path = path;
+						}
+
 						return;
 					}
 				}

test.js

@@ -21,6 +21,23 @@ it('wraps a function in profile block', () => {
 	expect(code).toMatchSnapshot();
 });
 
+it('does not wrap a function if comment does not exactly match', () => {
+	const example = `
+	const b = () => {
+		// wraps a profile
+		return 42;
+	}
+
+	const a = () => {
+		// bug in something else called profile
+		return b();
+	}
+	`;
+
+	const code = babel.transform(example, { plugins: [ plugin ] }).code;
+	expect(code).toMatchSnapshot();
+});
+
 it('caters to the case of multiple returns', () => {
 	const example = `
 	const a = () => {