return context object as result - fixes #37

index.js

@@ -74,8 +74,11 @@ class Listr {
 		return tasks
 			.then(() => {
 				this._renderer.end();
+
+				return context;
 			})
 			.catch(err => {
+				err.context = context;
 				this._renderer.end(err);
 				throw err;
 			});

readme.md

@@ -201,6 +201,9 @@ const tasks = new Listr([
 
 tasks.run({
 	foo: 'bar'
+}).then(ctx => {
+	console.log(ctx);
+	//=> {foo: 'bar', unicorn: 'rainbow'}
 });
 ```
 
@@ -322,7 +325,7 @@ Task object or multiple task objects.
 
 #### run([context])
 
-Start executing the tasks.
+Start executing the tasks. Returns a `Promise` for the context object.
 
 ##### context
 

test/test.js

@@ -128,7 +128,11 @@ test('context', async t => {
 		}
 	]);
 
-	await list.run();
+	const result = await list.run();
+
+	t.deepEqual(result, {
+		foo: 'bar'
+	});
 });
 
 test('subtask context', async t => {
@@ -155,5 +159,35 @@ test('subtask context', async t => {
 		}
 	]);
 
-	await list.run({foo: 'bar'});
+	const result = await list.run({foo: 'bar'});
+
+	t.deepEqual(result, {
+		foo: 'bar',
+		fiz: 'biz'
+	});
+});
+
+test('context is attached to error object', async t => {
+	const list = new Listr([
+		{
+			title: 'foo',
+			task: context => {
+				context.foo = 'bar';
+			}
+		},
+		{
+			title: 'unicorn',
+			task: () => Promise.reject(new Error('foo bar'))
+		}
+	]);
+
+	try {
+		await list.run();
+		t.fail('Should throw error');
+	} catch (err) {
+		t.is(err.message, 'foo bar');
+		t.deepEqual(err.context, {
+			foo: 'bar'
+		});
+	}
 });