spec/unit/syntax/nodes/test_contract.rb in qrb-0.2.0 vs spec/unit/syntax/nodes/test_contract.rb in qrb-0.3.0

@@ -8,10 +8,14 @@
 
     let(:contract){
       compiled.values.first
     }
 
+    let(:ast){
+      subject.to_ast
+    }
+
     context 'No converter and a class' do
       let(:input){ '<rgb> {r: .Integer, g: .Integer, b: .Integer}' }
 
       let(:compiled){
         subject.compile(type_factory, Color)
@@ -19,13 +23,28 @@
 
       it 'compiles to the expected Hash' do
         compiled.should be_a(Hash)
         compiled.keys.should eq([:rgb])
         contract.should be_a(Array)
-        contract.first.should be_a(TupleType)
-        contract.last.should be_a(Method)
+        contract[0].should be_a(TupleType)
+        contract[1].should be_a(Method)
+        contract[2].should be_a(Proc)
       end
+
+      it 'has expected AST' do
+        ast.should eq([
+          :contract,
+          "rgb",
+          [:tuple_type,
+            [:heading,
+              [:attribute, "r", [:builtin_type, "Integer"]],
+              [:attribute, "g", [:builtin_type, "Integer"]],
+              [:attribute, "b", [:builtin_type, "Integer"]]
+            ]
+          ]
+        ])
+      end
     end
 
     context 'No converter and no class' do
       let(:input){ '<rgb> {r: .Integer, g: .Integer, b: .Integer}' }
 
@@ -35,13 +54,28 @@
 
       it 'compiles to the expected Hash' do
         compiled.should be_a(Hash)
         compiled.keys.should eq([:rgb])
         contract.should be_a(Array)
-        contract.first.should be_a(TupleType)
-        contract.last.should be(Qrb::IDENTITY)
+        contract[0].should be_a(TupleType)
+        contract[1].should be(Qrb::IDENTITY)
+        contract[2].should be(Qrb::IDENTITY)
       end
+
+      it 'has expected AST' do
+        ast.should eq([
+          :contract,
+          "rgb",
+          [:tuple_type,
+            [:heading,
+              [:attribute, "r", [:builtin_type, "Integer"]],
+              [:attribute, "g", [:builtin_type, "Integer"]],
+              [:attribute, "b", [:builtin_type, "Integer"]]
+            ]
+          ]
+        ])
+      end
     end
 
     context 'A contract with explicit converters' do
       let(:input){ '<iso> .String \( s | DateTime.parse(s) ) \( d | d.to_s )' }
 
@@ -53,9 +87,47 @@
         compiled.should be_a(Hash)
         compiled.keys.should eq([:iso])
         contract.should be_a(Array)
         contract.first.should be_a(BuiltinType)
         contract.last.should be_a(Proc)
+      end
+
+      it 'has expected AST' do
+        ast.should eq([
+          :contract,
+          "iso",
+          [:builtin_type, "String"],
+          [:inline_pair,
+            [:fn, [:parameters, "s"], [:source, "DateTime.parse(s)"]],
+            [:fn, [:parameters, "d"], [:source, "d.to_s"]]
+          ]
+        ])
+      end
+    end
+
+    context 'A contract with external dressers' do
+      let(:input){ '<iso> .String .ExternalContract' }
+
+      let(:compiled){
+        subject.compile(type_factory, nil)
+      }
+
+      it 'compiles to the expected Hash' do
+        compiled.should be_a(Hash)
+        compiled.keys.should eq([:iso])
+        contract.should be_a(Array)
+        contract[0].should be_a(BuiltinType)
+        contract[1].should be_a(Method)
+        contract[2].should be_a(Method)
+      end
+
+      it 'has expected AST' do
+        ast.should eq([
+          :contract,
+          "iso",
+          [:builtin_type, "String"],
+          [:external_pair, "ExternalContract"]
+        ])
       end
     end
 
   end
 end