fix unit tests

AppScale · Aug 21, 2019 · 8099b21 · 8099b21
1 parent a5ef9af
commit 8099b21
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/test/test_appscale.py b/test/test_appscale.py
@@ -429,8 +429,6 @@ def testDeployWithCloudAppScalefile(self):
     }
     yaml_dumped_contents = yaml.dump(contents)
     self.addMockForAppScalefile(appscale, yaml_dumped_contents)
-    flexmock(appscale)
-    appscale.should_receive('is_dispatch_yaml').and_return(False)
     # finally, mock out the actual appscale-run-instances call
     fake_port = 8080
     fake_host = 'fake_host'
@@ -440,6 +438,7 @@ def testDeployWithCloudAppScalefile(self):
     AppScaleTools.should_receive('update_indexes')
     AppScaleTools.should_receive('update_cron')
     AppScaleTools.should_receive('update_queues')
+    AppScaleTools.should_receive('update_dispatch')
     app = '/bar/app'
     (host, port) = appscale.deploy(app)
     self.assertEquals(fake_host, host)
@@ -500,8 +499,6 @@ def testDeployWithCloudAppScalefileAndTestFlag(self):
     }
     yaml_dumped_contents = yaml.dump(contents)
     self.addMockForAppScalefile(appscale, yaml_dumped_contents)
-    flexmock(appscale)
-    appscale.should_receive('is_dispatch_yaml').and_return(False)
     # finally, mock out the actual appscale-run-instances call
     fake_port = 8080
     fake_host = 'fake_host'
@@ -511,6 +508,7 @@ def testDeployWithCloudAppScalefileAndTestFlag(self):
     AppScaleTools.should_receive('update_indexes')
     AppScaleTools.should_receive('update_cron')
     AppScaleTools.should_receive('update_queues')
+    AppScaleTools.should_receive('update_dispatch')
     app = '/bar/app'
     (host, port) = appscale.deploy(app)
     self.assertEquals(fake_host, host)