helper.rb in datasift-2.0.0

- old
+ new
@@ -10,15 +10,89 @@
 
 class Test::Unit::TestCase
 	def init()
 		@config = YAML::load(File.open(File.join(File.dirname(__FILE__), '..', 'config.yml')))
 		@testdata = YAML::load(File.open(File.join(File.dirname(__FILE__), 'testdata.yml')))
+
+		# Initialise the test data (parse the dates, etc).
+		@testdata['historic_start'] = DateTime.parse(@testdata['historic_start'])
+		@testdata['historic_end'] = DateTime.parse(@testdata['historic_end'])
+		@testdata['historic_created_at'] = DateTime.parse(@testdata['historic_created_at'])
+		@testdata['push_created_at'] = DateTime.parse(@testdata['push_created_at'])
+		@testdata['push_last_request'] = DateTime.parse(@testdata['push_last_request'])
+		@testdata['push_last_success'] = DateTime.parse(@testdata['push_last_success'])
+		@testdata['historic_sources'] = @testdata['historic_sources'].split(',')
+
+		@user = DataSift::User.new(@config['username'], @config['api_key'])
+		@user.setApiClient(DataSift::MockApiClient.new())
+		@user.api_client.clearResponse()
 	end
 
-	def initUser(mock = true)
-			@user = DataSift::User.new(@config['username'], @config['api_key'])
-			if mock
-				@user.setApiClient(DataSift::MockApiClient.new())
-				@user.api_client.clearResponse()
-			end
+	def set204Response()
+		@user.api_client.setResponse(204, {}, 200, 150)
+	end
+
+	def setResponseToASingleHistoric(changes = {})
+		data = {
+				'id' => @testdata['historic_playback_id'],
+				'definition_id' => @testdata['definition_hash'],
+				'name' => @testdata['historic_name'],
+				'start' => @testdata['historic_start'].strftime('%s'),
+				'end' => @testdata['historic_end'].strftime('%s'),
+				'created_at' => @testdata['historic_created_at'].strftime('%s'),
+				'status' => @testdata['historic_status'],
+				'progress' => 0,
+				'sources' => @testdata['historic_sources'],
+				'sample' => @testdata['historic_sample'],
+				'volume_info' => {
+					'digg' => 9
+				},
+			}
+		@user.api_client.setResponse(200, data.merge(changes), 200, 150)
+	end
+
+	def configurePushDefinition(push)
+		push.output_type = @testdata['push_output_type']
+		@testdata['push_output_params'].each { |k,v| push.output_params[k] = v }
+	end
+
+	def setResponseToASingleSubscription(changes = {})
+		data = {
+				'id'            => @testdata['push_id'],
+				'name'          => @testdata['push_name'],
+				'created_at'    => @testdata['push_created_at'].strftime('%s'),
+				'status'        => @testdata['push_status'],
+				'hash'          => @testdata['definition_hash'],
+				'hash_type'     => @testdata['push_hash_type'],
+				'output_type'   => @testdata['push_output_type'],
+				'output_params' => {
+					'delivery_frequency' => @testdata['push_output_params']['delivery_frequency'],
+					'url'                => @testdata['push_output_params']['url'],
+					'auth'               => {
+						'type'      => @testdata['push_output_params']['auth_type'],
+						'username'	=> @testdata['push_output_params']['auth_username'],
+						'password'	=> @testdata['push_output_params']['auth_password'],
+					},
+				},
+				'last_request'  => @testdata['push_last_request'].strftime('%s'),
+				'last_success'  => @testdata['push_last_success'].strftime('%s'),
+			}
+		@user.api_client.setResponse(200, data.merge(changes), 200, 150)
+	end
+
+	def checkSubscription(subscription)
+		assert_equal @testdata['push_id'],                                  subscription.id
+		assert_equal @testdata['push_name'],                                subscription.name
+		assert_equal @testdata['push_created_at'],                          subscription.created_at
+		assert_equal @testdata['push_status'],                              subscription.status
+		assert_equal @testdata['definition_hash'],                          subscription.hash
+		assert_equal @testdata['push_hash_type'],                           subscription.hash_type
+		assert_equal @testdata['push_output_type'],                         subscription.output_type
+		assert_equal @testdata['push_output_params']['delivery_frequency'], subscription.output_params['delivery_frequency']
+		assert_equal @testdata['push_output_params']['url'],                subscription.output_params['url']
+		assert_equal @testdata['push_output_params']['auth_type'],          subscription.output_params['auth.type']
+		assert_equal @testdata['push_output_params']['auth_username'],      subscription.output_params['auth.username']
+		assert_equal @testdata['push_output_params']['auth_password'],      subscription.output_params['auth.password']
+		assert_equal @testdata['push_last_request'],                        subscription.last_request
+		assert_equal @testdata['push_last_success'],                        subscription.last_success
 	end
 end