From af3c63994b450fd4473f262fe0316e612bbdb3e9 Mon Sep 17 00:00:00 2001
From: MoyuScript <i@moyu.moe>
Date: Mon, 19 Jan 2015 20:31:31 +0200
Subject: [PATCH] Add test for adding class prior to rendering (#508)

---
 tests/mocha/options.onclone.html | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
diff --git a/tests/mocha/options.onclone.html b/tests/mocha/options.onclone.html
index 478694f..44c8873 100644
--- a/tests/mocha/options.onclone.html
+++ b/tests/mocha/options.onclone.html
@@ -7,11 +7,21 @@
     <script src="../assets/jquery-1.6.2.js"></script>
     <script src="lib/expect.js"></script>
     <script src="lib/mocha.js"></script>
+    <style>
+        #block2 {
+            background: red;
+        }
+
+        .my-css-class #block2 {
+            background: green;
+        }
+    </style>
 </head>
 <body>
 <div id="mocha"></div>
 <script>mocha.setup('bdd')</script>
 <div style="background: red; width: 200px; height:200px;" id="block"></div>
+<div style="width: 200px; height:200px;" id="block2"></div>
 <script>
     describe("options.onclone", function() {
         it("with a function", function(done) {
@@ -46,6 +56,19 @@
                 done(error);
             });
         });
+
+        it("add class to node", function(done) {
+            html2canvas(document.querySelector("#block2"), {onclone: function(document) {
+                document.documentElement.className = "my-css-class";
+            }}).then(function(canvas) {
+                expect(canvas.width).to.equal(200);
+                expect(canvas.height).to.equal(200);
+                validCanvasPixels(canvas);
+                done();
+            }).catch(function(error) {
+                done(error);
+            });
+        });
     });
 
     function validCanvasPixels(canvas) {
