sacrifice 0.0.9 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b2a480e21311ddad2b06ed17eec6db93900d6ecf
-  data.tar.gz: 4698c126d90037eca38deceb3d46c64c54412a8d
+  metadata.gz: c74606f632f956725e2bd4fea9789a22cee4b201
+  data.tar.gz: f409039dd61db39757abdcd176a9661150f05b05
 SHA512:
-  metadata.gz: 9d77a7ec5f833a477a5fcf6fe0f396f1551ac9eb561cb5b31e06e657ab4ead589fbb0bcefacdb2b96847e537c418d9f3fbce1f0e7db3b5c52afd8557cf66c556
-  data.tar.gz: c539c41a2e6001c929883a846d89f4363feadebfa940886edeb4240192514c46b5fd348c5b7c9ee83b7d34b815cccecaa9580f2ce0ee451fad48e5a44753fa37
+  metadata.gz: a513769df721b332c6a0517f18d3869a4091fdcd75d6a316125fc40bfd2dc2328767854251933776693db43c7a5a5280a0074d2091d25cf6a08b8279c66a6ddc
+  data.tar.gz: 6914f39fad533258a3861e1481794a0ca2a47cd4050440d715f0e0967974e983738a8223e801769ab579a6a2a5b434f4e641d265642568a2eedf340c16d83c77

data/lib/sacrifice/app.rb

@@ -70,6 +70,12 @@ class App
     @access_token ||= AccessToken.get(id, secret)
   end
 
+  def find_user(user_id)
+    users.find { |user|
+      user.id.to_s == user_id.to_s
+    }
+  end
+
   private
 
   def users_url
@@ -101,10 +107,4 @@ class App
     end
     app
   end
-
-  def find_user(user_id)
-    users.find { |user|
-      user.id.to_s == user_id.to_s
-    }
-  end
 end

data/lib/sacrifice/csv.rb

@@ -5,12 +5,14 @@ require 'json'
 class Csv
   CREATE_OPTIONS = [:name, :install, :locale]
   CHANGE_OPTIONS = [:password]
-  OUTPUT_ORDER = [:id, :email, :password, :login_url, :access_token]
+  OUTPUT_ORDER = [:id, :name, :email, :password, :login_url, :access_token]
 
-  def self.generate app_name, file
+  def self.generate app_name, file, friends_file
     app = App.find! app_name
     output = init_output(app_name, file)
 
+    friends = ids(app, friends_file)
+
     headers = []
     CSV.read(file, headers: true, header_converters: :symbol).each { |data|
       if headers.empty?
@@ -40,12 +42,13 @@ class Csv
       # execute change
       if change_options.any? and user.change(change_options)
         user.password = change_options[:password] if change_options[:password]
-      else
+      elsif change_options[:password]
         puts "Failed to update password to #{change_options[:password]}"
       end
 
       # print created user
       user_map = user.attrs
+      user_map[:name] = create_options[:name] if create_options[:name]
       puts user_map.map { |key, value| "#{key.upcase} : #{value}" }
 
       # log created user
@@ -56,6 +59,9 @@ class Csv
         }
         line << output_line
       } unless user.id.nil?
+
+      user = app.find_user(user.id)
+      friends user, friends
     }
   end
 
@@ -79,6 +85,23 @@ class Csv
     }
   end
 
+  def self.ids app, csv_file
+    ids = []
+    CSV.read(csv_file, headers: true, header_converters: :symbol).each { |data|
+      ids.push app.find_user(data[:id]) or raise Thor::Error, "No user found w/id #{user.inspect}"
+    } if csv_file
+    ids
+  end
+
+  def self.friends user, friends
+    if friends.any?
+      friends.each { |friend|
+        user.send_friend_request_to(friend)
+        friend.send_friend_request_to(user)
+      }
+    end
+  end
+
   def self.init_output(app_name, file)
     output = "sacrificed_#{app_name}_#{file}"
     File.open(output, 'a').close

data/lib/sacrifice/users.rb

@@ -52,8 +52,8 @@ class Users < Thor
 
   desc 'friend', 'Make two of an app\'s users friends'
   method_option 'app', :aliases => %w[-a], :type => :string, :required => true, :banner => 'Name of the app'
-  method_option 'user1', :aliases => %w[-1 -u1], :type => :string, :required => true, :banner => 'First user ID'
-  method_option 'user2', :aliases => %w[-2 -u2], :type => :string, :required => true, :banner => 'Second user ID'
+  method_option 'user1', :aliases => %w[-1], :type => :string, :required => true, :banner => 'First user ID'
+  method_option 'user2', :aliases => %w[-2], :type => :string, :required => true, :banner => 'Second user ID'
 
   def friend
     users = App.find!(options[:app]).users
@@ -64,7 +64,7 @@ class Users < Thor
     }
 
     # The first request is just a request, the second request accepts the first request.
-    friends.each_index { |idx| friends[idx].send_friend_request_to[(idx + 1) % 2] }
+    friends.each_index { |idx| friends[idx].send_friend_request_to(friends[(idx + 1) % 2]) }
   end
 
   desc 'change', 'Change a test user\'s name and/or password'
@@ -145,9 +145,10 @@ class Users < Thor
   desc 'generate', 'Generate facebook test users'
   method_option 'app', aliases: %w[-a], type: :string, banner: 'app name that test user generate on' #, required: true
   method_option 'file', aliases: %w[-f], type: :string, banner: 'csv file to read (required in type csv)', required: true
+  method_option 'friends', type: :string, banner: 'csv file that include friend ids for generated users(required in type csv)'
 
   def generate
-    Csv.generate options[:app], options[:file]
+    Csv.generate options[:app], options[:file], options[:friends]
   end
 
   desc 'erase', 'Erase facebook test users'
@@ -157,4 +158,16 @@ class Users < Thor
   def erase
     Csv.erase options[:app], options[:file]
   end
+
+
+  desc 'friends', 'Make friends for user'
+  method_option 'app', :aliases => %w[-a], :type => :string, :required => true, :banner => 'Name of the app'
+  method_option 'user', :aliases => %w[-u], :type => :string, :required => true, :banner => 'User ID'
+  method_option 'friends', :aliases => %w[-f], :type => :string, :required => true, :banner => 'Csv file lists friends IDs'
+
+  def friends
+    app = App.find!(options[:app])
+    user = app.find_user(options[:user])
+    Csv.friends user, Csv.ids(app, options[:friends])
+  end
 end

data/lib/sacrifice/version.rb

@@ -1,3 +1,3 @@
 module Sacrifice
-  VERSION = "0.0.9"
+  VERSION = "0.0.12"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sacrifice
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.12
 platform: ruby
 authors:
 - toiroakr
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-09-23 00:00:00.000000000 Z
+date: 2016-11-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler