diff --git a/src/Adapter/Http/Writer.php b/src/Adapter/Http/Writer.php index 9c1e518..8863c47 100644 --- a/src/Adapter/Http/Writer.php +++ b/src/Adapter/Http/Writer.php @@ -2,11 +2,14 @@ namespace InfluxDB\Adapter\Http; use GuzzleHttp\Client; +use InfluxDB\Adapter\WriterTrait; use InfluxDB\Adapter\Http\Options; -use InfluxDB\Adapter\WriterAbstract; +use InfluxDB\Adapter\WritableInterface; -class Writer extends WriterAbstract +class Writer implements WritableInterface { + use WriterTrait; + private $httpClient; private $options; diff --git a/src/Adapter/Udp/Writer.php b/src/Adapter/Udp/Writer.php index dbe49aa..7f19d77 100644 --- a/src/Adapter/Udp/Writer.php +++ b/src/Adapter/Udp/Writer.php @@ -1,11 +1,14 @@ getMockBuilder("InfluxDB\\Adapter\\WriterAbstract") - ->getMockForAbstractClass(); + $helper = $this->getMockBuilder("InfluxDB\\Adapter\\WriterTrait") + ->getMockForTrait(); $method = new ReflectionMethod(get_class($helper), "pointsToString"); $method->setAccessible(true);