[test] add native module buffer case

rogerwang · rogerwang · commit 1c319ab4e1ec · 2017-11-28T14:03:30.000+08:00
diff --git a/test/sanity/native-module-buffer/binding.gyp b/test/sanity/native-module-buffer/binding.gyp
@@ -0,0 +1,9 @@
+{
+  "targets": [
+    {
+        "target_name": "buffer_example",
+        "sources": [ "buffer_example.cpp" ], 
+        "include_dirs" : ["<!(node -e \"require('nan')\")"]
+    }
+  ]
+}
diff --git a/test/sanity/native-module-buffer/buffer_example.cpp b/test/sanity/native-module-buffer/buffer_example.cpp
@@ -0,0 +1,30 @@
+#include <nan.h>
+#include <stdio.h>
+using namespace Nan;
+using namespace v8;
+
+void buffer_delete_callback(char* data, void* the_vector) {
+  fputs("DELETE CALLBACK\n", stderr);
+}
+
+NAN_METHOD(rotate) {
+    char* buffer = (char*) node::Buffer::Data(info[0]->ToObject());
+    size_t size = node::Buffer::Length(info[0]);
+    unsigned int rot = info[1]->Uint32Value();
+   
+    char * retval = new char[size];
+    for(unsigned int i = 0; i < size; i++ ) {
+        retval[i] = buffer[i] - rot;
+        buffer[i] += rot;
+    }   
+    
+    //info.GetReturnValue().Set(Nan::NewBuffer(retval, size, buffer_delete_callback, nullptr).ToLocalChecked());
+    info.GetReturnValue().Set(Nan::NewBuffer(retval, size).ToLocalChecked());
+}
+
+NAN_MODULE_INIT(Init) {
+   Nan::Set(target, New<String>("rotate").ToLocalChecked(),
+        GetFunction(New<FunctionTemplate>(rotate)).ToLocalChecked());
+}
+
+NODE_MODULE(buffer_example, Init)
diff --git a/test/sanity/native-module-buffer/index.html b/test/sanity/native-module-buffer/index.html
@@ -0,0 +1,8 @@
+<html>
+<body>
+  <script src="index.js"></script>
+  <h1 id='result'>result pending...</h1>
+  <script>
+    document.getElementById('result').innerHTML = result.toString('ascii');
+  </script>
+</body>
diff --git a/test/sanity/native-module-buffer/index.js b/test/sanity/native-module-buffer/index.js
@@ -0,0 +1,16 @@
+'use strict';
+const addon = require('./build/Release/buffer_example');
+
+const buffer = Buffer.from("ABC");
+
+for (var i = 0; i < 1000; i++) {
+  var b = Buffer.from("CDE");
+  addon.rotate(b, i);
+}
+
+// synchronous, rotates each character by +13
+var result = addon.rotate(buffer, 13);
+
+console.log(buffer.toString('ascii'));
+console.log(result.toString('ascii'));
+
diff --git a/test/sanity/native-module-buffer/package.json b/test/sanity/native-module-buffer/package.json
@@ -0,0 +1,13 @@
+{
+  "name": "buffer_example",
+  "main": "index.html",
+  "version": "0.0.1",
+  "private": true,
+  "gypfile": true,
+  "scripts": {
+    "start": "node index.js"
+  },
+  "dependencies": {
+    "nan": "*"
+  }
+}
diff --git a/test/sanity/native-module-buffer/test.py b/test/sanity/native-module-buffer/test.py
@@ -0,0 +1,95 @@
+import time
+import os
+import shutil
+import subprocess
+import platform
+from subprocess import Popen, PIPE
+import sys
+
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+chrome_options = Options()
+testdir = os.path.dirname(os.path.abspath(__file__))
+chrome_options.add_argument("nwapp=" + testdir)
+nw_tools  = os.path.normpath(os.path.join(testdir, os.pardir, os.pardir, os.pardir, 'tools'))
+sys.path.append(nw_tools)
+import getnwversion
+import getnwisrelease
+
+def find_executable(executable, path=None):
+    """Find if 'executable' can be run. Looks for it in 'path'
+    (string that lists directories separated by 'os.pathsep';
+    defaults to os.environ['PATH']). Checks for all executable
+    extensions. Returns full path or None if no command is found.
+    """
+    if path is None:
+        path = os.environ['PATH']
+    paths = path.split(os.pathsep)
+    extlist = ['']
+    if os.name == 'os2':
+        (base, ext) = os.path.splitext(executable)
+        # executable files on OS/2 can have an arbitrary extension, but
+        # .exe is automatically appended if no dot is present in the name
+        if not ext:
+            executable = executable + ".exe"
+    elif sys.platform == 'win32':
+        pathext = os.environ['PATHEXT'].lower().split(os.pathsep)
+        (base, ext) = os.path.splitext(executable)
+        if ext.lower() not in pathext:
+            extlist = pathext
+    for ext in extlist:
+        execname = executable + ext
+        if os.path.isfile(execname):
+            return execname
+        else:
+            for p in paths:
+                f = os.path.join(p, execname)
+                if os.path.isfile(f):
+                    return f
+    else:
+        return None
+
+os.chdir(testdir)
+
+header_path = os.path.abspath("../../../tmp/node")
+
+nw_version = getnwversion.nw_version
+if getnwisrelease.release == 0:
+    nw_version += getnwisrelease.postfix
+
+arch = ''
+_arch = platform.architecture()[0]
+if _arch == '64bit':
+    arch = 'x64'
+elif _arch == '32bit':
+    arch = 'ia32'
+else:
+    print 'Unsupported arch: ' + _arch
+    exit(-1)
+
+nw_gyp_path = find_executable('nw-gyp')
+npm_path = find_executable('npm')
+
+print "nw_gyp: ", nw_gyp_path
+print "npm_path: ", npm_path
+print "header path: ", header_path
+
+npm_env = {'npm_config_nodedir': header_path, 'npm_config_target': nw_version,
+           'npm_config_arch': arch, 'npm_config_target_arch': arch,
+           'npm_config_runtime': 'node-webkit', 'npm_config_build_from_source': "true",
+           'npm_config_node_gyp': nw_gyp_path, 'PATH': os.getenv('PATH')}
+
+proc = Popen([npm_path, 'install'], stdout=PIPE, stderr=PIPE, env=npm_env)
+out, err = proc.communicate()
+print out
+print err
+assert(proc.returncode == 0)
+
+driver = webdriver.Chrome(executable_path=os.environ['CHROMEDRIVER'], chrome_options=chrome_options)
+try:
+    print driver.current_url
+    result = driver.find_element_by_id('result').get_attribute('innerHTML')
+    print result
+    assert("456" == result)
+finally:
+    driver.quit()
