roku_builder 4.25.4 → 4.26.0

data/test/roku_builder/plugins/test_packager.rb

@@ -10,9 +10,12 @@ module RokuBuilder
       RokuBuilder.setup_plugins
       register_plugins(Packager)
       @requests = []
+      @device_manager = Minitest::Mock.new
     end
     def teardown
       @requests.each {|req| remove_request_stub(req)}
+      @device_manager.verify
+      clean_device_locks()
     end
     def test_packager_parse_options_long
       parser = OptionParser.new
@@ -61,6 +64,10 @@ module RokuBuilder
     end
     def test_packager_package_failed
       config, options = build_config_options_objects(PackagerTest, {package: true, stage: "production"}, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       @requests.push(stub_request(:post, "http://192.168.0.100:8060/keypress/Home").
         to_return(status: 200, body: "", headers: {}))
       @requests.push(stub_request(:post, "http://192.168.0.100/plugin_install").
@@ -68,8 +75,10 @@ module RokuBuilder
       @requests.push(stub_request(:post, "http://192.168.0.100/plugin_package").
         to_return(status: 200, body: "Failed: Error.", headers: {}))
       packager = Packager.new(config: config)
-      assert_raises ExecutionError do
-        packager.package(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        assert_raises ExecutionError do
+          packager.package(options: options)
+        end
       end
     end
     def test_packager_package
@@ -88,9 +97,13 @@ module RokuBuilder
       @requests.push(stub_request(:get, "http://192.168.0.100/pkgs/pkg_url").
         to_return(status: 200, body: body, headers: {}))
 
-      loader.expect(:sideload, nil, [Hash])
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
+      loader.expect(:sideload, nil, options: Hash, device: device)
       io.expect(:write, nil, ["package_body"])
-      inspector.expect(:inspect, nil, [Hash])
+      inspector.expect(:inspect, nil, options: Hash, device: device)
 
       logger.expect(:debug, nil, [String])
       io.expect(:each_line, nil)
@@ -101,12 +114,14 @@ module RokuBuilder
       Logger.class_variable_set(:@@instance, logger)
       packager = Packager.new(config: config)
       dev_id = Proc.new {"#{Random.rand(999999999999)}"}
-      Loader.stub(:new, loader) do
-        Time.stub(:now, Time.at(0)) do
-          File.stub(:open, nil, io) do
-            Inspector.stub(:new, inspector) do
-              packager.stub(:dev_id, dev_id) do
-                packager.package(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Loader.stub(:new, loader) do
+          Time.stub(:now, Time.at(0)) do
+            File.stub(:open, nil, io) do
+              Inspector.stub(:new, inspector) do
+                packager.stub(:dev_id, dev_id) do
+                  packager.package(options: options)
+                end
               end
             end
           end
@@ -126,6 +141,10 @@ module RokuBuilder
       config[:projects][:project1][:stages][:production][:squash] = true
       config, options = build_config_options_objects(PackagerTest, {package: true, stage: "production", inspect_package: true, verbose: true}, false, config)
 
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       @requests.push(stub_request(:post, "http://192.168.0.100/plugin_inspect").
         to_return(status: 200, body: "", headers: {}).times(2))
       body = "<a href=\"pkgs\">pkg_url</a>"
@@ -135,10 +154,10 @@ module RokuBuilder
       @requests.push(stub_request(:get, "http://192.168.0.100/pkgs/pkg_url").
         to_return(status: 200, body: body, headers: {}))
 
-      loader.expect(:sideload, nil, [Hash])
-      loader.expect(:squash, nil, [Hash])
+      loader.expect(:sideload, nil, options: Hash, device: device)
+      loader.expect(:squash, nil, options: Hash, device: device)
       io.expect(:write, nil, ["package_body"])
-      inspector.expect(:inspect, nil, [Hash])
+      inspector.expect(:inspect, nil, options: Hash, device: device)
 
       logger.expect(:debug, nil, [String])
       io.expect(:each_line, nil)
@@ -149,12 +168,14 @@ module RokuBuilder
       Logger.class_variable_set(:@@instance, logger)
       packager = Packager.new(config: config)
       dev_id = Proc.new {"#{Random.rand(999999999999)}"}
-      Loader.stub(:new, loader) do
-        Time.stub(:now, Time.at(0)) do
-          File.stub(:open, nil, io) do
-            Inspector.stub(:new, inspector) do
-              packager.stub(:dev_id, dev_id) do
-                packager.package(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Loader.stub(:new, loader) do
+          Time.stub(:now, Time.at(0)) do
+            File.stub(:open, nil, io) do
+              Inspector.stub(:new, inspector) do
+                packager.stub(:dev_id, dev_id) do
+                  packager.package(options: options)
+                end
               end
             end
           end
@@ -171,8 +192,16 @@ module RokuBuilder
         to_return(status: 200, body: body, headers: {}))
 
       config = build_config_options_objects(PackagerTest, {key: true, stage: "production"}, false)[0]
+
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       packager = Packager.new(config: config)
-      dev_id = packager.dev_id
+      dev_id = nil
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        dev_id = packager.dev_id
+      end
 
       assert_equal "dev_id", dev_id
     end
@@ -182,8 +211,15 @@ module RokuBuilder
         to_return(status: 200, body: body, headers: {}))
 
       config = build_config_options_objects(PackagerTest, {key: true, stage: "production"}, false)[0]
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       packager = Packager.new(config: config)
-      dev_id = packager.dev_id
+      dev_id = nil
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        dev_id = packager.dev_id
+      end
 
       assert_equal "dev_id", dev_id
     end
@@ -195,10 +231,16 @@ module RokuBuilder
       logger.expect(:debug, nil) {|s| s =~ /\d* -> \d*/}
       dev_id = Proc.new {"#{Random.rand(999999999999)}"}
       config, options = build_config_options_objects(PackagerTest, {key: true, stage: "production"}, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       packager = Packager.new(config: config)
       Logger.class_variable_set(:@@instance, logger)
-      packager.stub(:dev_id, dev_id) do
-        packager.key(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        packager.stub(:dev_id, dev_id) do
+          packager.key(options: options)
+        end
       end
     end
 
@@ -210,9 +252,32 @@ module RokuBuilder
       logger.expect(:debug, nil) {|s| s =~ /\d* -> \d*/}
       dev_id = Proc.new {"#{Random.rand(999999999999)}"}
       config, options = build_config_options_objects(PackagerTest, {key: true, stage: "production"}, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       packager = Packager.new(config: config)
       Logger.class_variable_set(:@@instance, logger)
-      packager.stub(:dev_id, dev_id) do
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        packager.stub(:dev_id, dev_id) do
+          packager.key(options: options)
+        end
+      end
+    end
+
+    def test_packager_key_same_device
+      config, options = build_config_options_objects(PackagerTest, {key: true, stage: "production"}, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
+      body = "<p> Your Dev ID: <font face=\"Courier\">dev_id</font> </p>"
+      @requests.push(stub_request(:get, "http://192.168.0.100/plugin_package").
+        to_return(status: 200, body: body, headers: {}))
+      @requests.push(stub_request(:post, "http://192.168.0.100/plugin_inspect").
+        to_return(status: 200, body: "", headers: {}))
+      packager = Packager.new(config: config)
+      RokuBuilder.stub(:device_manager, @device_manager) do
         packager.key(options: options)
       end
     end
@@ -230,9 +295,15 @@ module RokuBuilder
       connection.expect(:close, nil, [])
 
       config = build_config_options_objects(PackagerTest, {genkey: true}, false)[0]
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       packager = Packager.new(config: config)
-      Net::Telnet.stub(:new, connection) do
-        packager.send(:generate_new_key)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          packager.send(:generate_new_key)
+        end
       end
     end
 
@@ -240,18 +311,22 @@ module RokuBuilder
       config = good_config(PackagerTest)
       config[:projects][:project1][:stages][:production].delete(:key)
       config, options = build_config_options_objects(PackagerTest, {key: true, stage: "production"}, false, config)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
       packager = Packager.new(config: config)
       dev_id = Proc.new {"#{Random.rand(999999999999)}"}
       assert_raises ExecutionError do
-        packager.stub(:dev_id, dev_id) do
-          packager.key(options: options)
+        RokuBuilder.stub(:device_manager, @device_manager) do
+          packager.stub(:dev_id, dev_id) do
+            packager.key(options: options)
+          end
         end
       end
     end
 
     def test_packager_genkey
-      loader = Minitest::Mock.new
-      loader.expect(:sideload, nil, [Hash])
 
       body = "<a href=\"pkgs\">pkg_url</a>"
       @requests.push(stub_request(:post, "http://192.168.0.100/plugin_package").
@@ -259,13 +334,22 @@ module RokuBuilder
       @requests.push(stub_request(:get, "http://192.168.0.100/pkgs/pkg_url").
         to_return(status: 200, body: "", headers: {}))
       config, options = build_config_options_objects(PackagerTest, {genkey: true}, false)
+
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: false)
+      @device_manager.expect(:release_device, nil, [device])
+
+      loader = Minitest::Mock.new
+      loader.expect(:sideload, nil, options: Hash)
+
       packager = Packager.new(config: config)
-      Loader.stub(:new, loader) do
-        packager.stub(:generate_new_key, ["password", "dev_id"]) do
-          packager.genkey(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Loader.stub(:new, loader) do
+          packager.stub(:generate_new_key, ["password", "dev_id"]) do
+            packager.genkey(options: options)
+          end
         end
       end
-
       loader.verify
     end
   end

data/test/roku_builder/plugins/test_profiler.rb

@@ -11,6 +11,10 @@ module RokuBuilder
       unless RokuBuilder.plugins.include?(Profiler)
         RokuBuilder.register_plugin(Profiler)
       end
+      @device_manager = Minitest::Mock.new
+    end
+    def teardown
+      @device_manager.verify
     end
     def test_profiler_parse_options_long
       parser = OptionParser.new
@@ -23,12 +27,16 @@ module RokuBuilder
     def test_profiler_node_tracking
       options = {profile: "stats"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = "<All_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></All_Nodes>\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = "<All_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></All_Nodes>\n"
+      blk.call(txt)
+      true
       end
       count = 0
       read_stub = Proc.new do |size|
@@ -43,10 +51,12 @@ module RokuBuilder
       connection.expect(:waitfor, nil, &waitfor)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:printf, nil) do
-          STDIN.stub(:read, read_stub) do
-            profiler.node_tracking(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:printf, nil) do
+            STDIN.stub(:read, read_stub) do
+              profiler.node_tracking(options: options)
+            end
           end
         end
       end
@@ -56,12 +66,16 @@ module RokuBuilder
     def test_profiler_stats
       options = {profile: "stats"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = "<All_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></All_Nodes>\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = "<All_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></All_Nodes>\n"
+      blk.call(txt)
+      true
       end
       connection = Minitest::Mock.new
       profiler = Profiler.new(config: config)
@@ -70,9 +84,11 @@ module RokuBuilder
       connection.expect(:waitfor, nil, &waitfor)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:printf, nil) do
-          profiler.profile(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:printf, nil) do
+            profiler.profile(options: options)
+          end
         end
       end
 
@@ -81,12 +97,16 @@ module RokuBuilder
     def test_profiler_all
       options = {profile: "all"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = "<All_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></All_Nodes>\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = "<All_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></All_Nodes>\n"
+      blk.call(txt)
+      true
       end
       connection = Minitest::Mock.new
       profiler = Profiler.new(config: config)
@@ -95,9 +115,11 @@ module RokuBuilder
       connection.expect(:waitfor, nil, &waitfor)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:print, nil) do
-          profiler.profile(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:print, nil) do
+            profiler.profile(options: options)
+          end
         end
       end
 
@@ -106,12 +128,16 @@ module RokuBuilder
     def test_profiler_roots
       options = {profile: "roots"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = "<Root_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></Root_Nodes>\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = "<Root_Nodes><NodeA /><NodeB /><NodeC><NodeD /></NodeC></Root_Nodes>\n"
+      blk.call(txt)
+      true
       end
       connection = Minitest::Mock.new
       profiler = Profiler.new(config: config)
@@ -120,9 +146,11 @@ module RokuBuilder
       connection.expect(:waitfor, nil, &waitfor)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:print, nil) do
-          profiler.profile(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:print, nil) do
+            profiler.profile(options: options)
+          end
         end
       end
       connection.verify
@@ -130,12 +158,16 @@ module RokuBuilder
     def test_profiler_node
       options = {profile: "nodeId"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = "<nodeId><NodeA /><NodeB /><NodeC><NodeD /></NodeC></nodeId>\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = "<nodeId><NodeA /><NodeB /><NodeC><NodeD /></NodeC></nodeId>\n"
+      blk.call(txt)
+      true
       end
       connection = Minitest::Mock.new
       profiler = Profiler.new(config: config)
@@ -144,9 +176,11 @@ module RokuBuilder
       connection.expect(:waitfor, nil, &waitfor)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:print, nil) do
-          profiler.profile(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:print, nil) do
+            profiler.profile(options: options)
+          end
         end
       end
       connection.verify
@@ -154,12 +188,16 @@ module RokuBuilder
     def test_profiler_images
       options = {profile: "images"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = " RoGraphics instance\n0x234 1 2 3 4\nAvailable memory\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = " RoGraphics instance\n0x234 1 2 3 4\nAvailable memory\n"
+      blk.call(txt)
+      true
       end
       connection = Minitest::Mock.new
       profiler = Profiler.new(config: config)
@@ -168,9 +206,11 @@ module RokuBuilder
       connection.expect(:waitfor, nil, &waitfor)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:print, nil) do
-          profiler.profile(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:print, nil) do
+            profiler.profile(options: options)
+          end
         end
       end
 
@@ -180,12 +220,16 @@ module RokuBuilder
     def test_profiler_memmory
       options = {profile: "memmory"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = " RoGraphics instance 0x123\nAvailable memory 123 used 456 max 579\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = " RoGraphics instance 0x123\nAvailable memory 123 used 456 max 579\n"
+      blk.call(txt)
+      true
       end
       print_count = 0
       print_stub = Proc.new do |message|
@@ -216,9 +260,11 @@ module RokuBuilder
       connection.expect(:puts, nil, &puts_stub)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:print, print_stub) do
-          profiler.profile(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:print, print_stub) do
+            profiler.profile(options: options)
+          end
         end
       end
 
@@ -228,15 +274,19 @@ module RokuBuilder
     def test_profiler_textures
       options = {profile: "textures"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       waitfor = Proc.new do |telnet_config, &blk|
-        assert_equal(/.+/, telnet_config["Match"])
-        assert_equal(1, telnet_config["Timeout"])
-        txt = "*******\ntexture\n"
-        blk.call(txt)
-        true
+      assert_equal(/.+/, telnet_config["Match"])
+      assert_equal(1, telnet_config["Timeout"])
+      txt = "*******\ntexture\n"
+      blk.call(txt)
+      true
       end
       timeout = Proc.new do |telnet_config, &blk|
-        raise ::Net::ReadTimeout
+      raise ::Net::ReadTimeout
       end
       connection = Minitest::Mock.new
       profiler = Profiler.new(config: config)
@@ -246,9 +296,11 @@ module RokuBuilder
       connection.expect(:waitfor, nil, &timeout)
       connection.expect(:close, nil)
 
-      Net::Telnet.stub(:new, connection) do
-        profiler.stub(:print, nil) do
-          profiler.profile(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:print, nil) do
+            profiler.profile(options: options)
+          end
         end
       end
 
@@ -257,19 +309,30 @@ module RokuBuilder
     def test_profiler_devlog
       options = {devlog: "rendezvous", devlog_function: "on"}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
 
       connection = Minitest::Mock.new
       connection.expect(:puts, nil, ["enhanced_dev_log rendezvous on\n"])
 
       profiler = Profiler.new(config: config)
-      Net::Telnet.stub(:new, connection) do
-        profiler.devlog(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          profiler.stub(:print, nil) do
+            profiler.devlog(options: options)
+          end
+        end
       end
       connection.verify
     end
     def test_profiler_sgperf
       options = {sgperf: true}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       profiler = Profiler.new(config: config)
 
       connection = Object.new
@@ -298,11 +361,15 @@ module RokuBuilder
         raise Net::ReadTimeout
       }
 
-      Net::Telnet.stub(:new, connection) do
-        connection.stub(:puts, puts_stub) do
-          txt = ">>thread node calls: create     0 + op    24  @ 100.0% rendezvous"
-          connection.stub(:waitfor, waitfor, txt) do
-            profiler.sgperf(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          connection.stub(:puts, puts_stub) do
+            txt = ">>thread node calls: create     0 + op    24  @ 100.0% rendezvous"
+            connection.stub(:waitfor, waitfor, txt) do
+              profiler.stub(:print, nil) do
+                profiler.sgperf(options: options)
+              end
+            end
           end
         end
       end
@@ -315,6 +382,10 @@ module RokuBuilder
     def test_profiler_sgperf_multi_lines
       options = {sgperf: true}
       config, options = build_config_options_objects(ProfilerTest, options, false)
+      device = RokuBuilder::Device.new("roku", config.raw[:devices][:roku])
+      @device_manager.expect(:reserve_device, device, no_lock: true)
+      @device_manager.expect(:release_device, nil, [device])
+
       profiler = Profiler.new(config: config)
 
       connection = Object.new
@@ -357,11 +428,15 @@ module RokuBuilder
         call_count += 1
       }
 
-      Net::Telnet.stub(:new, connection) do
-        connection.stub(:puts, connection_puts) do
-          profiler.stub(:get_command_response, command_response) do
-            profiler.stub(:puts, profiler_puts) do
-              profiler.sgperf(options: options)
+      RokuBuilder.stub(:device_manager, @device_manager) do
+        Net::Telnet.stub(:new, connection) do
+          connection.stub(:puts, connection_puts) do
+            profiler.stub(:get_command_response, command_response) do
+              profiler.stub(:puts, profiler_puts) do
+                profiler.stub(:print, nil) do
+                  profiler.sgperf(options: options)
+                end
+              end
             end
           end
         end