statementByLabel utility

sorear · Jul 5, 2015 · 41ccfdf · 41ccfdf
1 parent 738c0f5
commit 41ccfdf
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/lib/smm/scoper.js b/lib/smm/scoper.js
@@ -467,6 +467,11 @@ class MMScoper {
     }
 }
 
+MMOMDatabase.prototype.statementByLabel = function(label) {
+    let sym = this.scoper.lookup(label);
+    return sym ? sym.labelled : null;
+};
+
 var HIGHSEG = MMScoper.HIGHSEG = -1 >>> 1;
 // MM files larger than 2GB are likely to cause no end of problems...
 

diff --git a/test/scoper.js b/test/scoper.js
@@ -1,6 +1,7 @@
 var MMOM = require('../lib/smm/mmom');
 var expect = require('chai').expect;
 require('../lib/smm/scoper');
+import { describeDB } from './lib/util';
 var db, errlist;
 function src(x) {
     before(function () {
@@ -348,9 +349,16 @@ var cases = [
     },
 ];
 
-cases.forEach(function (cc) {
-    describe(`${cc.name}:`, function () {
-        src(cc.src);
-        errs(cc.errs);
+describe('scoper i/o:', () => {
+    cases.forEach(function (cc) {
+        describe(`${cc.name}:`, function () {
+            src(cc.src);
+            errs(cc.errs);
+        });
     });
 });
+
+describeDB('scoper utilities', '$c foo $.  bar $a foo $.', dbt => {
+    it('statementByLabel (fail)', () => expect(dbt().statementByLabel('bar2')).to.equal(null));
+    it('statementByLabel (success)', () => expect(dbt().statementByLabel('bar').math).to.eql(['foo']));
+});