(PDOC-206) adds more tests

lib/puppet-strings/yard/code_objects/task.rb

@@ -21,8 +21,7 @@ class PuppetStrings::Yard::CodeObjects::Task < PuppetStrings::Yard::CodeObjects:
   attr_reader :statement
 
   # Initializes a JSON task code object.
-  # @param [String] source The task's JSON file source
-  # @param [String] filepath Path to task's .json file
+  # @param statement TaskStatement object
   # @return [void]
   def initialize(statement)
     @name = statement.name

lib/puppet-strings/yard/parsers/json/task_statement.rb

@@ -1,7 +1,7 @@
 module PuppetStrings::Yard::Parsers::JSON
   # Represents the Puppet Task statement.
   class TaskStatement
-    attr_reader :line, :comments, :comments_range, :json, :file
+    attr_reader :line, :comments, :comments_range, :json, :file, :source
 
     def initialize(json, source, file)
       @file = file

spec/unit/puppet-strings/yard/code_objects/task_spec.rb

@@ -1,9 +1,77 @@
 require 'spec_helper'
 require 'puppet-strings/yard/code_objects/task'
+require 'puppet-strings/yard/parsers/json/task_statement'
 
 describe PuppetStrings::Yard::CodeObjects::Task do
+  let(:source) { <<-SOURCE
+{
+  "description": "Allows you to backup your database to local file.",
+  "input_method": "stdin",
+  "parameters": {
+    "database": {
+      "description": "Database to connect to",
+      "type": "Optional[String[1]]"
+    },
+    "user": {
+      "description": "The user",
+      "type": "Optional[String[1]]"
+    },
+    "password": {
+      "description": "The password",
+      "type": "Optional[String[1]]"
+    },
+     "sql": {
+      "description": "Path to file you want backup to",
+      "type": "String[1]"
+    }
+  }
+}
+SOURCE
+  }
+  let(:json) { JSON.parse(source) }
+  let(:statement) { PuppetStrings::Yard::Parsers::JSON::TaskStatement.new(json, source, "test.json") }
+  subject { PuppetStrings::Yard::CodeObjects::Task.new(statement) }
 
-  it do
-    subject
+  describe '#type' do
+    it 'returns the correct type' do
+      expect(subject.type).to eq(:puppet_task)
+    end
+  end
+
+  describe '#source' do
+    it 'returns the source' do
+      expect(subject.source).to eq(source)
+    end
+  end
+
+  describe '#to_hash' do
+    let(:expected) do
+      { :name => "test",
+        "description" => "Allows you to backup your database to local file.",
+        "input_method" => "stdin",
+        "parameters" =>
+        {"database" => {
+            "description" => "Database to connect to",
+            "type" => "Optional[String[1]]"
+          },
+          "user"=> {
+            "description"=>"The user",
+            "type"=>"Optional[String[1]]"
+          },
+          "password"=> {
+            "description"=>"The password",
+            "type"=>"Optional[String[1]]"
+          },
+          "sql"=> {
+            "description"=>"Path to file you want backup to",
+            "type"=>"String[1]"
+          }
+        }
+      }
+    end
+
+    it 'returns the correct hash' do
+      expect(subject.to_hash).to eq(expected)
+    end
   end
 end