Cover close methods

Signed-off-by: Pavol Loffay <ploffay@redhat.com>

plugin/storage/cassandra/spanstore/writer.go

@@ -118,7 +118,7 @@ func NewSpanWriter(
 
 // Close closes SpanWriter
 func (s *SpanWriter) Close() error {
-	s.Close()
+	s.session.Close()
 	return nil
 }
 

plugin/storage/cassandra/spanstore/writer_test.go

@@ -53,6 +53,14 @@ func withSpanWriter(writeCacheTTL time.Duration, fn func(w *spanWriterTest)) {
 
 var _ spanstore.Writer = &SpanWriter{} // check API conformance
 
+func TestClientClose(t *testing.T) {
+	withSpanWriter(0, func(w *spanWriterTest) {
+		w.session.On("Close").Return(nil)
+		w.writer.Close()
+		w.session.AssertNumberOfCalls(t, "Close", 1)
+	})
+}
+
 func TestSpanWriter(t *testing.T) {
 	testCases := []struct {
 		caption                        string

plugin/storage/es/spanstore/writer_test.go

@@ -55,6 +55,14 @@ func withSpanWriter(fn func(w *spanWriterTest)) {
 
 var _ spanstore.Writer = &SpanWriter{} // check API conformance
 
+func TestClientClose(t *testing.T) {
+	withSpanWriter(func(w *spanWriterTest) {
+		w.client.On("Close").Return(nil)
+		w.writer.Close()
+		w.client.AssertNumberOfCalls(t, "Close", 1)
+	})
+}
+
 // This test behaves as a large test that checks WriteSpan's behavior as a whole.
 // Extra tests for individual functions are below.
 func TestSpanWriter_WriteSpan(t *testing.T) {