diff --git a/bitelab/__main__.py b/bitelab/__main__.py index f857e42..af2971e 100644 --- a/bitelab/__main__.py +++ b/bitelab/__main__.py @@ -286,6 +286,8 @@ Attributes: self.assertEqual(ret, 0) self.assertEqual(stdout, output) + gspk.assert_called_with(None) + ac.assert_called_with(base_url='http://someserver/') #args = { 'sshpubkey': keydata } @@ -327,6 +329,8 @@ Attributes: json=dict(sshpubkey=keydata), auth=BiteAuth('thisisanapikey'), **_httpxargs) + gspk.assert_called_with('bogusfilename') + @patch('bitelab.__main__.get_sshpubkey') @patch.dict(sys.__dict__, dict(argv=[ '', 'release', 'cora-z7s' ])) def test_release(self, gspk):