influxdb-client 1.0.0.pre.183 → 1.1.0.pre.380

data/lib/influxdb2/client/models/query.rb

@@ -0,0 +1,284 @@
+=begin
+#Influx API Service
+
+#No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+OpenAPI spec version: 0.1.0
+
+Generated by: https://openapi-generator.tech
+OpenAPI Generator version: 3.3.4
+
+=end
+
+require 'date'
+
+module InfluxDB2
+  # Query influx with specific return formatting.
+  class Query
+    attr_accessor :extern
+
+    # Query script to execute.
+    attr_accessor :query
+
+    # The type of query.
+    attr_accessor :type
+
+    # Required for `influxql` type queries.
+    attr_accessor :db
+
+    # Required for `influxql` type queries.
+    attr_accessor :rp
+
+    # Required for `influxql` type queries.
+    attr_accessor :cluster
+
+    attr_accessor :dialect
+
+    class EnumAttributeValidator
+      attr_reader :datatype
+      attr_reader :allowable_values
+
+      def initialize(datatype, allowable_values)
+        @allowable_values = allowable_values.map do |value|
+          case datatype.to_s
+          when /Integer/i
+            value.to_i
+          when /Float/i
+            value.to_f
+          else
+            value
+          end
+        end
+      end
+
+      def valid?(value)
+        !value || allowable_values.include?(value)
+      end
+    end
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'extern' => :'extern',
+        :'query' => :'query',
+        :'type' => :'type',
+        :'db' => :'db',
+        :'rp' => :'rp',
+        :'cluster' => :'cluster',
+        :'dialect' => :'dialect'
+      }
+    end
+
+    # Attribute type mapping.
+    def self.openapi_types
+      {
+        :'extern' => :'File',
+        :'query' => :'String',
+        :'type' => :'String',
+        :'db' => :'String',
+        :'rp' => :'String',
+        :'cluster' => :'String',
+        :'dialect' => :'Dialect'
+      }
+    end
+
+    # Initializes the object
+    # @param [Hash] attributes Model attributes in the form of hash
+    def initialize(attributes = {})
+      return unless attributes.is_a?(Hash)
+
+      # convert string to symbol for hash key
+      attributes = attributes.each_with_object({}) { |(k, v), h| h[k.to_sym] = v }
+
+      if attributes.has_key?(:'extern')
+        self.extern = attributes[:'extern']
+      end
+
+      if attributes.has_key?(:'query')
+        self.query = attributes[:'query']
+      end
+
+      if attributes.has_key?(:'type')
+        self.type = attributes[:'type']
+      else
+        self.type = 'flux'
+      end
+
+      if attributes.has_key?(:'db')
+        self.db = attributes[:'db']
+      end
+
+      if attributes.has_key?(:'rp')
+        self.rp = attributes[:'rp']
+      end
+
+      if attributes.has_key?(:'cluster')
+        self.cluster = attributes[:'cluster']
+      end
+
+      if attributes.has_key?(:'dialect')
+        self.dialect = attributes[:'dialect']
+      end
+    end
+
+    # Show invalid properties with the reasons. Usually used together with valid?
+    # @return Array for valid properties with the reasons
+    def list_invalid_properties
+      invalid_properties = Array.new
+      if @query.nil?
+        invalid_properties.push('invalid value for "query", query cannot be nil.')
+      end
+
+      invalid_properties
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    def valid?
+      return false if @query.nil?
+      type_validator = EnumAttributeValidator.new('String', ['flux', 'influxql'])
+      return false unless type_validator.valid?(@type)
+      true
+    end
+
+    # Custom attribute writer method checking allowed values (enum).
+    # @param [Object] type Object to be assigned
+    def type=(type)
+      validator = EnumAttributeValidator.new('String', ['flux', 'influxql'])
+      unless validator.valid?(type)
+        fail ArgumentError, 'invalid value for "type", must be one of #{validator.allowable_values}.'
+      end
+      @type = type
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param [Object] Object to be compared
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class &&
+          extern == o.extern &&
+          query == o.query &&
+          type == o.type &&
+          db == o.db &&
+          rp == o.rp &&
+          cluster == o.cluster &&
+          dialect == o.dialect
+    end
+
+    # @see the `==` method
+    # @param [Object] Object to be compared
+    def eql?(o)
+      self == o
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Fixnum] Hash code
+    def hash
+      [extern, query, type, db, rp, cluster, dialect].hash
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def build_from_hash(attributes)
+      return nil unless attributes.is_a?(Hash)
+      self.class.openapi_types.each_pair do |key, type|
+        if type =~ /\AArray<(.*)>/i
+          # check to ensure the input is an array given that the the attribute
+          # is documented as an array but the input is not
+          if attributes[self.class.attribute_map[key]].is_a?(Array)
+            self.send("#{key}=", attributes[self.class.attribute_map[key]].map { |v| _deserialize($1, v) })
+          end
+        elsif !attributes[self.class.attribute_map[key]].nil?
+          self.send("#{key}=", _deserialize(type, attributes[self.class.attribute_map[key]]))
+        end # or else data not found in attributes(hash), not an issue as the data can be optional
+      end
+
+      self
+    end
+
+    # Deserializes the data based on type
+    # @param string type Data type
+    # @param string value Value to be deserialized
+    # @return [Object] Deserialized data
+    def _deserialize(type, value)
+      case type.to_sym
+      when :DateTime
+        DateTime.parse(value)
+      when :Date
+        Date.parse(value)
+      when :String
+        value.to_s
+      when :Integer
+        value.to_i
+      when :Float
+        value.to_f
+      when :BOOLEAN
+        if value.to_s =~ /\A(true|t|yes|y|1)\z/i
+          true
+        else
+          false
+        end
+      when :Object
+        # generic object (usually a Hash), return directly
+        value
+      when /\AArray<(?<inner_type>.+)>\z/
+        inner_type = Regexp.last_match[:inner_type]
+        value.map { |v| _deserialize(inner_type, v) }
+      when /\AHash<(?<k_type>.+?), (?<v_type>.+)>\z/
+        k_type = Regexp.last_match[:k_type]
+        v_type = Regexp.last_match[:v_type]
+        {}.tap do |hash|
+          value.each do |k, v|
+            hash[_deserialize(k_type, k)] = _deserialize(v_type, v)
+          end
+        end
+      else # model
+        temp_model = InfluxDB2.const_get(type).new
+        temp_model.build_from_hash(value)
+      end
+    end
+
+    # Returns the string representation of the object
+    # @return [String] String presentation of the object
+    def to_s
+      to_hash.to_s
+    end
+
+    # to_body is an alias to to_hash (backward compatibility)
+    # @return [Hash] Returns the object in the form of hash
+    def to_body
+      to_hash
+    end
+
+    # Returns the object in the form of hash
+    # @return [Hash] Returns the object in the form of hash
+    def to_hash
+      hash = {}
+      self.class.attribute_map.each_pair do |attr, param|
+        value = self.send(attr)
+        next if value.nil?
+        hash[param] = _to_hash(value)
+      end
+      hash
+    end
+
+    # Outputs non-array value in the form of hash
+    # For object, use to_hash. Otherwise, just return the value
+    # @param [Object] value Any valid value
+    # @return [Hash] Returns the value in the form of hash
+    def _to_hash(value)
+      if value.is_a?(Array)
+        value.compact.map { |v| _to_hash(v) }
+      elsif value.is_a?(Hash)
+        {}.tap do |hash|
+          value.each { |k, v| hash[k] = _to_hash(v) }
+        end
+      elsif value.respond_to? :to_hash
+        value.to_hash
+      else
+        value
+      end
+    end
+  end
+end

