pigspec 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2052e77573cb0ffe2219065466cd794acd6717c7
-  data.tar.gz: 33245bb7b47d1468c1829b8778db76f469ce6ec0
+  metadata.gz: 6f7b3322d7176a21b68b6eecf8108f249fed30f6
+  data.tar.gz: f353261cbbfa282cd776f1068c80d3c8e72184e7
 SHA512:
-  metadata.gz: c1f03b07a03297e0ec463f71d3965472cdf4f03ff7d85af86a829985aa48ca6c8c2f7af46bb21bc3d3943ff25d0c72767eccbc21a839e33d2f855140668ee744
-  data.tar.gz: 923084c10aece6677bc88682c193c4958ae8902d8c53c453d113b41a260dc93a27f3a50ce754f121b0ba6a9bdd78efab55ee38d1a24e757c9b805e1d65c653ce
+  metadata.gz: c39701ceeacc8b0ed4e2e591e25b5f4b0e428ded03cf1fefafc92c6062ae4c9d69002e7b391adb76336191a5f8a4dc3fb0d8932c278a39356d564d859fc0aa02
+  data.tar.gz: 248f03a67b1289afb7a68b7f0ba910b37f34a5115c60ea61acd8c41030d89e0ea18001d02e3bdfd384763a8822e53c56a18b36243336fad8f3462bd901049d78

data/lib/pigspec.rb

@@ -58,8 +58,8 @@ module PigSpec
     end
     expose :with_args
 
-    def override(name, value)
-      @override.push name: name, value: value
+    def override(name, value, options = {})
+      @override.push name: name, value: value, schema: options[:as]
     end
     expose :override
 
@@ -83,16 +83,31 @@ module PigSpec
 
     def apply_override(test)
       @override.each do |item|
-        temp = Test.bridge.create_hdfs_temp
-        Test.bridge.upload_text item[:value], temp
-
-        schema = Test.bridge.schema item[:name]
-        query  = "#{item[:name]} = LOAD '#{temp}' USING PigStorage('\\t')"
-        query += " AS #{schema}" unless schema.nil?
-        query += ';'
+        value_file = upload_override_value item
+        query = compose_override_query item, value_file
         test.override item[:name], query
       end
     end
+
+    def upload_override_value(item)
+      temp = Test.bridge.create_hdfs_temp
+      Test.bridge.upload_text item[:value], temp
+      temp
+    end
+
+    def compose_override_query(item, value_file)
+      schema = compose_override_schema item
+      query  = "#{item[:name]} = LOAD '#{value_file}' USING PigStorage('\\t')"
+      query += " AS #{schema}" unless schema.nil?
+      query += ';'
+      query
+    end
+
+    def compose_override_schema(item)
+      schema = item[:schema]
+      schema ||= Test.bridge.schema item[:name]
+      schema
+    end
   end # class Test
 
 module_function

data/lib/pigspec/version.rb

@@ -1,4 +1,4 @@
 # PigSpec version constant
 module PigSpec
-  VERSION = '0.0.6'
+  VERSION = '0.0.7'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pigspec
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - shiracha