test for capture

tests/integration/WC/TestWooCommerceCapture.php

+<?php
+
+use Monolog\Logger;
+use WPDesk\Logger\WC\Exception\WCLoggerAlreadyCaptured;
+use WPDesk\Logger\WC\WooCommerceCapture;
+use WPDesk\Logger\WC\WooCommerceMonologPlugin;
+
+class TestWooCommerceCapture extends WP_UnitTestCase {
+    public function testIfCanCaptureWcLogger() {
+	    $wcCapture = new WooCommerceCapture($this->createMock(Logger::class));
+        $wcCapture->captureWcLogger();
+
+        $this->assertInstanceOf(WooCommerceMonologPlugin::class, wc_get_logger(), "Logger should be captured.");
+    }
+
+    public function testIfCanFreeWcLogger() {
+        $wcCapture = new WooCommerceCapture($this->createMock(Logger::class));
+        $wcCapture->captureWcLogger();
+        $wcCapture->freeWcLogger();
+        $this->assertNotInstanceOf(WooCommerceMonologPlugin::class, wc_get_logger(), "Logger should be restored to original");
+
+        $wcCapture->captureWcLogger();
+        $this->assertInstanceOf(WooCommerceMonologPlugin::class, wc_get_logger(), "Logger should be captured.");
+        $wcCapture->freeWcLogger();
+        $this->assertNotInstanceOf(WooCommerceMonologPlugin::class, wc_get_logger(), "Logger should be restored to original - twice");
+    }
+
+    public function testIfCantCaptureTwice() {
+        $this->expectException(WCLoggerAlreadyCaptured::class);
+        $wcCapture = new WooCommerceCapture($this->createMock(Logger::class));
+        $wcCapture->captureWcLogger();
+        $wcCapture->captureWcLogger();
+    }
+
+}
+