Add test for promptLabel as a functioN

Author: astralarya

src/react-console.tsx

@@ -104,10 +104,10 @@ export interface LogEntry {
 }
 
 export interface ConsoleProps{
-	handler(command: string): any;
-	cancel?(): any;
-	complete?(words: string[], curr: number, promptText: string): string[];
-	continue?(promptText: string): boolean;
+	handler: (command: string)=>any;
+	cancel?: ()=>any;
+	complete?: (words: string[], curr: number, promptText: string)=>string[];
+	continue?: (promptText: string)=>boolean;
 	autofocus?: boolean;
 	promptLabel?: string | (()=>string);
 	welcomeMessage?: string;
@@ -168,7 +168,6 @@ export default class extends React.Component<ConsoleProps,ConsoleState> {
 		promptLabel: '> ',
 		continue: function() { return false; },
 		cancel: function() {},
-		handler: function() { this.return() },
 	};
 	child: {
 		typer?: HTMLTextAreaElement;
@@ -190,12 +189,16 @@ export default class extends React.Component<ConsoleProps,ConsoleState> {
 			log: log,
 		}, this.scrollIfBottom() );
 	}
-	return = () => {
+	done = () => {
 		this.setState({
 			acceptInput: true,
 			currLabel: this.nextLabel(),
 		}, this.scrollIfBottom() );
 	}
+	return = () => {
+		console.log("WARNING: return() is deprecated.  Use done() or the callback (2nd arg of handler) instead.");
+		this.done();
+	}
 	// Component Lifecycle
 	componentDidMount() {
 		if(this.props.autofocus) {
@@ -528,7 +531,11 @@ export default class extends React.Component<ConsoleProps,ConsoleState> {
 				lastCommand: ConsoleCommand.Default,
 			}, () => {
 				this.scrollToBottom();
-				this.props.handler(command);
+				if(this.props.handler) {
+					this.props.handler(command)
+				} else {
+					this.done();
+				}
 			});
 		}
 	}

test/test.tsx

@@ -138,6 +138,22 @@ describe('<Console />', function() {
 			var wrapper = enzyme.mount(<Console promptLabel='ababa: '/>);
 			expect(wrapper.find('.react-console-prompt-label').text()).equals('ababa: ');
 		});
+		it('Calls function when passed as promptLabel and uses returned values as labels', function() {
+			var count = 0;
+			function counter() {
+				return count++;
+			}
+			var wrapper = enzyme.mount(<Console promptLabel={counter}/>);
+			var typer = wrapper.find('.react-console-typer');
+			typer.simulate('keyDown', { keyCode: 13 /* Return */ });
+			typer.simulate('keyDown', { keyCode: 13 /* Return */ });
+			typer.simulate('keyDown', { keyCode: 13 /* Return */ });
+			var labels = wrapper.find('.react-console-prompt-label');
+			expect(labels).length(4);
+			expect(count).equals(4);
+			expect(labels.first().text()).equals('0');
+			expect(labels.last().text()).equals('3');
+		});
 	});
 	describe('[Property] welcomeMessage: ', function () {
 		it('Doesn\'t have class `react-console-welcome` when welcomeMessage undefined', function() {