How to use write method of toxiproxy Package

Best Toxiproxy code snippet using toxiproxy.write

proxy_test.go

Source:proxy_test.go Github

copy

Full Screen

1package toxiproxy_test2import (3 "bytes"4 "encoding/hex"5 "io"6 "io/ioutil"7 "net"8 "testing"9 "time"10 "github.com/Shopify/toxiproxy"11 "github.com/sirupsen/logrus"12 tomb "gopkg.in/tomb.v1"13)14func init() {15 logrus.SetLevel(logrus.FatalLevel)16}17func NewTestProxy(name, upstream string) *toxiproxy.Proxy {18 proxy := toxiproxy.NewProxy()19 proxy.Name = name20 proxy.Listen = "localhost:0"21 proxy.Upstream = upstream22 return proxy23}24func WithTCPServer(t *testing.T, f func(string, chan []byte)) {25 ln, err := net.Listen("tcp", "localhost:0")26 if err != nil {27 t.Fatal("Failed to create TCP server", err)28 }29 defer ln.Close()30 response := make(chan []byte, 1)31 tomb := tomb.Tomb{}32 go func() {33 defer tomb.Done()34 src, err := ln.Accept()35 if err != nil {36 select {37 case <-tomb.Dying():38 default:39 t.Fatal("Failed to accept client")40 }41 return42 }43 ln.Close()44 val, err := ioutil.ReadAll(src)45 if err != nil {46 t.Fatal("Failed to read from client")47 }48 response <- val49 }()50 f(ln.Addr().String(), response)51 tomb.Killf("Function body finished")52 ln.Close()53 tomb.Wait()54 close(response)55}56func TestSimpleServer(t *testing.T) {57 WithTCPServer(t, func(addr string, response chan []byte) {58 conn, err := net.Dial("tcp", addr)59 if err != nil {60 t.Error("Unable to dial TCP server", err)61 }62 msg := []byte("hello world")63 _, err = conn.Write(msg)64 if err != nil {65 t.Error("Failed writing to TCP server", err)66 }67 err = conn.Close()68 if err != nil {69 t.Error("Failed to close TCP connection", err)70 }71 resp := <-response72 if !bytes.Equal(resp, msg) {73 t.Error("Server didn't read bytes from client")74 }75 })76}77func WithTCPProxy(t *testing.T, f func(proxy net.Conn, response chan []byte, proxyServer *toxiproxy.Proxy)) {78 WithTCPServer(t, func(upstream string, response chan []byte) {79 proxy := NewTestProxy("test", upstream)80 proxy.Start()81 conn := AssertProxyUp(t, proxy.Listen, true)82 f(conn, response, proxy)83 proxy.Stop()84 })85}86func TestProxySimpleMessage(t *testing.T) {87 WithTCPProxy(t, func(conn net.Conn, response chan []byte, proxy *toxiproxy.Proxy) {88 msg := []byte("hello world")89 _, err := conn.Write(msg)90 if err != nil {91 t.Error("Failed writing to TCP server", err)92 }93 err = conn.Close()94 if err != nil {95 t.Error("Failed to close TCP connection", err)96 }97 resp := <-response98 if !bytes.Equal(resp, msg) {99 t.Error("Server didn't read correct bytes from client", resp)100 }101 })102}103func TestProxyToDownUpstream(t *testing.T) {104 proxy := NewTestProxy("test", "localhost:20009")105 proxy.Start()106 conn := AssertProxyUp(t, proxy.Listen, true)107 // Check to make sure the connection is closed108 conn.SetReadDeadline(time.Now().Add(100 * time.Millisecond))109 _, err := conn.Read(make([]byte, 1))110 if err != io.EOF {111 t.Error("Proxy did not close connection when upstream down", err)112 }113 proxy.Stop()114}115func TestProxyBigMessage(t *testing.T) {116 WithTCPProxy(t, func(conn net.Conn, response chan []byte, proxy *toxiproxy.Proxy) {117 buf := make([]byte, 32*1024)118 msg := make([]byte, len(buf)*2)119 hex.Encode(msg, buf)120 _, err := conn.Write(msg)121 if err != nil {122 t.Error("Failed writing to TCP server", err)123 }124 err = conn.Close()125 if err != nil {126 t.Error("Failed to close TCP connection", err)127 }128 resp := <-response129 if !bytes.Equal(resp, msg) {130 t.Error("Server didn't read correct bytes from client", resp)131 }132 })133}134func TestProxyTwoPartMessage(t *testing.T) {135 WithTCPProxy(t, func(conn net.Conn, response chan []byte, proxy *toxiproxy.Proxy) {136 msg1 := []byte("hello world")137 msg2 := []byte("hello world")138 _, err := conn.Write(msg1)139 if err != nil {140 t.Error("Failed writing to TCP server", err)141 }142 _, err = conn.Write(msg2)143 if err != nil {144 t.Error("Failed writing to TCP server", err)145 }146 err = conn.Close()147 if err != nil {148 t.Error("Failed to close TCP connection", err)149 }150 msg1 = append(msg1, msg2...)151 resp := <-response152 if !bytes.Equal(resp, msg1) {153 t.Error("Server didn't read correct bytes from client", resp)154 }155 })156}157func TestClosingProxyMultipleTimes(t *testing.T) {158 WithTCPProxy(t, func(conn net.Conn, response chan []byte, proxy *toxiproxy.Proxy) {159 proxy.Stop()160 proxy.Stop()161 proxy.Stop()162 })163}164func TestStartTwoProxiesOnSameAddress(t *testing.T) {165 WithTCPProxy(t, func(conn net.Conn, response chan []byte, proxy *toxiproxy.Proxy) {166 proxy2 := NewTestProxy("proxy_2", "localhost:3306")167 proxy2.Listen = proxy.Listen168 if err := proxy2.Start(); err == nil {169 t.Fatal("Expected an err back from start")170 }171 })172}173func TestStopProxyBeforeStarting(t *testing.T) {174 WithTCPServer(t, func(upstream string, response chan []byte) {175 proxy := NewTestProxy("test", upstream)176 AssertProxyUp(t, proxy.Listen, false)177 proxy.Stop()178 err := proxy.Start()179 if err != nil {180 t.Error("Proxy failed to start", err)181 }182 err = proxy.Start()183 if err != toxiproxy.ErrProxyAlreadyStarted {184 t.Error("Proxy did not fail to start when already started", err)185 }186 AssertProxyUp(t, proxy.Listen, true)187 proxy.Stop()188 AssertProxyUp(t, proxy.Listen, false)189 })190}191func TestProxyUpdate(t *testing.T) {192 WithTCPServer(t, func(upstream string, response chan []byte) {193 proxy := NewTestProxy("test", upstream)194 err := proxy.Start()195 if err != nil {196 t.Error("Proxy failed to start", err)197 }198 AssertProxyUp(t, proxy.Listen, true)199 before := proxy.Listen200 input := &toxiproxy.Proxy{Listen: "localhost:0", Upstream: proxy.Upstream, Enabled: true}201 err = proxy.Update(input)202 if err != nil {203 t.Error("Failed to update proxy", err)204 }205 if proxy.Listen == before || proxy.Listen == input.Listen {206 t.Errorf("Proxy update didn't change listen address: %s to %s", before, proxy.Listen)207 }208 AssertProxyUp(t, proxy.Listen, true)209 input.Listen = proxy.Listen210 err = proxy.Update(input)211 if err != nil {212 t.Error("Failed to update proxy", err)213 }214 AssertProxyUp(t, proxy.Listen, true)215 input.Enabled = false216 err = proxy.Update(input)217 if err != nil {218 t.Error("Failed to update proxy", err)219 }220 AssertProxyUp(t, proxy.Listen, false)221 })222}223func TestRestartFailedToStartProxy(t *testing.T) {224 WithTCPServer(t, func(upstream string, response chan []byte) {225 proxy := NewTestProxy("test", upstream)226 conflict := NewTestProxy("test2", upstream)227 err := conflict.Start()228 if err != nil {229 t.Error("Proxy failed to start", err)230 }231 AssertProxyUp(t, conflict.Listen, true)232 proxy.Listen = conflict.Listen233 err = proxy.Start()234 if err == nil || err == toxiproxy.ErrProxyAlreadyStarted {235 t.Error("Proxy started when it should have conflicted")236 }237 conflict.Stop()238 AssertProxyUp(t, conflict.Listen, false)239 err = proxy.Start()240 if err != nil {241 t.Error("Proxy failed to start after conflict went away", err)242 }243 AssertProxyUp(t, proxy.Listen, true)244 proxy.Stop()245 AssertProxyUp(t, proxy.Listen, false)246 })247}248func AssertProxyUp(t *testing.T, addr string, up bool) net.Conn {249 conn, err := net.Dial("tcp", addr)250 if err != nil && up {251 t.Error("Expected proxy to be up:", err)252 } else if err == nil && !up {253 t.Error("Expected proxy to be down")254 }255 return conn256}...

Full Screen

Full Screen

timeout_test.go

Source:timeout_test.go Github

copy

Full Screen

...32 }33 defer conn.Close()34 serverConn := <-serverConnRecv35 defer serverConn.Close()36 writeAndReceive := func(from, to net.Conn) {37 data := []byte("foobar")38 _, err := from.Write(data)39 if err != nil {40 t.Fatal(err)41 }42 err = testhelper.TimeoutAfter(time.Second, func() {43 resp := make([]byte, len(data))44 to.Read(resp)45 if !bytes.Equal(resp, data) {46 t.Fatalf("expected '%s' but got '%s'", string(data), string(resp))47 }48 })49 if err != nil {50 t.Fatal(err)51 }52 }53 // Make sure we can send and receive data before continuing.54 writeAndReceive(conn, serverConn)55 writeAndReceive(serverConn, conn)56 f(conn, serverConn, proxy)57}58func TestTimeoutToxicDoesNotCauseHang(t *testing.T) {59 WithEstablishedProxy(t, func(conn, _ net.Conn, proxy *toxiproxy.Proxy) {60 proxy.Toxics.AddToxicJson(ToxicToJson(t, "might_block", "latency", "upstream", &toxics.LatencyToxic{Latency: 10}))61 proxy.Toxics.AddToxicJson(ToxicToJson(t, "timeout", "timeout", "upstream", &toxics.TimeoutToxic{Timeout: 0}))62 for i := 0; i < 5; i++ {63 _, err := conn.Write([]byte("hello"))64 if err != nil {65 t.Fatal("Unable to write to proxy", err)66 }67 time.Sleep(200 * time.Millisecond) // Shitty sync waiting for latency toxi to get data.68 }69 err := testhelper.TimeoutAfter(time.Second, func() {70 proxy.Toxics.RemoveToxic("might_block")71 })72 if err != nil {73 t.Fatal(err)74 }75 })76}77func TestTimeoutToxicClosesConnectionOnRemove(t *testing.T) {78 WithEstablishedProxy(t, func(conn, serverConn net.Conn, proxy *toxiproxy.Proxy) {79 proxy.Toxics.AddToxicJson(ToxicToJson(t, "to_delete", "timeout", "upstream", &toxics.TimeoutToxic{Timeout: 0}))...

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 toxiproxyClient := toxiproxy.NewClient("localhost:8474")4 toxiproxyClient.CreateProxy("redis", "localhost:6379", "localhost:16379")5 toxiproxyClient.CreateProxy("mysql", "localhost:3306", "localhost:13306")6 toxiproxyClient.CreateProxy("postgres", "localhost:5432", "localhost:15432")7 toxiproxyClient.CreateProxy("memcached", "localhost:11211", "localhost:11111")8 toxiproxyClient.CreateProxy("elasticsearch", "localhost:9200", "localhost:19200")9 toxiproxyClient.CreateProxy("mongodb", "localhost:27017", "localhost:127017")10 toxiproxyClient.CreateProxy("rabbitmq", "localhost:5672", "localhost:15672")11 toxiproxyClient.CreateProxy("cassandra", "localhost:9042", "localhost:19042")12 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7000", "localhost:17000")13 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7001", "localhost:17001")14 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7002", "localhost:17002")15 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7003", "localhost:17003")16 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7004", "localhost:17004")17 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7005", "localhost:17005")18 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7006", "localhost:17006")19 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7007", "localhost:17007")20 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7008", "localhost:17008")21 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7009", "localhost:17009")22 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7010", "localhost:17010")23 toxiproxyClient.CreateProxy("redis-cluster", "localhost:7011", "localhost:17011

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 toxiproxyClient.CreateProxy("my_proxy", "localhost:5000", "localhost:5001")4 toxiproxyClient.EnableProxy("my_proxy")5 toxiproxyClient.WriteToxic("my_proxy", "latency", "downstream", toxiproxy.Toxic{6 Attributes: map[string]interface{}{"latency": 1000, "jitter": 200},7 })8 toxiproxyClient.WriteToxic("my_proxy", "latency", "upstream", toxiproxy.Toxic{9 Attributes: map[string]interface{}{"latency": 1000, "jitter": 200},10 })11 toxiproxyClient.DeleteProxy("my_proxy")12}13import (14func main() {15 toxiproxyClient.CreateProxy("my_proxy", "localhost:5000", "localhost:5001")16 toxiproxyClient.EnableProxy("my_proxy")17 toxiproxyClient.WriteToxic("my_proxy", "latency", "downstream", toxiproxy.Toxic{18 Attributes: map[string]interface{}{"latency": 1000, "jitter": 200},19 })20 toxiproxyClient.WriteToxic("my_proxy", "latency", "upstream", toxiproxy.Toxic{

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 toxiproxyClient.CreateProxy("myproxy", "localhost:6379", "localhost:6380")4 toxiproxyClient.AddToxic("myproxy", "latency", "downstream", 0, toxiproxy.Attributes{"latency": 500, "jitter": 200})5 toxiproxyClient.AddToxic("myproxy", "bandwidth", "downstream", 0, toxiproxy.Attributes{"rate": 102400})6 toxiproxyClient.AddToxic("myproxy", "slow_close", "downstream", 0, toxiproxy.Attributes{"delay": 500})7 toxiproxyClient.AddToxic("myproxy", "timeout", "downstream", 0, toxiproxy.Attributes{"timeout": 500})8 toxiproxyClient.AddToxic("myproxy", "slicer", "downstream", 0, toxiproxy.Attributes{"average_size": 1024, "size_variation": 10, "delay": 100})9 toxiproxyClient.AddToxic("myproxy", "limit_data", "downstream", 0, toxiproxy.Attributes{"bytes": 1024})10 toxiproxyClient.AddToxic("myproxy", "limit_data", "upstream", 0, toxiproxy.Attributes{"bytes": 1024})11 toxiproxyClient.AddToxic("myproxy", "limit_data", "downstream", 0, toxiproxy.Attributes{"bytes": 1024})12 toxiproxyClient.AddToxic("myproxy", "limit_data", "upstream", 0, toxiproxy.Attributes{"bytes": 1024})13 toxiproxyClient.AddToxic("myproxy", "limit_data", "downstream", 0, toxiproxy.Attributes{"bytes": 1024})14 toxiproxyClient.AddToxic("myproxy", "limit_data", "upstream", 0, toxiproxy.Attributes{"bytes": 1024})15 toxiproxyClient.AddToxic("myproxy", "limit_data", "downstream", 0, toxiproxy.Attributes{"bytes": 1024})

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 client, err := toxiproxy.NewClient("localhost:8474")4 if err != nil {5 fmt.Println("Error creating client:", err)6 }7 proxy, err := client.CreateProxy("redis", "localhost:6379", "localhost:16379")8 if err != nil {9 fmt.Println("Error creating proxy:", err)10 }11 fmt.Println("Proxy created:", proxy.Name)12 latencyToxic, err := proxy.Toxics().Create("latency", "downstream", -1, toxiproxy.Attributes{"latency": 1000, "jitter": 500})13 if err != nil {14 fmt.Println("Error creating latency toxic:", err)15 }16 fmt.Println("Latency toxic added:", latencyToxic.Name)17 bandwidthToxic, err := proxy.Toxics().Create("bandwidth", "downstream", -1, toxiproxy.Attributes{"rate": 1024})18 if err != nil {19 fmt.Println("Error creating bandwidth toxic:", err)20 }21 fmt.Println("Bandwidth toxic added:", bandwidthToxic.Name)22 toxics, err := proxy.Toxics().All()23 if err != nil {24 fmt.Println("Error listing toxics:", err)25 }26 fmt.Println("Toxics:")27 for _, toxic := range toxics {28 fmt.Println(" ", toxic.Name)29 }30 err = proxy.Toxics().Delete("latency")31 if err != nil {32 fmt.Println("Error deleting toxic:", err)33 }34 fmt.Println("Latency toxic deleted")35 err = client.DeleteProxy("redis")36 if err != nil {37 fmt.Println("Error deleting proxy:", err)38 }39 fmt.Println("Proxy deleted")40}

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 client := toxiproxy.NewClient("localhost:8474")4 toxics, err := client.Toxics("proxy1")5 if err != nil {6 panic(err)7 }8 toxic := toxiproxy.Toxic{9 Attributes: toxiproxy.Attributes{10 },11 }12 err = toxics.Create(&toxic)13 if err != nil {14 panic(err)15 }16}

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 c, err := client.NewClient("localhost:8474")4 if err != nil {5 fmt.Println(err)6 }7 err = c.Reset()8 if err != nil {9 fmt.Println(err)10 }11 proxy, err := c.CreateProxy("test", "localhost:12345", "localhost:8080")12 if err != nil {13 fmt.Println(err)14 }15 proxy.AddToxic("latency", "latency", "downstream", 1, client.Attributes{16 })17 proxy.AddToxic("slow_close", "slow_close", "downstream", 1, client.Attributes{18 })19 time.Sleep(1 * time.Second)20 proxy.RemoveToxic("latency")21 proxy.RemoveToxic("slow_close")22}

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 proxy, err := client.CreateProxy("my_proxy", "localhost:8080", "localhost:8081")4 if err != nil {5 panic(err)6 }7 fmt.Println("Proxy created")8 time.Sleep(5 * time.Second)9 err = proxy.Delete()10 if err != nil {11 panic(err)12 }13 fmt.Println("Proxy deleted")14}15import (16func main() {17 proxy, err := client.CreateProxy("my_proxy", "localhost:8080", "localhost:8081")18 if err != nil {19 panic(err)20 }21 fmt.Println("Proxy created")22 time.Sleep(5 * time.Second)23 err = proxy.Delete()24 if err != nil {25 panic(err)26 }27 fmt.Println("Proxy deleted")28}29import (30func main() {31 proxy, err := client.CreateProxy("my_proxy", "localhost:8080", "localhost:8081")32 if err != nil {33 panic(err)34 }35 fmt.Println("Proxy created")36 time.Sleep(5 * time.Second)37 err = proxy.Delete()38 if err != nil {39 panic(err)40 }41 fmt.Println("Proxy deleted")42}43import (44func main() {45 proxy, err := client.CreateProxy("my_proxy", "localhost:8080", "localhost:8081

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1func main() {2 toxiproxy := toxiproxy.Toxiproxy{3 }4 err := client.AddProxy(toxiproxy)5 if err != nil {6 log.Fatal(err)7 }8 err = client.SetLatency("test", 2000)9 if err != nil {10 log.Fatal(err)11 }12 err = client.SetBandwidth("test", 1000000)13 if err != nil {14 log.Fatal(err)15 }16 err = client.SetSlowClose("test", 2000)17 if err != nil {18 log.Fatal(err)19 }20 err = client.SetTimeout("test", 2000)21 if err != nil {22 log.Fatal(err)23 }24 err = client.SetLatency("test", 2000)25 if err != nil {26 log.Fatal(err)27 }28 err = client.SetBandwidth("test", 1000000)29 if err != nil {30 log.Fatal(err)31 }32 err = client.SetSlowClose("test", 2000)33 if err != nil {34 log.Fatal(err)35 }36 err = client.SetTimeout("test", 2000)37 if err != nil {38 log.Fatal(err)39 }40 err = client.SetLatency("test", 2000)41 if err != nil {42 log.Fatal(err)43 }44 err = client.SetBandwidth("test", 1000000)45 if err != nil {46 log.Fatal(err)47 }48 err = client.SetSlowClose("

Full Screen

Full Screen

write

Using AI Code Generation

copy

Full Screen

1proxy := client.Proxy("test")2err := proxy.Create()3if err != nil {4 panic(err)5}6proxy := client.Proxy("test")7err := proxy.Create()8if err != nil {9 panic(err)10}11proxy := client.Proxy("test")12err := proxy.Create()13if err != nil {14 panic(err)15}16proxy := client.Proxy("test")17err := proxy.Create()18if err != nil {19 panic(err)20}21proxy := client.Proxy("test")22err := proxy.Create()23if err != nil {24 panic(err)25}

Full Screen

Full Screen

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run Toxiproxy automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

... View sample repo... View DocumentationSign up Free
›
_

Most used method in

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

...
Sign up with GoogleSign up with Email
Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful