github.com/elfadel/cilium@v1.6.12/pkg/datapath/loader/template_test.go (about) 1 // Copyright 2019 Authors of Cilium 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 // +build !privileged_tests 16 17 package loader 18 19 import ( 20 "bytes" 21 22 "github.com/cilium/cilium/pkg/checker" 23 "github.com/cilium/cilium/pkg/datapath/linux" 24 "github.com/cilium/cilium/pkg/testutils" 25 26 . "gopkg.in/check.v1" 27 ) 28 29 func (s *LoaderTestSuite) TestWrap(c *C) { 30 var ( 31 realEPBuffer bytes.Buffer 32 templateBuffer bytes.Buffer 33 ) 34 35 realEP := testutils.NewTestEndpoint() 36 template := wrap(&realEP, nil) 37 dp := linux.NewDatapath(linux.DatapathConfiguration{}, nil) 38 39 // Write the configuration that should be the same, and verify it is. 40 err := dp.WriteTemplateConfig(&realEPBuffer, &realEP) 41 c.Assert(err, IsNil) 42 err = dp.WriteTemplateConfig(&templateBuffer, template) 43 c.Assert(err, IsNil) 44 c.Assert(realEPBuffer.String(), checker.DeepEquals, templateBuffer.String()) 45 46 // Write with the static data, and verify that the buffers differ. 47 // Note this isn't an overly strong test because it only takes one 48 // character to change for this test to pass, but we would ideally 49 // define every bit of static data differently in the templates. 50 realEPBuffer.Reset() 51 templateBuffer.Reset() 52 err = dp.WriteEndpointConfig(&realEPBuffer, &realEP) 53 c.Assert(err, IsNil) 54 err = dp.WriteEndpointConfig(&templateBuffer, template) 55 c.Assert(err, IsNil) 56 c.Assert(realEPBuffer.String(), Not(Equals), templateBuffer.String()) 57 }