~mpontillo/maas/beaconing--replies-from-maas--wip

« back to all changes in this revision

Viewing changes to src/maasserver/triggers/tests/test_system_listener.py

  • Committer: Mike Pontillo
  • Date: 2017-06-19 22:29:18 UTC
  • mfrom: (6085.2.8 maas)
  • Revision ID: mike.pontillo@canonical.com-20170619222918-4pzgkvgr9b7cbob2
Merge trunk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
3713
3713
                % (json.dumps(kms_host_old), json.dumps(kms_host_new))))
3714
3714
 
3715
3715
 
3716
 
class TestProxySubnetListener(
 
3716
class TestProxyListener(
3717
3717
        MAASTransactionServerTestCase, TransactionalHelpersMixin):
3718
3718
    """End-to-end test for the proxy triggers code."""
3719
3719
 
3787
3787
            yield dv.get(timeout=2)
3788
3788
        finally:
3789
3789
            yield listener.stopService()
 
3790
 
 
3791
    @wait_for_reactor
 
3792
    @inlineCallbacks
 
3793
    def test_sends_message_for_config_insert_enable_proxy(self):
 
3794
        yield deferToDatabase(register_system_triggers)
 
3795
        dv = DeferredValue()
 
3796
        listener = self.make_listener_without_delay()
 
3797
        listener.register(
 
3798
            "sys_proxy", lambda *args: dv.set(args))
 
3799
        yield listener.startService()
 
3800
        try:
 
3801
            yield deferToDatabase(self.create_config, "enable_proxy", True)
 
3802
            yield dv.get(timeout=2)
 
3803
        finally:
 
3804
            yield listener.stopService()
 
3805
 
 
3806
    @wait_for_reactor
 
3807
    @inlineCallbacks
 
3808
    def test_sends_message_for_config_insert_use_peer_proxy(self):
 
3809
        yield deferToDatabase(register_system_triggers)
 
3810
        dv = DeferredValue()
 
3811
        listener = self.make_listener_without_delay()
 
3812
        listener.register(
 
3813
            "sys_proxy", lambda *args: dv.set(args))
 
3814
        yield listener.startService()
 
3815
        try:
 
3816
            yield deferToDatabase(self.create_config, "use_peer_proxy", True)
 
3817
            yield dv.get(timeout=2)
 
3818
        finally:
 
3819
            yield listener.stopService()
 
3820
 
 
3821
    @wait_for_reactor
 
3822
    @inlineCallbacks
 
3823
    def test_sends_message_for_config_insert_http_proxy(self):
 
3824
        yield deferToDatabase(register_system_triggers)
 
3825
        dv = DeferredValue()
 
3826
        listener = self.make_listener_without_delay()
 
3827
        listener.register(
 
3828
            "sys_proxy", lambda *args: dv.set(args))
 
3829
        yield listener.startService()
 
3830
        try:
 
3831
            yield deferToDatabase(
 
3832
                self.create_config, "http_proxy", "http://proxy.example.com")
 
3833
            yield dv.get(timeout=2)
 
3834
        finally:
 
3835
            yield listener.stopService()
 
3836
 
 
3837
    @wait_for_reactor
 
3838
    @inlineCallbacks
 
3839
    def test_sends_message_for_config_update_enable_proxy(self):
 
3840
        yield deferToDatabase(register_system_triggers)
 
3841
        yield deferToDatabase(self.create_config, "enable_proxy", True)
 
3842
        dv = DeferredValue()
 
3843
        listener = self.make_listener_without_delay()
 
3844
        listener.register(
 
3845
            "sys_proxy", lambda *args: dv.set(args))
 
3846
        yield listener.startService()
 
3847
        try:
 
3848
            yield deferToDatabase(self.set_config, "enable_proxy", False)
 
3849
            yield dv.get(timeout=2)
 
3850
        finally:
 
3851
            yield listener.stopService()
 
3852
 
 
3853
    @wait_for_reactor
 
3854
    @inlineCallbacks
 
3855
    def test_sends_message_for_config_update_use_peer_proxy(self):
 
3856
        yield deferToDatabase(register_system_triggers)
 
3857
        yield deferToDatabase(self.create_config, "use_peer_proxy", True)
 
3858
        dv = DeferredValue()
 
3859
        listener = self.make_listener_without_delay()
 
3860
        listener.register(
 
3861
            "sys_proxy", lambda *args: dv.set(args))
 
3862
        yield listener.startService()
 
3863
        try:
 
3864
            yield deferToDatabase(self.set_config, "use_peer_proxy", False)
 
3865
            yield dv.get(timeout=2)
 
3866
        finally:
 
3867
            yield listener.stopService()
 
3868
 
 
3869
    @wait_for_reactor
 
3870
    @inlineCallbacks
 
3871
    def test_sends_message_for_config_update_http_proxy(self):
 
3872
        yield deferToDatabase(register_system_triggers)
 
3873
        yield deferToDatabase(
 
3874
            self.create_config, "http_proxy", "http://proxy1.example.com")
 
3875
        dv = DeferredValue()
 
3876
        listener = self.make_listener_without_delay()
 
3877
        listener.register(
 
3878
            "sys_proxy", lambda *args: dv.set(args))
 
3879
        yield listener.startService()
 
3880
        try:
 
3881
            yield deferToDatabase(
 
3882
                self.set_config, "http_proxy", "http://proxy2.example.com")
 
3883
            yield dv.get(timeout=2)
 
3884
        finally:
 
3885
            yield listener.stopService()