env_command_spec.rb in kontena-cli-1.4.0.pre6

- old
+ new

@@ -29,18 +29,11 @@
       ]
     end
   end
 
   context 'for a node without any token' do
-    let :node_token do
-      {
-        "id" => nil,
-        "token" => nil,
-      }
-    end
-
     before do
-      expect(client).to receive(:get).with('nodes/test-grid/node-1/token').and_return(node_token)
+      expect(client).to receive(:get).with('nodes/test-grid/node-1/token').and_raise(Kontena::Errors::StandardError.new(404, "Host node does not have a node token"))
     end
 
     it 'uses the grid token' do
       expect{subject.run(['node-1'])}.to exit_with_error.and output(" [error] Node node-1 was not created with a node token. Use `kontena grid env` instead\n").to_stderr
     end