class TestDfsr(unittest.TestCase):
def setUp(self):
self.dfsr = DfsrQuery('servername')
self.dfsr.wmi = Mock()
self.query_result = self.dfsr.wmi.make_query
def tearDown(self):
pass
def test_get_dfsr_state(self):
mock_state = Mock()
mock_state.State = 'STATE'
self.query_result.return_value = [ mock_state ]
state = self.dfsr.get_dfsr_state()
self.assertEqual(state, 'STATE')
def test_get_dfsr_state_down(self):
self.query_result.return_value = []
state = self.dfsr.get_dfsr_state()
self.assertEqual(state, 'Service offline')
def test_get_replication_status_counts(self):
result = self._make_results(['State','ReplicationGroupName'])
self.query_result.return_value = [
result('State1','Name1'),
result('State1','Name2'),
result('State2','Name3'),
]
results = self.dfsr.get_replication_status_counts()
self.assertEqual(results['State1'], ['Name1','Name2'])
self.assertEqual(results['State2'], ['Name3'])
def test_get_replication_status_counts_invalid(self):
self.dfsr.wmi.make_query.return_value = []
results = self.dfsr.get_replication_status_counts()
self.assertEqual(len(results), 0)
def _make_results(self, properties):
query_result = namedtuple('query_result', properties)
return query_result
def setUp(self):
self.dfsr = DfsrQuery('servername')
self.dfsr.wmi = Mock()
self.query_result = self.dfsr.wmi.make_query