data/lib/influxdb2/client/query_api.rb

@@ -0,0 +1,93 @@
+# The MIT License
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+require_relative 'models/dialect'
+require_relative 'models/query'
+require_relative 'flux_csv_parser'
+require 'json'
+
+module InfluxDB2
+  # The client of the InfluxDB 2.0 that implement Query HTTP API endpoint.
+  #
+  class QueryApi < DefaultApi
+    DEFAULT_DIALECT = InfluxDB2::Dialect.new(header: true, delimiter: ',', comment_prefix: '#',
+                                             annotations: %w[datatype group default])
+
+    # @param [Hash] options The options to be used by the client.
+    def initialize(options:)
+      super(options: options)
+    end
+
+    # @param [Object] query the flux query to execute. The data could be represent by [String], [Query]
+    # @param [String] org specifies the source organization
+    # @return [String] result of query
+    def query_raw(query: nil, org: nil, dialect: DEFAULT_DIALECT)
+      _post_query(query: query, org: org, dialect: dialect).read_body
+    end
+
+    # @param [Object] query the flux query to execute. The data could be represent by [String], [Query]
+    # @param [String] org specifies the source organization
+    # @return [Array] list of FluxTables which are matched the query
+    def query(query: nil, org: nil, dialect: DEFAULT_DIALECT)
+      response = query_raw(query: query, org: org, dialect: dialect)
+      parser = InfluxDB2::FluxCsvParser.new(response)
+
+      parser.parse
+      parser.tables
+    end
+
+    # @param [Object] query the flux query to execute. The data could be represent by [String], [Query]
+    # @param [String] org specifies the source organization
+    # @return stream of Flux Records
+    def query_stream(query: nil, org: nil, dialect: DEFAULT_DIALECT)
+      response = _post_query(query: query, org: org, dialect: dialect)
+
+      InfluxDB2::FluxCsvParser.new(response, stream: true)
+    end
+
+    private
+
+    def _post_query(query: nil, org: nil, dialect: DEFAULT_DIALECT)
+      org_param = org || @options[:org]
+      _check('org', org_param)
+
+      payload = _generate_payload(query, dialect)
+      return nil if payload.nil?
+
+      uri = URI.parse(File.join(@options[:url], '/api/v2/query'))
+      uri.query = URI.encode_www_form(org: org_param)
+
+      _post(payload.to_body.to_json, uri)
+    end
+
+    def _generate_payload(query, dialect)
+      if query.nil?
+        nil
+      elsif query.is_a?(Query)
+        query
+      elsif query.is_a?(String)
+        if query.empty?
+          nil
+        else
+          Query.new(query: query, dialect: dialect, type: nil)
+        end
+      end
+    end
+  end
+end

