← Back to branch summary

~free.ekanayaka/landscape-client/amp-cleanup-7

« back to all changes in this revision

Viewing changes to landscape/tests/helpers.py

  • Committer: Free Ekanayaka
  • Date: 2013-04-15 07:16:16 UTC
  • Revision ID: free.ekanayaka@canonical.com-20130415071616-yf3xhpbuteh9m3bx
Add successResultOf and failureResultOf to LandscapeTest, taken from Twisted 12.3.0 [trivial] [r=cglass]

Show diffs side-by-side

added added

removed removed

Lines of Context:
landscape/tests/helpers.py
109
109
        HelperTestCase.tearDown(self)
110
110
        MockerTestCase.tearDown(self)
111
111
 
 
112
    def successResultOf(self, deferred):
 
113
        """See C{twisted.trial._synctest._Assertions.successResultOf}.
 
114
 
 
115
        This is a copy of the original method, which is available only
 
116
        since Twisted 12.3.0 (from 2012-12-20).
 
117
        """
 
118
        result = []
 
119
        deferred.addBoth(result.append)
 
120
        if not result:
 
121
            self.fail(
 
122
                "Success result expected on %r, found no result instead" % (
 
123
                    deferred,))
 
124
        elif isinstance(result[0], failure.Failure):
 
125
            self.fail(
 
126
                "Success result expected on %r, "
 
127
                "found failure result (%r) instead" % (deferred, result[0]))
 
128
        else:
 
129
            return result[0]
 
130
 
 
131
    def failureResultOf(self, deferred):
 
132
        """See C{twisted.trial._synctest._Assertions.failureResultOf}.
 
133
 
 
134
        This is a copy of the original method, which is available only
 
135
        since Twisted 12.3.0 (from 2012-12-20).
 
136
        """
 
137
        result = []
 
138
        deferred.addBoth(result.append)
 
139
        if not result:
 
140
            self.fail(
 
141
                "Failure result expected on %r, found no result instead" % (
 
142
                    deferred,))
 
143
        elif not isinstance(result[0], failure.Failure):
 
144
            self.fail(
 
145
                "Failure result expected on %r, "
 
146
                "found success result (%r) instead" % (deferred, result[0]))
 
147
        else:
 
148
            return result[0]
 
149
 
112
150
    def assertDeferredSucceeded(self, deferred):
113
151
        self.assertTrue(isinstance(deferred, Deferred))
114
152
        called = []