extract endpoint.

1 files changed, 4 insertions, 3 deletions

diff --git a/lib/tasks/agent.rake b/lib/tasks/agent.rake
index 98aa2d5..26d3982 100644
--- a/lib/tasks/agent.rake
+++ b/lib/tasks/agent.rake
@@ -1,21 +1,22 @@
 namespace :agent do
   require 'fake_agent'
+  ENDPOINT='http://localhost:3000'
 
   desc "watch all files"
   task watch: :environment do
-    agent = FakeAgent.new(Agent.first.id, 'http://localhost:3000')
+    agent = FakeAgent.new(Agent.first.id, ENDPOINT)
     agent.watch(Dir.pwd)
   end
 
   desc "scan directory"
   task scan: :environment do
-    agent = FakeAgent.new(Agent.first.id, 'http://localhost:3000')
+    agent = FakeAgent.new(Agent.first.id, ENDPOINT)
     agent.scan(Dir.pwd)
   end
 
   desc "scan network traffic"
   task nfm: :environment do
-    agent = FakeAgent.new(Agent.first.id, 'http://localhost:3000')
+    agent = FakeAgent.new(Agent.first.id, ENDPOINT)
     agent.sniff('en1')
   end
 end