def test_AddUnitWithoutLogin(self, put):
am = AppManager("target", self.dbn)
capture = py.io.StdCaptureFD(in_=False)
am.unitadd("myapp", 5)
out, err = capture.reset()
put.not_called()
self.assertEquals(out.strip(), ELoginFirst)
def test_AddUnitWithLogin(self, put):
self.loggedin()
am = AppManager("target", self.dbn)
capture = py.io.StdCaptureFD(in_=False)
am.unitadd("myapp", 5)
out, err = capture.reset()
put.assert_called_with("target/apps/myapp/units", data=str(5), headers=TestAuthHeader)
self.assertEquals(out.strip(), IAddUnit)
