From 0b890ced19571e270fce9c5ef6030117396faa60 Mon Sep 17 00:00:00 2001
From: Ryan Gaus <rsg1egoman@gmail.com>
Date: Thu, 20 Jul 2017 18:22:18 -0400
Subject: [PATCH] Code generator tests (#5847)

---
 packages/babel-generator/test/index.js | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/packages/babel-generator/test/index.js b/packages/babel-generator/test/index.js
index da7a2733b5..95056f589e 100644
--- a/packages/babel-generator/test/index.js
+++ b/packages/babel-generator/test/index.js
@@ -1,5 +1,5 @@
 import Printer from "../lib/printer";
-import generate from "../lib";
+import generate, { CodeGenerator } from "../lib";
 import assert from "assert";
 import { parse } from "babylon";
 import chai from "chai";
@@ -351,6 +351,14 @@ describe("programmatic generation", function() {
   });
 });
 
+describe("CodeGenerator", function() {
+  it("generate", function() {
+    const codeGen = new CodeGenerator(t.numericLiteral(123));
+    const code = codeGen.generate().code;
+    assert.equal(parse(code).program.body[0].expression.value, 123);
+  });
+});
+
 const suites = fixtures(`${__dirname}/fixtures`);
 
 suites.forEach(function(testSuite) {