<s> '''<STR_LIT>''' <EOL> from __future__ import absolute_import <EOL> from salttesting import skipIf , TestCase <EOL> from salttesting . mock import ( <EOL> NO_MOCK , <EOL> NO_MOCK_REASON , <EOL> patch ) <EOL> from salttesting . helpers import ensure_in_syspath <EOL> ensure_in_syspath ( '<STR_LIT>' ) <EOL> from salt . modules import riak <EOL> riak . __salt__ = { } <EOL> @ skipIf ( NO_MOCK , NO_MOCK_REASON ) <EOL> class RiakTestCase ( TestCase ) : <EOL> '''<STR_LIT>''' <EOL> def test_start ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : '<STR_LIT:success>' } ) : <EOL> self . assertEqual ( <EOL> riak . start ( ) , { '<STR_LIT:success>' : True , '<STR_LIT>' : '<STR_LIT:success>' } <EOL> ) <EOL> def test_stop ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : '<STR_LIT:success>' } ) : <EOL> self . assertEqual ( <EOL> riak . stop ( ) , { '<STR_LIT:success>' : True , '<STR_LIT>' : '<STR_LIT:success>' } <EOL> ) <EOL> def test_cluster_join ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : '<STR_LIT:success>' } ) : <EOL> self . assertEqual ( <EOL> riak . cluster_join ( '<STR_LIT:A>' , '<STR_LIT:B>' ) , { '<STR_LIT:success>' : True , '<STR_LIT>' : '<STR_LIT:success>' } <EOL> ) <EOL> def test_cluster_leave ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : '<STR_LIT:success>' } ) : <EOL> self . assertEqual ( <EOL> riak . cluster_leave ( '<STR_LIT:A>' , '<STR_LIT:B>' ) , { '<STR_LIT:success>' : True , '<STR_LIT>' : '<STR_LIT:success>' } <EOL> ) <EOL> def test_cluster_plan ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : '<STR_LIT:success>' } ) : <EOL> self . assertTrue ( riak . cluster_plan ( ) ) <EOL> def test_cluster_commit ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : '<STR_LIT:success>' } ) : <EOL> self . assertEqual ( <EOL> riak . cluster_commit ( ) , { '<STR_LIT:success>' : True , '<STR_LIT>' : '<STR_LIT:success>' } <EOL> ) <EOL> def test_member_status ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : '<STR_LIT>' } ) : <EOL> self . assertDictEqual ( riak . member_status ( ) , <EOL> { '<STR_LIT>' : { } , <EOL> '<STR_LIT>' : { '<STR_LIT:A>' : '<STR_LIT:a>' , '<STR_LIT:C>' : '<STR_LIT:c>' , '<STR_LIT:B>' : '<STR_LIT:b>' , <EOL> '<STR_LIT:D>' : '<STR_LIT:d>' , '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : <NUM_LIT:0> } } ) <EOL> def test_status ( self ) : <EOL> '''<STR_LIT>''' <EOL> ret = { '<STR_LIT>' : '<STR_LIT>' } <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = ret ) : <EOL> self . assertEqual ( <EOL> riak . status ( ) , { '<STR_LIT>' : '<STR_LIT:0>' , '<STR_LIT>' : '<STR_LIT:0>' } <EOL> ) <EOL> def test_test ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : '<STR_LIT:success>' } ) : <EOL> self . assertEqual ( <EOL> riak . test ( ) , { '<STR_LIT:success>' : True , '<STR_LIT>' : '<STR_LIT:success>' } <EOL> ) <EOL> def test_services ( self ) : <EOL> '''<STR_LIT>''' <EOL> with patch . object ( riak , '<STR_LIT>' , return_value = { '<STR_LIT>' : '<STR_LIT>' } ) : <EOL> self . assertEqual ( riak . services ( ) , [ '<STR_LIT:a>' , '<STR_LIT:b>' , '<STR_LIT:c>' ] ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> from integration import run_tests <EOL> run_tests ( RiakTestCase , needs_daemon = False ) </s>