data/lib/influxdb2/client/version.rb

@@ -19,5 +19,5 @@
 # THE SOFTWARE.
 
 module InfluxDB2
-  VERSION = '1.0.0'.freeze
+  VERSION = '1.1.0'.freeze
 end

data/lib/influxdb2/client/write_api.rb

@@ -37,14 +37,10 @@ module InfluxDB2
 
   # Write time series data into InfluxDB.
   #
-  class WriteApi
-    DEFAULT_TIMEOUT = 10
-    DEFAULT_REDIRECT_COUNT = 10
-
+  class WriteApi < DefaultApi
     # @param [Hash] options The options to be used by the client.
     def initialize(options:)
-      @options = options
-      @max_redirect_count = @options[:max_redirect_count] || DEFAULT_REDIRECT_COUNT
+      super(options: options)
     end
 
     # Write data into specified Bucket.
@@ -98,39 +94,6 @@ module InfluxDB2
 
     private
 
-    def _post(payload, uri, limit = @max_redirect_count)
-      raise InfluxError.from_message("Too many HTTP redirects. Exceeded limit: #{@max_redirect_count}") if limit.zero?
-
-      http = Net::HTTP.new(uri.host, uri.port)
-      http.open_timeout = @options[:open_timeout] || DEFAULT_TIMEOUT
-      http.write_timeout = @options[:write_timeout] || DEFAULT_TIMEOUT if Net::HTTP.method_defined? :write_timeout
-      http.read_timeout = @options[:read_timeout] || DEFAULT_TIMEOUT
-      http.use_ssl = @options[:use_ssl].nil? ? true : @options[:use_ssl]
-
-      request = Net::HTTP::Post.new(uri.request_uri)
-      request['Authorization'] = "Token #{@options[:token]}"
-      request.body = payload
-
-      begin
-        response = http.request(request)
-        case response
-        when Net::HTTPSuccess then
-          response
-        when Net::HTTPRedirection then
-          location = response['location']
-          _post(payload, URI.parse(location), limit - 1)
-        else
-          raise InfluxError.from_response(response)
-        end
-      ensure
-        http.finish if http.started?
-      end
-    end
-
-    def _check(key, value)
-      raise ArgumentError, "The '#{key}' should be defined as argument or default option: #{@options}" if value.nil?
-    end
-
     def _generate_payload(data)
       if data.nil?
         nil

data/test/influxdb/flux_csv_parser_test.rb

@@ -0,0 +1,326 @@
+# The MIT License
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+require 'test_helper'
+
+class FluxCsvParserTest < MiniTest::Test
+  def test_multiple_values
+    data = "#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,string,string,string,string,long,long,string\n" \
+      "#group,false,false,true,true,true,true,true,true,false,false,false\n" \
+      "#default,_result,,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_field,_measurement,host,region,_value2,value1,value_str\n" \
+      ",,0,1677-09-21T00:12:43.145224192Z,2018-07-16T11:21:02.547596934Z,free,mem,A,west,121,11,test\n" \
+      ",,1,1677-09-21T00:12:43.145224192Z,2018-07-16T11:21:02.547596934Z,free,mem,B,west,484,22,test\n" \
+      ",,2,1677-09-21T00:12:43.145224192Z,2018-07-16T11:21:02.547596934Z,usage_system,cpu,A,west,1444,38,test\n" \
+      ',,3,1677-09-21T00:12:43.145224192Z,2018-07-16T11:21:02.547596934Z,user_usage,cpu,A,west,2401,49,test'
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+
+    column_headers = tables[0].columns
+    assert_equal 11, column_headers.size
+
+    values = [false, false, true, true, true, true, true, true, false, false, false]
+    _assert_columns(column_headers, values: values)
+    assert_equal 4, tables.size
+
+    _assert_multiple_record(tables)
+  end
+
+  def test_parse_shortcut
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,boolean\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,true\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,true\n"
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+
+    assert_equal 1, tables.size
+    assert_equal 1, tables[0].records.size
+
+    record = tables[0].records[0]
+
+    assert_equal _parse_time('1970-01-01T00:00:10Z'), record.start
+    assert_equal _parse_time('1970-01-01T00:00:20Z'), record.stop
+    assert_equal _parse_time('1970-01-01T00:00:10Z'), record.time
+    assert_equal 10, record.value
+    assert_equal 'free', record.field
+    assert_equal 'mem', record.measurement
+  end
+
+  def test_mapping_boolean
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,boolean\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,true\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,true\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,false\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,x\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n"
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+    records = tables[0].records
+
+    assert_equal true, records[0].values['value']
+    assert_equal false, records[1].values['value']
+    assert_equal false, records[2].values['value']
+    assert_equal true, records[3].values['value']
+  end
+
+  def test_mapping_unsigned_long
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,unsignedLong\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,17916881237904312345\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n"
+
+    expected = 17_916_881_237_904_312_345
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+    records = tables[0].records
+
+    assert_equal expected, records[0].values['value']
+    assert_nil records[1].values['value']
+  end
+
+  def test_mapping_double
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,double\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,12.25\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n" \
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+    records = tables[0].records
+
+    assert_equal 12.25, records[0].values['value']
+    assert_nil records[1].values['value']
+  end
+
+  def test_mapping_base64_binary
+    binary_data = 'test value'
+    encoded_data = Base64.encode64(binary_data)
+
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,base64Binary\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ',,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,' + encoded_data + "\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n"
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+    records = tables[0].records
+
+    value = records[0].values['value']
+
+    assert !value.nil?
+    assert_equal binary_data, value
+
+    assert_nil records[1].values['value']
+  end
+
+  def test_mapping_rfc3339
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,dateTime:RFC3339\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,1970-01-01T00:00:10Z\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n"
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+    records = tables[0].records
+
+    assert_equal Time.parse('1970-01-01T00:00:10Z').to_datetime.rfc3339, records[0].values['value']
+    assert_nil records[1].values['value']
+  end
+
+  def test_mapping_duration
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339' \
+      ",dateTime:RFC3339,long,string,string,string,duration\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,125\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n"
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+    records = tables[0].records
+
+    assert_equal 125, records[0].values['value']
+    assert_nil records[1].values['value']
+  end
+
+  def test_group_key
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,duration\n" \
+      "#group,false,false,false,false,true,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,125\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n" \
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+
+    assert_equal 10, tables[0].columns.size
+    assert_equal 2, tables[0].group_key.size
+  end
+
+  def test_unknown_type_as_string
+    data = '#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,' \
+      "dateTime:RFC3339,long,string,string,string,unknown\n" \
+      "#group,false,false,false,false,false,false,false,false,false,true\n" \
+      "#default,_result,,,,,,,,,\n" \
+      ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,12.25\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n"
+
+    tables = InfluxDB2::FluxCsvParser.new(data).parse.tables
+    records = tables[0].records
+
+    assert_equal '12.25', records[0].values['value']
+    assert_nil records[1].values['value']
+  end
+
+  private
+
+  def _parse_time(time)
+    Time.parse(time).to_datetime.rfc3339
+  end
+
+  def _assert_record(flux_record, values: nil, size: 0, value: nil)
+    values.keys.each do |key|
+      assert_equal values[key], flux_record.values[key]
+    end
+
+    if value.nil?
+      assert_nil value
+    else
+      assert_equal value, flux_record.value
+    end
+
+    assert_equal size, flux_record.values.size
+  end
+
+  def _assert_columns(column_headers, values: nil)
+    i = 0
+    values.each do |value|
+      assert_equal value, column_headers[i].group
+      i += 1
+    end
+  end
+
+  def _assert_multiple_record(tables)
+    # Record 1
+    table_records = tables[0].records
+    assert_equal 1, table_records.size
+
+    values = { 'table' => 0, 'host' => 'A', 'region' => 'west', 'value1' => 11, '_value2' => 121,
+               'value_str' => 'test' }
+
+    _assert_record(table_records[0], values: values, size: 11)
+
+    # Record 2
+    table_records = tables[1].records
+    assert_equal 1, table_records.size
+
+    values = { 'table' => 1, 'host' => 'B', 'region' => 'west', 'value1' => 22, '_value2' => 484,
+               'value_str' => 'test' }
+
+    _assert_record(table_records[0], values: values, size: 11)
+
+    # Record 3
+    table_records = tables[2].records
+    assert_equal 1, table_records.size
+
+    values = { 'table' => 2, 'host' => 'A', 'region' => 'west', 'value1' => 38, '_value2' => 1444,
+               'value_str' => 'test' }
+
+    _assert_record(table_records[0], values: values, size: 11)
+
+    # Record 4
+    table_records = tables[3].records
+    assert_equal 1, table_records.size
+
+    values = { 'table' => 3, 'host' => 'A', 'region' => 'west', 'value1' => 49, '_value2' => 2401,
+               'value_str' => 'test' }
+
+    _assert_record(table_records[0], values: values, size: 11)
+  end
+end
+
+class FluxCsvParserErrorTest < MiniTest::Test
+  def test_error
+    data = "#datatype,string,string\n" \
+      "#group,true,true\n" \
+      "#default,,\n" \
+      ",error,reference\n" \
+      ',failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time,897'
+
+    parser = InfluxDB2::FluxCsvParser.new(data)
+
+    error = assert_raises InfluxDB2::FluxQueryError do
+      parser.parse
+    end
+
+    assert_equal 'failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time',
+                 error.message
+    assert_equal 897, error.reference
+  end
+
+  def test_error_without_reference
+    data = "#datatype,string,string\n" \
+      "#group,true,true\n" \
+      "#default,,\n" \
+      ",error,reference\n" \
+      ',failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time,'
+
+    parser = InfluxDB2::FluxCsvParser.new(data)
+
+    error = assert_raises InfluxDB2::FluxQueryError do
+      parser.parse
+    end
+
+    assert_equal 'failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time',
+                 error.message
+    assert_equal 0, error.reference
+  end
+
+  def test_without_table_definition
+    data = ",result,table,_start,_stop,_time,_value,_field,_measurement,host,value\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,12.25\n" \
+      ",,0,1970-01-01T00:00:10Z,1970-01-01T00:00:20Z,1970-01-01T00:00:10Z,10,free,mem,A,\n"
+
+    parser = InfluxDB2::FluxCsvParser.new(data)
+
+    error = assert_raises InfluxDB2::FluxCsvParserError do
+      parser.parse
+    end
+
+    assert_equal 'Unable to parse CSV response. FluxTable definition was not found.', error.message
+  end
+end