1 files changed, 55 insertions, 0 deletions
diff --git a/troveclient/tests/osc/v1/test_database_logs.py b/troveclient/tests/osc/v1/test_database_logs.py
index 1654380..9def9e8 100644
--- a/troveclient/tests/osc/v1/test_database_logs.py
+++ b/troveclient/tests/osc/v1/test_database_logs.py
@@ -47,3 +47,58 @@ class TestLogList(TestLogs):
         columns, data = self.cmd.take_action(parsed_args)
         self.assertEqual(self.columns, columns)
         self.assertEqual(self.values, data)
+
+
+class TestShowDatabaseInstanceLog(TestLogs):
+
+    def setUp(self):
+        super(TestShowDatabaseInstanceLog, self).setUp()
+        self.cmd = database_logs.ShowDatabaseInstanceLog(self.app, None)
+        self.columns = (
+            'container',
+            'metafile',
+            'name',
+            'pending',
+            'prefix',
+            'published',
+            'status',
+            'type',
+        )
+
+    @mock.patch.object(utils, 'find_resource')
+    def test_show_instance_log(self, mock_find):
+        mock_find.return_value = 'fake_instance_id'
+        data = self.fake_logs.get_logs()[0]
+        self.instance_client.log_show.return_value = data
+
+        args = ['instance', 'logname']
+        parsed_args = self.check_parser(self.cmd, args, [])
+        columns, values = self.cmd.take_action(parsed_args)
+
+        self.assertEqual(self.columns, columns)
+        self.assertItemsEqual(data.to_dict().values(), values)
+
+
+class TestSetDatabaseInstanceLog(TestLogs):
+    def setUp(self):
+        super(TestSetDatabaseInstanceLog, self).setUp()
+        self.cmd = database_logs.SetDatabaseInstanceLog(self.app, None)
+
+    @mock.patch.object(utils, 'find_resource')
+    def test_set_instance_log(self, mock_find):
+        mock_find.return_value = 'fake_instance_id'
+        data = self.fake_logs.get_logs()[0]
+        data.status = 'Ready'
+        self.instance_client.log_action.return_value = data
+
+        args = ['instance1', 'log_name', '--enable']
+        parsed_args = self.check_parser(self.cmd, args, [])
+        self.cmd.take_action(parsed_args)
+
+        self.instance_client.log_action.assert_called_once_with(
+            'fake_instance_id', 'log_name',
+            enable=True,
+            disable=False,
+            discard=False,
+            publish=False
+        )