vorax 0.1.0pre

data/spec/conn_string_spec.rb

@@ -0,0 +1,53 @@
+# encoding: UTF-8
+
+include Vorax
+
+describe 'sqlplus' do
+
+  before(:all) do# {{{
+    @cstr = Parser::ConnString.new
+  end# }}}
+
+  it 'should parse a connection string containing just the user' do# {{{
+    @cstr.parse('scott')[:user].should eq('scott')
+  end# }}}
+
+  it 'should parse a connection string containing the user and the password' do# {{{
+    @cstr.parse('scott/tiger')[:user].should eq('scott')
+  end# }}}
+
+  it 'should parse a connection string containing the user and the database' do# {{{
+    @cstr.parse('scott@db')[:user].should eq('scott')
+  end# }}}
+
+  it 'should parse a connection string containing the user with special chars' do# {{{
+    @cstr.parse('"scott@man"@db')[:user].should eq('scott@man')
+    @cstr.parse('"scott/man"@db')[:user].should eq('scott/man')
+  end# }}}
+
+  it 'should parse a connection string containing an empty user' do# {{{
+    @cstr.parse('/')[:user].should eq('')
+  end# }}}
+
+  it 'should extract the role' do# {{{
+    @cstr.parse('talek/muci as sysdba').should eq({:prompt_for => nil, :user =>'talek', :password => 'muci', :db => '', :role => 'sysdba'})
+    @cstr.parse('talek/"muciBuci123" as  SYSASM').should eq({:prompt_for => nil, :user =>'talek', :password => 'muciBuci123', :db => '', :role => 'sysasm'})
+    @cstr.parse('scott/tiger@//fox:1521/fd.fox.ro as sysoper').should eq({:prompt_for => nil, :user =>'scott', :password => 'tiger', :db => '//fox:1521/fd.fox.ro', :role => 'sysoper'})
+    @cstr.parse('scott@//fox:1521/fd.fox.ro as sysoper').should eq({:prompt_for => :password, :user =>'scott', :password => '', :db => '//fox:1521/fd.fox.ro', :role => 'sysoper'})
+  end# }}}
+
+  it 'should work with OS auth' do# {{{
+    @cstr.parse('/').should eq({:user =>'', :password => '', :db => '', :role => '', :prompt_for => nil})
+    @cstr.parse('/ as sysdba').should eq({:prompt_for => nil, :user =>'', :password => '', :db => '', :role => 'sysdba'})
+    @cstr.parse('/@db').should eq({:prompt_for => nil, :user =>'', :password => '', :db => 'db', :role => ''})
+  end# }}}
+
+  it 'should prompt for password when just the user is given' do# {{{
+    @cstr.parse('scott').should eq({:user =>'scott', :password => '', :db => '', :role => '', :prompt_for => :password})
+  end# }}}
+
+  it 'should prompt for user when nothing is given' do# {{{
+    @cstr.parse('').should eq({:user =>'', :password => '', :db => '', :role => '', :prompt_for => :user})
+  end# }}}
+
+end

data/spec/package_spec_spec.rb

@@ -0,0 +1,48 @@
+# encoding: UTF-8
+
+include Vorax
+include Parser
+
+describe 'package_spec' do
+
+  it 'should work with a simple package' do
+    text = File.open("spec/sql/muci.spc", 'rb') { |file| file.read }
+    parser = Vorax::Parser::PackageSpec.new
+    parser.walk(text)
+
+    parser.constants.should eq(Set.new(["MY_CONSTANT1", "MY_CONSTANT2"]))
+    parser.exceptions.should eq(Set.new(["ex_no_data_found", "ex_custom"]))
+    parser.cursors.should eq(Set.new(["my_cursor"]))
+    parser.types.should eq(Set.new(["population_type"]))
+    parser.variables.should eq(Set.new(["g_var1", "g_var2"]))
+    parser.procedures.should eq(Set.new(["my_proc"]))
+    parser.functions.should eq(Set.new(["my_func"]))
+  end
+
+  it 'should work with a big package spec' do
+    text = File.open('spec/sql/dbms_stats.spc', 'rb') { |file| file.read }
+    parser = Vorax::Parser::PackageSpec.new
+    parser.walk(text)
+    parser.constants.should eq(Set.new(["AUTO_CASCADE", "AUTO_INVALIDATE", "AUTO_SAMPLE_SIZE", "DEFAULT_DEGREE", "AUTO_DEGREE", "DEFAULT_CASCADE", "DEFAULT_DEGREE_VALUE", "DEFAULT_ESTIMATE_PERCENT", "DEFAULT_METHOD_OPT", "DEFAULT_NO_INVALIDATE", "DEFAULT_GRANULARITY", "DEFAULT_PUBLISH", "DEFAULT_INCREMENTAL", "DEFAULT_STALE_PERCENT", "DEFAULT_AUTOSTATS_TARGET", "DEFAULT_STAT_CATEGORY", "PURGE_ALL"]))
+    parser.exceptions.should eq(Set.new([]))
+    parser.cursors.should eq(Set.new([]))
+    parser.types.should eq(Set.new(["numarray", "datearray", "chararray", "rawarray", "fltarray", "dblarray", "StatRec", "ObjectElem", "ObjectTab", "DiffRepElem", "DiffRepTab", "CContext"]))
+    parser.variables.should eq(Set.new([]))
+    parser.procedures.should eq(Set.new(["prepare_column_values", "prepare_column_values_nvarchar", "prepare_column_values_rowid", "set_param", "reset_param_defaults", "reset_global_pref_defaults", "set_global_prefs", "set_table_prefs", "delete_table_prefs", "export_table_prefs", "import_table_prefs", "set_schema_prefs", "delete_schema_prefs", "export_schema_prefs", "import_schema_prefs", "set_database_prefs", "delete_database_prefs", "export_database_prefs", "import_database_prefs", "init_package", "publish_pending_stats", "export_pending_stats", "delete_pending_stats", "resume_gather_stats", "set_column_stats", "set_index_stats", "set_table_stats", "convert_raw_value", "convert_raw_value_nvarchar", "convert_raw_value_rowid", "get_column_stats", "get_index_stats", "get_table_stats", "delete_column_stats", "delete_index_stats", "delete_table_stats", "delete_schema_stats", "delete_database_stats", "create_stat_table", "drop_stat_table", "upgrade_stat_table", "export_column_stats", "export_index_stats", "export_table_stats", "export_schema_stats", "export_database_stats", "import_column_stats", "import_index_stats", "import_table_stats", "import_schema_stats", "import_database_stats", "gather_index_stats", "gather_table_stats", "gather_schema_stats", "gather_database_stats", "generate_stats", "flush_database_monitoring_info", "alter_schema_tab_monitoring", "alter_database_tab_monitoring", "gather_system_stats", "get_system_stats", "set_system_stats", "delete_system_stats", "import_system_stats", "export_system_stats", "gather_fixed_objects_stats", "delete_fixed_objects_stats", "export_fixed_objects_stats", "import_fixed_objects_stats", "gather_dictionary_stats", "delete_dictionary_stats", "export_dictionary_stats", "import_dictionary_stats", "lock_table_stats", "lock_partition_stats", "lock_schema_stats", "unlock_table_stats", "unlock_partition_stats", "unlock_schema_stats", "restore_table_stats", "restore_schema_stats", "restore_database_stats", "restore_fixed_objects_stats", "restore_dictionary_stats", "restore_system_stats", "purge_stats", "alter_stats_history_retention", "copy_table_stats", "drop_extended_stats", "merge_col_usage", "seed_col_usage", "reset_col_usage", "gather_database_stats_job_proc", "cleanup_stats_job_proc"]))
+    parser.functions.should eq(Set.new(["get_param", "get_prefs", "to_cascade_type", "to_estimate_percent_type", "to_degree_type", "to_no_invalidate_type", "to_publish_type", "get_stats_history_retention", "get_stats_history_availability", "diff_table_stats_in_stattab", "diff_table_stats_in_history", "diff_table_stats_in_pending", "create_extended_stats", "show_extended_stats_name", "report_col_usage"]))
+  end
+
+  it 'should work with a SYS package spec' do
+    text = File.open('spec/sql/dbms_crypto.spc', 'rb') { |file| file.read }
+    parser = Vorax::Parser::PackageSpec.new
+    parser.walk(text)
+    parser.constants.should eq(Set.new(["HASH_MD4", "HASH_MD5", "HASH_SH1", "HMAC_MD5", "HMAC_SH1", "ENCRYPT_DES", "ENCRYPT_3DES_2KEY", "ENCRYPT_3DES", "ENCRYPT_AES", "ENCRYPT_PBE_MD5DES", "ENCRYPT_AES128", "ENCRYPT_AES192", "ENCRYPT_AES256", "CHAIN_CBC", "CHAIN_CFB", "CHAIN_ECB", "CHAIN_OFB", "PAD_PKCS5", "PAD_NONE", "PAD_ZERO", "PAD_ORCL", "ENCRYPT_RC4", "DES_CBC_PKCS5", "DES3_CBC_PKCS5", "AES_CBC_PKCS5"]))
+    parser.exceptions.should eq(Set.new(["CipherSuiteInvalid", "CipherSuiteNull", "KeyNull", "KeyBadSize", "DoubleEncryption"]))
+    parser.cursors.should eq(Set.new([]))
+    parser.types.should eq(Set.new([]))
+    parser.variables.should eq(Set.new([]))
+    parser.procedures.should eq(Set.new(["Encrypt", "Decrypt"]))
+    parser.functions.should eq(Set.new(["Encrypt", "Decrypt", "Hash", "Mac", "RandomBytes", "RandomNumber", "RandomInteger"]))
+  end
+
+end

data/spec/pagezip_spec.rb

@@ -0,0 +1,153 @@
+# encoding: UTF-8
+
+include Vorax
+
+describe 'pagezip layout' do
+
+  before(:each) do# {{{
+    @sp = Sqlplus.new('sqlplus')
+    @sp.default_convertor = :pagezip
+    @prep = [VORAX_CSTR, 
+             "set tab off", 
+             "set linesize 10000", 
+             "set markup html on", 
+             "set echo off",
+             "set pause off",
+             "set define &",
+             "set termout on",
+             "set verify off",
+             "set pagesize 5"].join("\n")
+    @result = ""
+  end# }}}
+
+  it 'should work with multi line records' do# {{{
+    @sp.exec("select * from departments where id in (1, 2, 6);", :prep => @prep)
+    @result << @sp.read_output(32767) while @sp.busy?
+    expected = <<OUTPUT
+
+SQL>
+
+ID NAME        DESCRIPTION                        
+-- ----------- -----------------------------------
+ 1 Bookkeeping This department is responsible for:
+               - financial reporting              
+               - analysis                         
+               - other boring tasks               
+ 2 Marketing                                      
+ 6 Management  The bad guys department            
+
+SQL>
+OUTPUT
+    @result.should eq(expected)
+  end# }}}
+
+  it 'should work with a single line record' do# {{{
+    @sp.exec("select * from departments where id=2;", :prep => @prep)
+    @result << @sp.read_output(32767) while @sp.busy?
+    expected = <<OUTPUT
+
+SQL>
+
+ID NAME      DESCRIPTION
+-- --------- -----------
+ 2 Marketing            
+
+SQL>
+OUTPUT
+    @result.should eq(expected)
+  end# }}}
+
+  it 'should work without headers' do# {{{
+    @sp.exec("select * from departments where id<=4;", :prep => "#@prep\nset pagesize 0")
+    @result << @sp.read_output(32767) while @sp.busy?
+    expected = <<OUTPUT
+
+SQL>
+
+1 Bookkeeping This department is responsible for:
+              - financial reporting              
+              - analysis                         
+              - other boring tasks               
+2 Marketing                                      
+3 Deliveries                                     
+4 CRM                                            
+
+SQL>
+OUTPUT
+    @result.should eq(expected)
+  end# }}}
+
+  it 'should work with special unicode chars' do# {{{
+    @sp.exec("select * from employees where id=1;", :prep => @prep)
+    @result << @sp.read_output(32767) while @sp.busy?
+    expected = <<OUTPUT
+
+SQL>
+
+ID NAME        SALARY DEPARTMENT_ID
+-- ----------- ------ -------------
+ 1 Tică Șerban    570             1
+
+SQL>
+OUTPUT
+    @result.should eq(expected)
+  end# }}}
+
+  it 'should work with multiple pages' do# {{{
+    @sp.exec("select * from departments where id<=10;", :prep => "#@prep\nset pagesize 4")
+    @result << @sp.read_output(32767) while @sp.busy?
+    expected = <<OUTPUT
+
+SQL>
+
+ID NAME        DESCRIPTION                        
+-- ----------- -----------------------------------
+ 1 Bookkeeping This department is responsible for:
+               - financial reporting              
+               - analysis                         
+               - other boring tasks               
+ 2 Marketing                                      
+ 3 Deliveries                                     
+ 4 CRM                                            
+
+ID NAME             DESCRIPTION            
+-- ---------------- -----------------------
+ 5 Legal Stuff                             
+ 6 Management       The bad guys department
+ 7 Cooking                                 
+ 8 Public Relations                        
+
+ID NAME        DESCRIPTION
+-- ----------- -----------
+ 9 Aquisitions            
+10 Cleaning               
+
+10 rows selected.
+
+SQL>
+OUTPUT
+    #puts @result
+    @result.should eq(expected)
+  end# }}}
+
+it 'should work with accept prompts' do# {{{
+  begin
+    pack_file = Tempfile.new(['vorax', '.sql'])
+    @sp.exec("accept var prompt \"Enter var: \"\nprompt &var", :prep => @prep, :pack_file => pack_file.path)
+    Timeout::timeout(10) {
+      @result << @sp.read_output(32767) while @result !~ /Enter var:\z/
+      @sp.send_text("muci\n")
+      @result << @sp.read_output(32767) while @result !~ /muci\n\z/
+    }
+  ensure
+    pack_file.unlink
+  end
+end# }}}
+
+  after(:each) do# {{{
+    @sp.terminate
+  end# }}}
+
+end
+
+

data/spec/parser_spec.rb

@@ -0,0 +1,299 @@
+# encoding: UTF-8
+
+include Vorax
+
+describe 'Parser' do
+
+  it 'should detect the end of a subprogram' do# {{{
+    text = "end;"
+    Parser.plsql_def(text)[:end_def].should be >0;
+    text = "end if;"
+    Parser.plsql_def(text)[:end_def].should be <0;
+    text = "end loop;"
+    Parser.plsql_def(text)[:end_def].should be <0;
+    text = "end my_func;"
+    Parser.plsql_def(text)[:end_def].should be >0;
+    text = "end/*test*/my_prog;"
+    Parser.plsql_def(text)[:end_def].should be >0;
+    text = "end/*test*/\"my_special_prog\";"
+    Parser.plsql_def(text)[:end_def].should be >0;
+    text = "end is_AdminBasket;\r\n\r\n  function get_BasketId(pi_impkey in integer, pi_tablename in varchar2)\r\n    "
+    Parser.plsql_def(text)[:end_def].should be >0;
+  end# }}}
+
+  it 'should detect the definition of a plsql module' do# {{{
+    text = "PACKAGE muci AS "
+    result = Parser.plsql_def(text)
+    result[:end_def].should be >0
+    result[:name].should == 'muci'
+    result[:type].should == 'SPEC'
+    text = "package\nbody muci.buci/* bla bla*/ as "
+    result = Parser.plsql_def(text)
+    result[:end_def].should be >0;
+    result[:name].should == 'muci.buci'
+    result[:type].should == 'BODY'
+    text = "package bdy muci.buci/* bla bla*/ as "
+    Parser.plsql_def(text)[:end_def].should be <0;
+  end# }}}
+
+  it 'should work with balanced paren' do# {{{
+    Parser.walk_balanced_paren('(a(b)c)').should eq("(a(b)c)")
+    Parser.walk_balanced_paren('(a(b)(xyz)c)').should eq("(a(b)(xyz)c)")
+    Parser.walk_balanced_paren("(a')'(b)c)bla bla bla").should eq("(a')'(b)c)")
+    Parser.walk_balanced_paren('(a")"(b)c)bla bla bla').should eq('(a")"(b)c)')
+    Parser.walk_balanced_paren("(a/*)*/(b)c)bla bla bla").should eq("(a/*)*/(b)c)")
+    Parser.walk_balanced_paren("(a--)\n(b)c)bla bla bla").should eq("(a--)\n(b)c)")
+    Parser.walk_balanced_paren("(aq[')]'(b)c)bla bla bla").should eq("(aq[')]'(b)c)")
+  end# }}}
+
+  it 'should add the proper end delimitator' do# {{{
+    text = "select * from cat"
+    Parser.prepare_exec(text).should eq("select * from cat;")
+
+    text = <<TEXT
+begin
+  null;
+end;
+TEXT
+    Parser.prepare_exec(text).should eq("begin\n  null;\nend;\n/\n")
+
+    text = "set serveroutput on"
+    Parser.prepare_exec(text).should eq("set serveroutput on")
+
+    text = "select * from dual;"
+    Parser.prepare_exec(text).should eq("select * from dual;")
+  end# }}}
+
+  describe 'Comments parsing' do# {{{
+
+  it 'should be smart enough to leave quotes unchanged' do# {{{
+    stmt = <<STR
+select '/*muci*/' from/*abc*/dual;
+STR
+    Parser.remove_all_comments(stmt).should eq("select '/*muci*/' from dual;\n")
+  end# }}}
+
+  it 'should remove all comments' do# {{{
+    stmt = <<STR
+select /* muci */ * from --muhah
+--bla
+dual
+/* muha
+ha*/
+;
+STR
+    Parser.remove_all_comments(stmt).should eq("select   * from   dual\n \n;\n")
+  end# }}}
+
+  it 'should remove all comments from incomplete statements' do# {{{
+    Parser.remove_all_comments('select /* bla').should eq('select /* bla')
+  end# }}}
+
+  it 'should remove trailing comments' do# {{{
+    text = <<TEXT
+select * from cat --comment
+-- alt comment
+-- si inca un comment
+TEXT
+    Parser.remove_trailing_comments(text).should eq("select * from cat ")
+
+    text = <<TEXT
+select * from cat
+/* multi
+line comment */
+/* inca
+unul */
+TEXT
+    Parser.remove_trailing_comments(text).should eq("select * from cat\n")
+
+    text = <<TEXT
+select * from cat
+/* multi
+line comment */
+-- single line comment
+/* inca
+unul */
+-- muhaha
+TEXT
+    Parser.remove_trailing_comments(text).should eq("select * from cat\n")
+  end# }}}
+
+  end# }}}
+
+  describe 'Arguments parsing' do# {{{
+
+  it 'should detect basic argument owner' do# {{{
+    text = 'select col1, 1+2, func(a, b, my_f(x, y), c, \'a\'\'bc\', "xxx".y, d,'
+    Parser.argument_belongs_to(text, 65).should eq('func')
+    Parser.argument_belongs_to(text, 37).should eq('my_f')
+  end# }}}
+
+  it 'should handle whitespaces between function name and open paran' do# {{{
+    Parser.argument_belongs_to('exec dbms_stats.gather_schema_stats   (owname => user, ').should eq('dbms_stats.gather_schema_stats')
+    Parser.argument_belongs_to("begin dbms_stats.gather_schema_stats   \n\t  (owname => user, ").should eq('dbms_stats.gather_schema_stats')
+  end# }}}
+
+  it 'should handle quoted identifiers' do# {{{
+    Parser.argument_belongs_to('exec "My user"   . "bla@hei.la" @ dblink.hop.hu(owname => user, ').should eq("\"My user\".\"bla@hei.la\"@dblink.hop.hu")
+    Parser.argument_belongs_to('exec "ABC!"."bla"(owname => user, ').should eq("\"ABC!\".\"bla\"")
+    Parser.argument_belongs_to('exec owner."pkg"."my func"(owname => user, ').should eq("owner.\"pkg\".\"my func\"")
+  end# }}}
+
+  it 'should ignore comments' do# {{{
+    Parser.argument_belongs_to('exec dbms_stats./*muci*/gather_schema_stats/*abc*/   (owname => user, ').should eq('dbms_stats.gather_schema_stats')
+  end# }}}
+
+  it 'should handle plsql quoting' do# {{{
+    text = "exec pkg.my_proc(q'[Isn't the \"(\" character sweet?]', "
+    Parser.argument_belongs_to(text).should eq('pkg.my_proc')
+    text = "exec pkg.my_proc(q'{Isn't the \"(\" character sweet?}', "
+    Parser.argument_belongs_to(text).should eq('pkg.my_proc')
+    text = "exec pkg.my_proc(q'<Isn't the \"(\" character sweet?>', "
+    Parser.argument_belongs_to(text).should eq('pkg.my_proc')
+  end# }}}
+
+  it 'should fallback to simple quoting' do# {{{
+    text = "exec pkg.my_proc('<Isn''t the \"(\" character sweet?>', "
+    Parser.argument_belongs_to(text).should eq('pkg.my_proc')
+  end# }}}
+
+  end# }}}
+
+  describe 'Statement type' do# {{{
+
+    it 'should detect an anonymous block' do# {{{
+      Parser.statement_type("begin dbms_output.put_line('xx');").should eq("ANONYMOUS")
+      Parser.statement_type("declare l_test varchar2(10);").should eq("ANONYMOUS")
+      Parser.statement_type("bbegin dbms_output.put_line('xx');").should be_nil
+      Parser.statement_type("ddeclare dbms_output.put_line('xx');").should be_nil
+      Parser.statement_type("select 1 from dual;").should be_nil
+    end# }}}
+
+    it 'should detect a function' do# {{{
+      Parser.statement_type("create or\nreplace function muci as").should eq("FUNCTION")
+      Parser.statement_type("create orreplace function muci as").should be_nil
+    end# }}}
+
+    it 'should detect a procedure' do# {{{
+      Parser.statement_type("create or replace\n procedure muci as").should eq("PROCEDURE")
+      Parser.statement_type("create or replace proceduremuci as").should be_nil
+    end# }}}
+
+    it 'should detect a trigger' do# {{{
+      Parser.statement_type("create or replace\n trigger muci as").should eq("TRIGGER")
+      Parser.statement_type("createor replace trigger as").should be_nil
+    end# }}}
+
+    it 'should detect a package spec' do# {{{
+      Parser.statement_type("create or replace\n package muci as").should eq("PACKAGE")
+      Parser.statement_type("createor replace package as").should be_nil
+    end# }}}
+
+    it 'should detect a package body' do# {{{
+      Parser.statement_type("create or replace\n package body muci as").should eq("PACKAGE BODY")
+      Parser.statement_type("create or replace packagebody as").should be_nil
+      Parser.statement_type("create or replace package bodyas").should_not eq("PACKAGE BODY")
+    end# }}}
+
+    it 'should detect a type spec' do# {{{
+      Parser.statement_type("create or replace\n type muci as").should eq("TYPE")
+      Parser.statement_type("createor replace type as").should be_nil
+    end# }}}
+
+    it 'should detect a type body' do# {{{
+      Parser.statement_type("create or replace\n type body muci as").should eq("TYPE BODY")
+      Parser.statement_type("create or replace typebody as").should be_nil
+      Parser.statement_type("create or replace type bodyas").should_not eq("TYPE BODY")
+    end# }}}
+
+    it 'should detect java source' do# {{{
+      Parser.statement_type("create or replace java muci as").should eq("JAVA")
+      Parser.statement_type("create java muci as").should eq("JAVA")
+      Parser.statement_type("create or replace and compile java muci as").should eq("JAVA")
+      Parser.statement_type("create or replace and resolve noforce java muci as").should eq("JAVA")
+      Parser.statement_type("create or replace and resolvenoforce java muci as").should_not eq("JAVA")
+    end# }}}
+
+    it 'should detect sqlplus command' do# {{{
+      Parser.statement_type("accept muci").should eq("SQLPLUS")
+      Parser.statement_type("acc muci").should eq("SQLPLUS")
+      Parser.statement_type("acce muci").should eq("SQLPLUS")
+      Parser.statement_type("@muci.sql").should eq("SQLPLUS")
+      Parser.statement_type("@@muci.sql").should eq("SQLPLUS")
+      Parser.statement_type("/").should eq("SQLPLUS")
+      Parser.statement_type("archive log list").should eq("SQLPLUS")
+      Parser.statement_type("attribute muci").should eq("SQLPLUS")
+      Parser.statement_type("break on whatever").should eq("SQLPLUS")
+      Parser.statement_type("btitle abc").should eq("SQLPLUS")
+      Parser.statement_type("btit abc").should eq("SQLPLUS")
+      Parser.statement_type("cle").should eq("SQLPLUS")
+      Parser.statement_type("colu abc format a15").should eq("SQLPLUS")
+      Parser.statement_type("compute on bla bla bla").should eq("SQLPLUS")
+      Parser.statement_type("connect muci/buci@db").should eq("SQLPLUS")
+      Parser.statement_type("copy").should eq("SQLPLUS")
+      Parser.statement_type("def var").should eq("SQLPLUS")
+      Parser.statement_type("desc my_table").should eq("SQLPLUS")
+      Parser.statement_type("discon").should eq("SQLPLUS")
+      Parser.statement_type("execu dbms_output.put_line('abc');").should eq("SQLPLUS")
+      Parser.statement_type("exit").should eq("SQLPLUS")
+      Parser.statement_type("quit").should eq("SQLPLUS")
+      Parser.statement_type("help").should eq("SQLPLUS")
+      Parser.statement_type("host ls -al").should eq("SQLPLUS")
+      Parser.statement_type("!ls -al").should eq("SQLPLUS")
+      Parser.statement_type("passw").should eq("SQLPLUS")
+      Parser.statement_type("password user").should eq("SQLPLUS")
+      Parser.statement_type("pause press any key").should eq("SQLPLUS")
+      Parser.statement_type("print curs").should eq("SQLPLUS")
+      Parser.statement_type("prom var").should eq("SQLPLUS")
+      Parser.statement_type("recover database").should eq("SQLPLUS")
+      Parser.statement_type("rem comment").should eq("SQLPLUS")
+      Parser.statement_type("repf on").should eq("SQLPLUS")
+      Parser.statement_type("rephe off").should eq("SQLPLUS")
+      Parser.statement_type("sav muci.sql").should eq("SQLPLUS")
+      Parser.statement_type("set pagesize 100").should eq("SQLPLUS")
+      Parser.statement_type("set transaction name muci;").should be_nil
+      Parser.statement_type("show pagesize").should eq("SQLPLUS")
+      Parser.statement_type("shutdown").should eq("SQLPLUS")
+      Parser.statement_type("spool muci.log").should eq("SQLPLUS")
+      Parser.statement_type("start muci.sql").should eq("SQLPLUS")
+      Parser.statement_type("startup").should eq("SQLPLUS")
+      Parser.statement_type("store set options.sql").should eq("SQLPLUS")
+      Parser.statement_type("timing stop").should eq("SQLPLUS")
+      Parser.statement_type("title abc").should eq("SQLPLUS")
+      Parser.statement_type("undef var").should eq("SQLPLUS")
+      Parser.statement_type("var muci").should eq("SQLPLUS")
+      Parser.statement_type("whenever oserror exit").should eq("SQLPLUS")
+      Parser.statement_type("xquery bla bla bla").should eq("SQLPLUS")
+      Parser.statement_type("\n-- APEX RMS DB UPDATES FILE\n").should be_nil
+    end# }}}
+
+  end# }}}
+
+  it 'should get the current statement' do# {{{
+    text = <<STRING
+select /* comment: ; */ 'muci;buci''s yea' from dual; -- interesting ha;ha?
+select * from cat;
+STRING
+    Parser.current_statement(text, 10)[:statement].should eq("select /* comment: ; */ 'muci;buci''s yea' from dual;")
+    Parser.current_statement(text, 85)[:statement].should eq(" -- interesting ha;ha?\nselect * from cat;")
+
+    text = <<STRING
+set serveroutput on
+column c1 format a10
+select 1 from dual;
+begin
+  null;
+end;
+/
+select c2 from t1
+/
+update t set x=1;
+STRING
+    Parser.current_statement(text, 10, :sqlplus_commands => false, :plsql_blocks => false)[:statement].should eq("set serveroutput on\ncolumn c1 format a10\nselect 1 from dual;")
+    Parser.current_statement(text, 10, :sqlplus_commands => true, :plsql_blocks => false)[:statement].should eq("set serveroutput on")
+    Parser.current_statement(text, 71, :sqlplus_commands => true, :plsql_blocks => false)[:statement].should eq("\nbegin\n  null;")
+    Parser.current_statement(text, 71, :sqlplus_commands => true, :plsql_blocks => true)[:statement].should eq("\nbegin\n  null;\nend;\n/\n")
+    Parser.current_statement(text, 88, :sqlplus_commands => true, :plsql_blocks => true)[:statement].should eq("select c2 from t1\n/\n")
+  end# }}}
+
+end

data/spec/plsql_structure_spec.rb

@@ -0,0 +1,44 @@
+# encoding: UTF-8
+
+include Vorax
+
+describe 'plsql structure' do
+
+  it 'should work for complex plsql code' do
+    text = File.open('spec/sql/test.pkg', 'rb') { |file| file.read }
+    #text = File.open('spec/sql/admin_toolkit.bdy', 'rb') { |file| file.read }
+    structure = Parser::PlsqlStructure.new(text)
+    @result = ''
+    def compute_tree(node, level = 0)
+      if node.is_root?
+        @result << "*"
+      else
+        @result << "|" unless node.parent.is_last_sibling?
+        @result << (' ' * (node.level - 1) * 4)
+        @result << (node.is_last_sibling? ? "+" : "|")
+        @result << "---"
+        @result << (node.has_children? ? "+" : ">")
+      end
+
+      @result << " #{node.name} - #{(node.content ? node.content.end_pos : '')}\n"
+
+      node.children { |child| compute_tree(child, level + 1)}
+    end
+
+    compute_tree(structure.tree)
+    @result.should eq(<<STR
+* root - 
+|---+ test[SPEC]: 25 - 154
+|    |---> test[PROCEDURE]: 70 - 70
+|    +---> muci[FUNCTION]: 99 - 99
+|---+ test[BODY]: 180 - 876
+|    |---> private_proc[PROCEDURE]: 241 - 394
+|    +---> xyz[PROCEDURE]: 414 - 451
+|---> test[PROCEDURE]: 890 - 0
++---> anonymous[BLOCK]: 969 - 1040
+STR
+)
+  end
+
+end
+

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+# encoding: UTF-8
+
+$LOAD_PATH << File.expand_path('../../lib', __FILE__)
+
+# Initialize the VORAX_TEST_CSTR environment variable with
+# the user/pwd@db connection string, pointing to VORAX test
+# user, or change the below line accordingly.
+VORAX_CSTR = "connect #{ENV['VORAX_TEST_CSTR']}"
+
+require 'tempfile'
+require 'timeout'
+require 'vorax'
+