// javajstests.cpp // /** * Copyright (C) 2009 10gen Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License, version 3, * as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . */ #include "stdafx.h" #include "../db/instance.h" #include "../stdafx.h" #include "../scripting/engine.h" #include "dbtests.h" #include "util/base64.h" namespace mongo { bool dbEval(const char *ns, BSONObj& cmd, BSONObjBuilder& result, string& errmsg); } // namespace mongo namespace JSTests { class Fundamental { public: void run() { // By calling JavaJSImpl() inside run(), we ensure the unit test framework's // signal handlers are pre-installed from JNI's perspective. This allows // JNI to catch signals generated within the JVM and forward other signals // as appropriate. ScriptEngine::setup(); globalScriptEngine->runTest(); } }; class BasicScope { public: void run(){ auto_ptr s; s.reset( globalScriptEngine->newScope() ); s->setNumber( "x" , 5 ); ASSERT( 5 == s->getNumber( "x" ) ); s->setNumber( "x" , 1.67 ); ASSERT( 1.67 == s->getNumber( "x" ) ); s->setString( "s" , "eliot was here" ); ASSERT( "eliot was here" == s->getString( "s" ) ); s->setBoolean( "b" , true ); ASSERT( s->getBoolean( "b" ) ); if ( 0 ){ s->setBoolean( "b" , false ); ASSERT( ! s->getBoolean( "b" ) ); } } }; class ResetScope { public: void run(){ // Not worrying about this for now SERVER-446. /* auto_ptr s; s.reset( globalScriptEngine->newScope() ); s->setBoolean( "x" , true ); ASSERT( s->getBoolean( "x" ) ); s->reset(); ASSERT( !s->getBoolean( "x" ) ); */ } }; class FalseTests { public: void run(){ Scope * s = globalScriptEngine->newScope(); ASSERT( ! s->getBoolean( "x" ) ); s->setString( "z" , "" ); ASSERT( ! s->getBoolean( "z" ) ); delete s ; } }; class SimpleFunctions { public: void run(){ Scope * s = globalScriptEngine->newScope(); s->invoke( "x=5;" , BSONObj() ); ASSERT( 5 == s->getNumber( "x" ) ); s->invoke( "return 17;" , BSONObj() ); ASSERT( 17 == s->getNumber( "return" ) ); s->invoke( "function(){ return 17; }" , BSONObj() ); ASSERT( 17 == s->getNumber( "return" ) ); s->setNumber( "x" , 1.76 ); s->invoke( "return x == 1.76; " , BSONObj() ); ASSERT( s->getBoolean( "return" ) ); s->setNumber( "x" , 1.76 ); s->invoke( "return x == 1.79; " , BSONObj() ); ASSERT( ! s->getBoolean( "return" ) ); s->invoke( "function( z ){ return 5 + z; }" , BSON( "" << 11 ) ); ASSERT_EQUALS( 16 , s->getNumber( "return" ) ); delete s; } }; class ObjectMapping { public: void run(){ Scope * s = globalScriptEngine->newScope(); BSONObj o = BSON( "x" << 17 << "y" << "eliot" << "z" << "sara" ); s->setObject( "blah" , o ); s->invoke( "return blah.x;" , BSONObj() ); ASSERT_EQUALS( 17 , s->getNumber( "return" ) ); s->invoke( "return blah.y;" , BSONObj() ); ASSERT_EQUALS( "eliot" , s->getString( "return" ) ); s->setThis( & o ); s->invoke( "return this.z;" , BSONObj() ); ASSERT_EQUALS( "sara" , s->getString( "return" ) ); s->invoke( "return this.z == 'sara';" , BSONObj() ); ASSERT_EQUALS( true , s->getBoolean( "return" ) ); s->invoke( "this.z == 'sara';" , BSONObj() ); ASSERT_EQUALS( true , s->getBoolean( "return" ) ); s->invoke( "this.z == 'asara';" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "return" ) ); s->invoke( "return this.x == 17;" , BSONObj() ); ASSERT_EQUALS( true , s->getBoolean( "return" ) ); s->invoke( "return this.x == 18;" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "return" ) ); s->invoke( "function(){ return this.x == 17; }" , BSONObj() ); ASSERT_EQUALS( true , s->getBoolean( "return" ) ); s->invoke( "function(){ return this.x == 18; }" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "return" ) ); s->invoke( "function (){ return this.x == 17; }" , BSONObj() ); ASSERT_EQUALS( true , s->getBoolean( "return" ) ); s->invoke( "function z(){ return this.x == 18; }" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "return" ) ); s->invoke( "function (){ this.x == 17; }" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "return" ) ); s->invoke( "function z(){ this.x == 18; }" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "return" ) ); s->invoke( "x = 5; for( ; x <10; x++){ a = 1; }" , BSONObj() ); ASSERT_EQUALS( 10 , s->getNumber( "x" ) ); delete s; } }; class ObjectDecoding { public: void run(){ Scope * s = globalScriptEngine->newScope(); s->invoke( "z = { num : 1 };" , BSONObj() ); BSONObj out = s->getObject( "z" ); ASSERT_EQUALS( 1 , out["num"].number() ); ASSERT_EQUALS( 1 , out.nFields() ); s->invoke( "z = { x : 'eliot' };" , BSONObj() ); out = s->getObject( "z" ); ASSERT_EQUALS( (string)"eliot" , out["x"].valuestr() ); ASSERT_EQUALS( 1 , out.nFields() ); BSONObj o = BSON( "x" << 17 ); s->setObject( "blah" , o ); out = s->getObject( "blah" ); ASSERT_EQUALS( 17 , out["x"].number() ); delete s; } }; class JSOIDTests { public: void run(){ #ifdef MOZJS Scope * s = globalScriptEngine->newScope(); s->localConnect( "blah" ); s->invoke( "z = { _id : new ObjectId() , a : 123 };" , BSONObj() ); BSONObj out = s->getObject( "z" ); ASSERT_EQUALS( 123 , out["a"].number() ); ASSERT_EQUALS( jstOID , out["_id"].type() ); OID save = out["_id"].__oid(); s->setObject( "a" , out ); s->invoke( "y = { _id : a._id , a : 124 };" , BSONObj() ); out = s->getObject( "y" ); ASSERT_EQUALS( 124 , out["a"].number() ); ASSERT_EQUALS( jstOID , out["_id"].type() ); ASSERT_EQUALS( out["_id"].__oid().str() , save.str() ); s->invoke( "y = { _id : new ObjectId( a._id ) , a : 125 };" , BSONObj() ); out = s->getObject( "y" ); ASSERT_EQUALS( 125 , out["a"].number() ); ASSERT_EQUALS( jstOID , out["_id"].type() ); ASSERT_EQUALS( out["_id"].__oid().str() , save.str() ); delete s; #endif } }; class SetImplicit { public: void run() { Scope *s = globalScriptEngine->newScope(); BSONObj o = BSON( "foo" << "bar" ); s->setObject( "a.b", o ); ASSERT( s->getObject( "a" ).isEmpty() ); BSONObj o2 = BSONObj(); s->setObject( "a", o2 ); s->setObject( "a.b", o ); ASSERT( s->getObject( "a" ).isEmpty() ); o2 = fromjson( "{b:{}}" ); s->setObject( "a", o2 ); s->setObject( "a.b", o ); ASSERT( !s->getObject( "a" ).isEmpty() ); } }; class ObjectModReadonlyTests { public: void run(){ Scope * s = globalScriptEngine->newScope(); BSONObj o = BSON( "x" << 17 << "y" << "eliot" << "z" << "sara" << "zz" << BSONObj() ); s->setObject( "blah" , o , true ); s->invoke( "blah.y = 'e'", BSONObj() ); BSONObj out = s->getObject( "blah" ); ASSERT( strlen( out["y"].valuestr() ) > 1 ); s->invoke( "blah.a = 19;" , BSONObj() ); out = s->getObject( "blah" ); ASSERT( out["a"].eoo() ); s->invoke( "blah.zz.a = 19;" , BSONObj() ); out = s->getObject( "blah" ); ASSERT( out["zz"].embeddedObject()["a"].eoo() ); s->setObject( "blah.zz", BSON( "a" << 19 ) ); out = s->getObject( "blah" ); ASSERT( out["zz"].embeddedObject()["a"].eoo() ); s->invoke( "delete blah['x']" , BSONObj() ); out = s->getObject( "blah" ); ASSERT( !out["x"].eoo() ); // read-only object itself can be overwritten s->invoke( "blah = {}", BSONObj() ); out = s->getObject( "blah" ); ASSERT( out.isEmpty() ); // test array - can't implement this in v8 // o = fromjson( "{a:[1,2,3]}" ); // s->setObject( "blah", o, true ); // out = s->getObject( "blah" ); // s->invoke( "blah.a[ 0 ] = 4;", BSONObj() ); // s->invoke( "delete blah['a'][ 2 ];", BSONObj() ); // out = s->getObject( "blah" ); // ASSERT_EQUALS( 1.0, out[ "a" ].embeddedObject()[ 0 ].number() ); // ASSERT_EQUALS( 3.0, out[ "a" ].embeddedObject()[ 2 ].number() ); delete s; } }; class OtherJSTypes { public: void run(){ Scope * s = globalScriptEngine->newScope(); { // date BSONObj o; { BSONObjBuilder b; b.appendDate( "d" , 123456789 ); o = b.obj(); } s->setObject( "x" , o ); s->invoke( "return x.d.getTime() != 12;" , BSONObj() ); ASSERT_EQUALS( true, s->getBoolean( "return" ) ); s->invoke( "z = x.d.getTime();" , BSONObj() ); ASSERT_EQUALS( 123456789 , s->getNumber( "z" ) ); s->invoke( "z = { z : x.d }" , BSONObj() ); BSONObj out = s->getObject( "z" ); ASSERT( out["z"].type() == Date ); } { // regex BSONObj o; { BSONObjBuilder b; b.appendRegex( "r" , "^a" , "i" ); o = b.obj(); } s->setObject( "x" , o ); s->invoke( "z = x.r.test( 'b' );" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "z" ) ); s->invoke( "z = x.r.test( 'a' );" , BSONObj() ); ASSERT_EQUALS( true , s->getBoolean( "z" ) ); s->invoke( "z = x.r.test( 'ba' );" , BSONObj() ); ASSERT_EQUALS( false , s->getBoolean( "z" ) ); s->invoke( "z = { a : x.r };" , BSONObj() ); BSONObj out = s->getObject("z"); ASSERT_EQUALS( (string)"^a" , out["a"].regex() ); ASSERT_EQUALS( (string)"i" , out["a"].regexFlags() ); } // array { BSONObj o = fromjson( "{r:[1,2,3]}" ); s->setObject( "x", o, false ); BSONObj out = s->getObject( "x" ); ASSERT_EQUALS( Array, out.firstElement().type() ); s->setObject( "x", o, true ); out = s->getObject( "x" ); ASSERT_EQUALS( Array, out.firstElement().type() ); } delete s; } }; class SpecialDBTypes { public: void run(){ Scope * s = globalScriptEngine->newScope(); BSONObjBuilder b; b.appendTimestamp( "a" , 123456789 ); b.appendMinKey( "b" ); b.appendMaxKey( "c" ); b.appendTimestamp( "d" , 1234000 , 9876 ); { BSONObj t = b.done(); ASSERT_EQUALS( 1234000U , t["d"].timestampTime() ); ASSERT_EQUALS( 9876U , t["d"].timestampInc() ); } s->setObject( "z" , b.obj() ); ASSERT( s->invoke( "y = { a : z.a , b : z.b , c : z.c , d: z.d }" , BSONObj() ) == 0 ); BSONObj out = s->getObject( "y" ); ASSERT_EQUALS( Timestamp , out["a"].type() ); ASSERT_EQUALS( MinKey , out["b"].type() ); ASSERT_EQUALS( MaxKey , out["c"].type() ); ASSERT_EQUALS( Timestamp , out["d"].type() ); ASSERT_EQUALS( 9876U , out["d"].timestampInc() ); ASSERT_EQUALS( 1234000U , out["d"].timestampTime() ); ASSERT_EQUALS( 123456789U , out["a"].date() ); delete s; } }; class TypeConservation { public: void run(){ Scope * s = globalScriptEngine->newScope(); // -- A -- BSONObj o; { BSONObjBuilder b ; b.append( "a" , (int)5 ); b.append( "b" , 5.6 ); o = b.obj(); } ASSERT_EQUALS( NumberInt , o["a"].type() ); ASSERT_EQUALS( NumberDouble , o["b"].type() ); s->setObject( "z" , o ); s->invoke( "return z" , BSONObj() ); BSONObj out = s->getObject( "return" ); ASSERT_EQUALS( 5 , out["a"].number() ); ASSERT_EQUALS( 5.6 , out["b"].number() ); ASSERT_EQUALS( NumberDouble , out["b"].type() ); ASSERT_EQUALS( NumberInt , out["a"].type() ); // -- B -- { BSONObjBuilder b ; b.append( "a" , (int)5 ); b.append( "b" , 5.6 ); o = b.obj(); } s->setObject( "z" , o , false ); s->invoke( "return z" , BSONObj() ); out = s->getObject( "return" ); ASSERT_EQUALS( 5 , out["a"].number() ); ASSERT_EQUALS( 5.6 , out["b"].number() ); ASSERT_EQUALS( NumberDouble , out["b"].type() ); ASSERT_EQUALS( NumberInt , out["a"].type() ); // -- C -- { BSONObjBuilder b ; { BSONObjBuilder c; c.append( "0" , 5.5 ); c.append( "1" , 6 ); b.appendArray( "a" , c.obj() ); } o = b.obj(); } ASSERT_EQUALS( NumberDouble , o["a"].embeddedObjectUserCheck()["0"].type() ); ASSERT_EQUALS( NumberInt , o["a"].embeddedObjectUserCheck()["1"].type() ); s->setObject( "z" , o , false ); out = s->getObject( "z" ); ASSERT_EQUALS( NumberDouble , out["a"].embeddedObjectUserCheck()["0"].type() ); ASSERT_EQUALS( NumberInt , out["a"].embeddedObjectUserCheck()["1"].type() ); s->invokeSafe( "z.z = 5;" , BSONObj() ); out = s->getObject( "z" ); ASSERT_EQUALS( 5 , out["z"].number() ); ASSERT_EQUALS( NumberDouble , out["a"].embeddedObjectUserCheck()["0"].type() ); // Commenting so that v8 tests will work // ASSERT_EQUALS( NumberDouble , out["a"].embeddedObjectUserCheck()["1"].type() ); // TODO: this is technically bad, but here to make sure that i understand the behavior // Eliot says I don't have to worry about this case // // -- D -- // // o = fromjson( "{a:3.0,b:4.5}" ); // ASSERT_EQUALS( NumberDouble , o["a"].type() ); // ASSERT_EQUALS( NumberDouble , o["b"].type() ); // // s->setObject( "z" , o , false ); // s->invoke( "return z" , BSONObj() ); // out = s->getObject( "return" ); // ASSERT_EQUALS( 3 , out["a"].number() ); // ASSERT_EQUALS( 4.5 , out["b"].number() ); // // ASSERT_EQUALS( NumberDouble , out["b"].type() ); // ASSERT_EQUALS( NumberDouble , out["a"].type() ); // delete s; } }; class NumberLong { public: void run() { Scope * s = globalScriptEngine->newScope(); s->localConnect( "blah" ); BSONObjBuilder b; long long val = (long long)( 0xbabadeadbeefbaddULL ); b.append( "a", val ); BSONObj in = b.obj(); s->setObject( "a", in ); BSONObj out = s->getObject( "a" ); ASSERT_EQUALS( mongo::NumberLong, out.firstElement().type() ); ASSERT( s->exec( "b = {b:a.a}", "foo", false, true, false ) ); out = s->getObject( "b" ); ASSERT_EQUALS( mongo::NumberLong, out.firstElement().type() ); ASSERT_EQUALS( val, out.firstElement().numberLong() ); ASSERT( s->exec( "c = {c:a.a.toString()}", "foo", false, true, false ) ); out = s->getObject( "c" ); stringstream ss; ss << val; ASSERT_EQUALS( ss.str(), out.firstElement().valuestr() ); ASSERT( s->exec( "d = {d:a.a.toNumber()}", "foo", false, true, false ) ); out = s->getObject( "d" ); ASSERT_EQUALS( NumberDouble, out.firstElement().type() ); ASSERT_EQUALS( double( val ), out.firstElement().number() ); } }; class WeirdObjects { public: BSONObj build( int depth ){ BSONObjBuilder b; b.append( "0" , depth ); if ( depth > 0 ) b.appendArray( "1" , build( depth - 1 ) ); return b.obj(); } void run(){ Scope * s = globalScriptEngine->newScope(); s->localConnect( "blah" ); for ( int i=5; i<100 ; i += 10 ){ s->setObject( "a" , build(i) , false ); s->invokeSafe( "tojson( a )" , BSONObj() ); s->setObject( "a" , build(5) , true ); s->invokeSafe( "tojson( a )" , BSONObj() ); } delete s; } }; void dummy_function_to_force_dbeval_cpp_linking() { BSONObj cmd; BSONObjBuilder result; string errmsg; dbEval( "", cmd, result, errmsg); } DBDirectClient client; class Utf8Check { public: Utf8Check() { reset(); } ~Utf8Check() { reset(); } void run() { if( !globalScriptEngine->utf8Ok() ) { log() << "warning: utf8 not supported" << endl; return; } string utf8ObjSpec = "{'_id':'\\u0001\\u007f\\u07ff\\uffff'}"; BSONObj utf8Obj = fromjson( utf8ObjSpec ); client.insert( ns(), utf8Obj ); client.eval( "unittest", "v = db.jstests.utf8check.findOne(); db.jstests.utf8check.remove( {} ); db.jstests.utf8check.insert( v );" ); check( utf8Obj, client.findOne( ns(), BSONObj() ) ); } private: void check( const BSONObj &one, const BSONObj &two ) { if ( one.woCompare( two ) != 0 ) { static string fail = string( "Assertion failure expected " ) + string( one ) + ", got " + string( two ); FAIL( fail.c_str() ); } } void reset() { client.dropCollection( ns() ); } static const char *ns() { return "unittest.jstests.utf8check"; } }; class LongUtf8String { public: LongUtf8String() { reset(); } ~LongUtf8String() { reset(); } void run() { if( !globalScriptEngine->utf8Ok() ) return; client.eval( "unittest", "db.jstests.longutf8string.save( {_id:'\\uffff\\uffff\\uffff\\uffff'} )" ); } private: void reset() { client.dropCollection( ns() ); } static const char *ns() { return "unittest.jstests.longutf8string"; } }; class CodeTests { public: void run(){ Scope * s = globalScriptEngine->newScope(); { BSONObjBuilder b; b.append( "a" , 1 ); b.appendCode( "b" , "function(){ out.b = 11; }" ); b.appendCodeWScope( "c" , "function(){ out.c = 12; }" , BSONObj() ); b.appendCodeWScope( "d" , "function(){ out.d = 13 + bleh; }" , BSON( "bleh" << 5 ) ); s->setObject( "foo" , b.obj() ); } s->invokeSafe( "out = {}; out.a = foo.a; foo.b(); foo.c();" , BSONObj() ); BSONObj out = s->getObject( "out" ); ASSERT_EQUALS( 1 , out["a"].number() ); ASSERT_EQUALS( 11 , out["b"].number() ); ASSERT_EQUALS( 12 , out["c"].number() ); // Guess we don't care about this //s->invokeSafe( "foo.d() " , BSONObj() ); //out = s->getObject( "out" ); //ASSERT_EQUALS( 18 , out["d"].number() ); delete s; } }; class DBRefTest { public: DBRefTest(){ _a = "unittest.dbref.a"; _b = "unittest.dbref.b"; reset(); } ~DBRefTest(){ //reset(); } void run(){ client.insert( _a , BSON( "a" << "17" ) ); { BSONObj fromA = client.findOne( _a , BSONObj() ); cout << "Froma : " << fromA << endl; BSONObjBuilder b; b.append( "b" , 18 ); b.appendDBRef( "c" , "dbref.a" , fromA["_id"].__oid() ); client.insert( _b , b.obj() ); } ASSERT( client.eval( "unittest" , "x = db.dbref.b.findOne(); assert.eq( 17 , x.c.fetch().a , 'ref working' );" ) ); // BSON DBRef <=> JS DBPointer ASSERT( client.eval( "unittest", "x = db.dbref.b.findOne(); db.dbref.b.drop(); x.c = new DBPointer( x.c.ns, x.c.id ); db.dbref.b.insert( x );" ) ); ASSERT_EQUALS( DBRef, client.findOne( "unittest.dbref.b", "" )[ "c" ].type() ); // BSON Object <=> JS DBRef ASSERT( client.eval( "unittest", "x = db.dbref.b.findOne(); db.dbref.b.drop(); x.c = new DBRef( x.c.ns, x.c.id ); db.dbref.b.insert( x );" ) ); ASSERT_EQUALS( Object, client.findOne( "unittest.dbref.b", "" )[ "c" ].type() ); ASSERT_EQUALS( string( "dbref.a" ), client.findOne( "unittest.dbref.b", "" )[ "c" ].embeddedObject().getStringField( "$ref" ) ); } void reset(){ client.dropCollection( _a ); client.dropCollection( _b ); } const char * _a; const char * _b; }; class BinDataType { public: void pp( const char * s , BSONElement e ){ int len; const char * data = e.binData( len ); cout << s << ":" << e.binDataType() << "\t" << len << endl; cout << "\t"; for ( int i=0; inewScope(); s->localConnect( "asd" ); const char * foo = "asdas\0asdasd"; const char * base64 = "YXNkYXMAYXNkYXNk"; BSONObj in; { BSONObjBuilder b; b.append( "a" , 7 ); b.appendBinData( "b" , 12 , ByteArray , foo ); in = b.obj(); s->setObject( "x" , in ); } s->invokeSafe( "myb = x.b; print( myb ); printjson( myb );" , BSONObj() ); s->invokeSafe( "y = { c : myb };" , BSONObj() ); BSONObj out = s->getObject( "y" ); ASSERT_EQUALS( BinData , out["c"].type() ); // pp( "in " , in["b"] ); // pp( "out" , out["c"] ); ASSERT_EQUALS( 0 , in["b"].woCompare( out["c"] , false ) ); // check that BinData js class is utilized s->invokeSafe( "q = x.b.toString();", BSONObj() ); stringstream expected; expected << "BinData( type: " << ByteArray << ", base64: \"" << base64 << "\" )"; ASSERT_EQUALS( expected.str(), s->getString( "q" ) ); delete s; } }; class VarTests { public: void run(){ Scope * s = globalScriptEngine->newScope(); ASSERT( s->exec( "a = 5;" , "a" , false , true , false ) ); ASSERT_EQUALS( 5 , s->getNumber("a" ) ); ASSERT( s->exec( "var b = 6;" , "b" , false , true , false ) ); ASSERT_EQUALS( 6 , s->getNumber("b" ) ); delete s; } }; class All : public Suite { public: All() : Suite( "js" ) { } void setupTests(){ add< Fundamental >(); add< BasicScope >(); add< ResetScope >(); add< FalseTests >(); add< SimpleFunctions >(); add< ObjectMapping >(); add< ObjectDecoding >(); add< JSOIDTests >(); add< SetImplicit >(); add< ObjectModReadonlyTests >(); add< OtherJSTypes >(); add< SpecialDBTypes >(); add< TypeConservation >(); add< NumberLong >(); add< WeirdObjects >(); add< Utf8Check >(); add< LongUtf8String >(); add< CodeTests >(); add< DBRefTest >(); add< BinDataType >(); add< VarTests >(); } } myall; } // namespace JavaJSTests